Update app.py

app.py

@@ -55,27 +55,22 @@ def clear_chat():
 """
 For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
 """
-with gr.Blocks() as demo:
-    chatbot = gr.Chatbot(label="Chat")
-    system_message = gr.Textbox(value="You are a friendly Chatbot.", label="System message")
-    max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
-    temperature = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
-    top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
-    personality = gr.Dropdown(choices=["friendly", "professional", "humorous", "serious"], value="friendly", label="Personality")
-    message = gr.Textbox(placeholder="Enter your message here...", label="Message")
-    submit_button = gr.Button("Send Message")
-    clear_button = gr.Button("Clear Chat")
-
-    submit_button.click(
-        respond,
-        inputs=[message, chatbot, system_message, max_tokens, temperature, top_p, personality],  # Pass personality as input
-        outputs=[chatbot]
-    )
-
-    clear_button.click(
-        clear_chat,
-        outputs=[message, chatbot]
-    )
+demo = gr.ChatInterface(
+    respond,
+    additional_inputs=[
+        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
+        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
+        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
+        gr.Slider(
+            minimum=0.1,
+            maximum=1.0,
+            value=0.95,
+            step=0.05,
+            label="Top-p (nucleus sampling)",
+        ),
+        gr.Dropdown(choices=["friendly", "professional", "humorous", "serious"], value="friendly", label="Personality"),
+   ]
+)
 
 
 if __name__ == "__main__":