Update app.py

app.py

@@ -68,14 +68,7 @@ def setup_chatbot_chain():
 
 chatbot_chain = setup_chatbot_chain()
 
-def respond(
-    message,
-    history,
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-):
+def respond(message, history):
     chat_history = "\n".join([f"Human: {h[0]}\nAI: {h[1]}" for h in history])
     
     response = chatbot_chain.run(
@@ -86,41 +79,44 @@ def respond(
     
     return response
 
-# Customize the ChatInterface
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(
-            value="You are an empathetic and supportive mental health chatbot. Provide warm, comforting responses and suggest coping strategies when appropriate. Always encourage seeking professional help for persistent issues.",
-            label="System message"
-        ),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=1.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)"),
-    ],
-    title="Mental Health Support Chatbot",
-    description="I'm here to listen and provide support. How are you feeling today?",
-    examples=[
-        ["I've been feeling anxious lately"],
-        ["I'm having trouble sleeping"],
-        ["I feel overwhelmed with work"],
-        ["I'm feeling sad and don't know why"]
-    ],
-)
-
-# Add a disclaimer
-disclaimer = gr.Markdown(
-    """
-    **Disclaimer:** This chatbot is for informational purposes only and is not a substitute for professional medical advice, 
-    diagnosis, or treatment. Always seek the advice of your physician or other qualified health provider 
-    with any questions you may have regarding a medical condition.
-    """
-)
-
-# Combine the chat interface and disclaimer
-with gr.Blocks() as combined_interface:
-    demo.render()
-    disclaimer.render()
+with gr.Blocks() as demo:
+    gr.Markdown("# Mental Health Support Chatbot")
+    gr.Markdown("I'm here to listen and provide support. How are you feeling today?")
+
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox()
+    clear = gr.Button("Clear")
+
+    def user(user_message, history):
+        return "", history + [[user_message, None]]
+
+    def bot(history):
+        bot_message = respond(history[-1][0], history[:-1])
+        history[-1][1] = bot_message
+        return history
+
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot, chatbot, chatbot
+    )
+    clear.click(lambda: None, None, chatbot, queue=False)
+
+    gr.Examples(
+        examples=[
+            "I've been feeling anxious lately",
+            "I'm having trouble sleeping",
+            "I feel overwhelmed with work",
+            "I'm feeling sad and don't know why"
+        ],
+        inputs=msg
+    )
+
+    gr.Markdown(
+        """
+        **Disclaimer:** This chatbot is for informational purposes only and is not a substitute for professional medical advice, 
+        diagnosis, or treatment. Always seek the advice of your physician or other qualified health provider 
+        with any questions you may have regarding a medical condition.
+        """
+    )
 
 if __name__ == "__main__":
-    combined_interface.launch()
+    demo.launch()