Input fixes

app.py

@@ -1,17 +1,18 @@
 import gradio as gr
 from transformers import AutoTokenizer
+import json
 
 tokenizer = AutoTokenizer.from_pretrained("HuggingFaceH4/zephyr-7b-beta")
 
-demo_conversation1 = [
+demo_conversation1 = """[
     {"role": "user", "content": "Hi there!"},
     {"role": "assistant", "content": "Hello, human!"}
-]
+]"""
 
-demo_conversation2 = [
+demo_conversation2 = """[
     {"role": "system", "content": "You are a helpful chatbot."},
     {"role": "user", "content": "Hi there!"}
-]
+]"""
 
 default_template = """{% for message in messages %}
 {{ "<|im_start|>" + message["role"] + "\\n" + message["content"] + "<|im_end|>\\n" }}
@@ -22,18 +23,23 @@ default_template = """{% for message in messages %}
 
 conversations = [demo_conversation1, demo_conversation2]
 
-def apply_chat_template(template):
+def apply_chat_template(template, test_conversation1, test_conversation2):
     tokenizer.chat_template = template
     outputs = []
-    for i, conversation in enumerate(conversations):
+    for i, conversation_str in enumerate(test_conversation1, test_conversation2):
+        conversation = json.loads(conversation)
         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} without generation prompt:\n\n{without_gen}\n\nConversation {i} with generation prompt:\n\n{with_gen}\n\n"
+        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)
     return tuple(outputs)
 
 iface = gr.Interface(
     fn=apply_chat_template,
-    inputs=gr.TextArea(value=default_template, lines=10, max_lines=30, label="Chat Template"),
-    outputs=["text"] * len(conversations))
+    inputs=[
+        gr.TextArea(value=default_template, lines=10, max_lines=30, label="Chat Template"),
+        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"])
 iface.launch()