Update app.py

app.py

@@ -107,17 +107,16 @@ def respond(
         response += token
         yield response
 
-demo = gr.Interface(
-    fn=respond,
-    inputs=[
-        gr.Textbox(label="Input message"),
+demo = gr.ChatInterface(
+    respond,
+
+    additional_inputs=[
         gr.Textbox(value="", label="System message", visible=False),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens", visible=False),
         gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature", visible=False),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P", visible=False),
         gr.Dropdown(choices=list(mood_prompts.keys()), label="Mood", value="Casual"),
     ],
-    outputs=gr.Textbox(label="Response"),
     css=css,
     theme="allenai/gradio-theme",
 )