Update app.py

app.py

@@ -73,19 +73,33 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
     return response, updated_history
 
 # Create Gradio Interface
-demo = gr.ChatInterface(
-    fn=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)"),
-    ],
-    inputs=["text", "state", "state", "number", "number", "number"],
-    outputs=["text", "state"],
-    title="Chatbot with BlenderBot",
-    description="A chatbot interface using the facebook/blenderbot-400M-distill model."
-)
+demo = gr.Blocks()
+
+with demo:
+    gr.Markdown("# Chatbot with BlenderBot")
+
+    with gr.Row():
+        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)")
+
+    chatbot = gr.Chatbot(label="Chatbot Interface")
+    state = gr.State([])
+
+    with gr.Row():
+        user_input = gr.Textbox(placeholder="Type a message...", show_label=False)
+        submit_btn = gr.Button("Submit")
+
+    def submit_message(user_input, history, system_message, max_tokens, temperature, top_p):
+        response, history = respond(user_input, history, system_message, max_tokens, temperature, top_p)
+        return gr.update(chatbot=history), history
+
+    submit_btn.click(
+        submit_message,
+        inputs=[user_input, state, system_message, max_tokens, temperature, top_p],
+        outputs=[chatbot, state]
+    )
 
 if __name__ == "__main__":
     demo.launch()