Update app.py

app.py

@@ -45,18 +45,15 @@ additional_inputs = [
     gr.Slider(label="Repetition penalty", value=1.2, minimum=1.0, maximum=2.0, step=0.05, interactive=True, info="Penalize repeated tokens")
 ]
 
-with gr.Blocks() as demo:
-    chatbot = gr.Chatbot(show_label=True, show_share_button=False, show_copy_button=True, likeable=True, layout="panel")
-    with gr.Row():
-        with gr.Column(scale=3):
-            user_input = gr.Textbox(label="Your message", placeholder="Type your message here...")
-        with gr.Column(scale=1, height=40):
-            submit_button = gr.Button("➢", shape="square")
-
-    submit_button.click(
-        fn=generate,
-        inputs=[user_input, chatbot, gr.Textbox(label="System Prompt", max_lines=1, interactive=True)],
-        outputs=chatbot
-    )
-
-demo.launch(show_api=False)
+gr.ChatInterface(
+    fn=generate,
+    chatbot=gr.Chatbot(show_label=True, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
+    additional_inputs=additional_inputs,
+    title="ConvoLite",
+    submit_btn="➢",
+    retry_btn="Retry",
+    undo_btn="↩ Undo",
+    clear_btn="Clear (New chat)",
+    stop_btn="Stop ▢",
+    concurrency_limit=20,
+).launch(show_api=False)