Update app.py

app.py

@@ -71,7 +71,7 @@ mood_prompts = {
 
 def respond(
     message,
-    history,
+    history: list[tuple[str, str]],
     system_message,
     max_tokens,
     temperature,
@@ -107,42 +107,27 @@ def respond(
         response += token
         yield response
 
-# Define the layout
-def create_interface():
-    with gr.Blocks(css=css, theme="allenai/gradio-theme") as demo:
-        gr.Markdown("# Chat Interface")
-        
-        with gr.Row():
-            with gr.Column():
-                message_input = gr.Textbox(placeholder="Type your message here...", label="Message")
-                history_input = gr.Textbox(placeholder="Chat history (if any)...", label="History")
-                system_message_input = gr.Textbox(value="", label="System message")
-                max_tokens_slider = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
-                temperature_slider = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
-                top_p_slider = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P")
-
-        with gr.Row():
-            mood_dropdown = gr.Dropdown(
-                choices=list(mood_prompts.keys()),
-                label="Mood",
-                value="Casual"
-            )
-        
-        chat_interface = gr.Interface(
-            fn=respond,
-            inputs=[
-                message_input,
-                history_input,
-                system_message_input,
-                max_tokens_slider,
-                temperature_slider,
-                top_p_slider,
-                mood_dropdown
-            ],
-            outputs=gr.Textbox(label="Response")
+demo = gr.ChatInterface(
+    respond,
+    additional_inputs=[
+        gr.Textbox(value="", 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",
+        ),
+        gr.Dropdown(
+            choices=list(mood_prompts.keys()),
+            label="Mood",
+            value="Casual"
         )
-        
-        chat_interface.launch()
-
+    ],
+    css=css,
+    theme="allenai/gradio-theme",
+)
 if __name__ == "__main__":
-    create_interface()
+    demo.launch()