Misc proposals

app.py

@@ -19,7 +19,6 @@ default_template = """{% for message in messages %}
 {% endif %}"""
 
 description_text = """### This space is a helper app for writing [Chat Templates](https://huggingface.co/docs/transformers/main/en/chat_templating).
-
 ### When you're happy with the outputs from your template, you can use the code block at the end to add it to a PR!"""
 
 def apply_chat_template(template, test_conversation, add_generation_prompt, cleanup_whitespace):
@@ -36,20 +35,29 @@ def apply_chat_template(template, test_conversation, add_generation_prompt, clea
     )
     pr_snippet = "\n".join(pr_snippet)
     formatted = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=add_generation_prompt)
-    return formatted, pr_snippet
-
-iface = gr.Interface(
-    description=description_text,
-    fn=apply_chat_template,
-    inputs=[
-        gr.TextArea(value=default_template, lines=10, max_lines=30, label="Chat Template"),
-        gr.TextArea(value=demo_conversation, lines=6, label="Conversation"),
-        gr.Checkbox(value=False, label="Add generation prompt"),
-        gr.Checkbox(value=True, label="Cleanup template whitespace"),
-    ],
-    outputs=[
-        gr.TextArea(label="Formatted conversation"),
-        gr.TextArea(label="Code snippet to create PR", lines=3, show_label=True, show_copy_button=True)
-    ]
-)
-iface.launch()
+    return formatted #, pr_snippet
+
+def open_pr(template, model_repo):
+  tokenizer = AutoTokenizer.from_pretrained(model_repo)
+  tokenizer.chat_template = template
+  tokenizer.push_to_hub(model_repo, create_pr=True)
+
+
+with gr.Blocks() as demo:
+    gr.Markdown(description_text)
+
+    with gr.Row():
+      with gr.Column():
+          template = gr.TextArea(value=default_template, lines=10, max_lines=30, label="Chat Template")
+          example = gr.TextArea(value=demo_conversation, lines=6, label="Conversation")
+          generate_prompt = gr.Checkbox(value=False, label="Add generation prompt")
+          clean_whitespace = gr.Checkbox(value=True, label="Cleanup template whitespace")
+          btn = gr.Button("Submit")
+      with gr.Column():
+        output = gr.TextArea(label="Formatted conversation")
+        model_repo = gr.Textbox(label='Model repo to open a PR')
+        btn_pr = gr.Button("Open a PR with template update")
+    btn.click(fn=apply_chat_template, inputs=[template, example, generate_prompt, clean_whitespace], outputs=[output])
+    btn_pr.click(fn=open_pr, inputs=[template, model_repo])
+
+demo.launch()