Proper output formats

app.py

@@ -28,12 +28,11 @@ def apply_chat_template(template, test_conversation1, test_conversation2):
     outputs = []
     for i, conversation_str in enumerate((test_conversation1, test_conversation2)):
         conversation = json.loads(conversation_str)
-        without_gen = tokenizer.apply_chat_template(conversation, tokenize=False)
-        with_gen = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
-        out = f"Conversation {i}:\n\n{conversation_str}\n\nOutput without generation prompt:\n\n{without_gen}\n\nOutput with generation prompt:\n\n{with_gen}\n\n"
-        outputs.append(out)
+        outputs.append(tokenizer.apply_chat_template(conversation, tokenize=False))
+        outputs.append(with_gen = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True))
     return tuple(outputs)
 
+output_names = ["Conversation 1 without generation prompt", "Conversation 1 with generation prompt", "Conversation 2 without generation prompt", "Conversation 2 with generation prompt"]
 iface = gr.Interface(
     fn=apply_chat_template,
     inputs=[
@@ -41,5 +40,5 @@ iface = gr.Interface(
         gr.TextArea(value=str(demo_conversation1), lines=5, label="Conversation 1"),
         gr.TextArea(value=str(demo_conversation2), lines=5, label="Conversation 2")
     ],
-    outputs=["text", "text"])
+    outputs=[gr.TextArea, label=output_name for output_name in output_names])
 iface.launch()