Update app.py

app.py

@@ -107,25 +107,20 @@ def respond(
         response += token
         yield response
 
-# Layout with separate row for Mood
-with gr.Blocks(css=css, theme="allenai/gradio-theme") as demo:
-    with gr.Row():
-        system_message = gr.Textbox(value="", label="System message")
-        max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
-    with gr.Row():
-        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")
-    with gr.Row():
-        mood = gr.Dropdown(
-            choices=list(mood_prompts.keys()),
-            label="Mood",
-            value="Casual"
-        )
-    
-    gr.ChatInterface(
-        fn=respond,
-        inputs=[system_message, max_tokens, temperature, top_p, mood],
-    )
+demo = gr.ChatInterface(
+    respond,
+    additional_inputs=[
+        gr.Row(gr.Textbox(value="", label="System message")),
+        gr.Row(
+            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.Row(gr.Dropdown(choices=list(mood_prompts.keys()), label="Mood", value="Casual")),
+    ],
+    css=css,
+    theme="allenai/gradio-theme",
+)
 
 if __name__ == "__main__":
     demo.launch()