Update app.py

app.py

@@ -27,7 +27,7 @@ model.to(device)
 # ----------------------
 def chat_with_bot(message, history):
     try:
-        # Combine history into a conversation string
+        # Convert history to conversation string
         conversation = ""
         for user_msg, bot_msg in history:
             conversation += f"User: {user_msg}\nBot: {bot_msg}\n"
@@ -49,7 +49,7 @@ def chat_with_bot(message, history):
                 early_stopping=True
             )
         
-        # Decode only new part
+        # Decode only the new part
         reply = tokenizer.decode(outputs[:, inputs.shape[-1]:][0], skip_special_tokens=True)
         return reply.strip() if reply.strip() else "I'm not sure how to answer that."
         
@@ -57,24 +57,23 @@ def chat_with_bot(message, history):
         return f"⚠️ Error: {str(e)}"
 
 # ----------------------
-# Gradio ChatBlock
+# Use ChatInterface instead of Blocks
 # ----------------------
-with gr.Blocks() as demo:
-    gr.Markdown("## Fitlien Chatbot 🤖\nChat with your fitness assistant!")
-    
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="Your message")
-    clear_btn = gr.Button("Clear Chat")
-    
-    def respond(message, history):
-        reply = chat_with_bot(message, history)
-        history.append((message, reply))
-        return "", history
-    
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    clear_btn.click(lambda: [], None, chatbot)
-    
-    demo.api_name = "chat"
+demo = gr.ChatInterface(
+    fn=chat_with_bot,
+    title="My Fitlien Chatbot",
+    description="A chatbot fine-tuned from DialoGPT-medium for fitness conversations.",
+    examples=[
+        "Hello! How are you?",
+        "What's a good workout routine?",
+        "Tell me about nutrition",
+        "How can I stay motivated?"
+    ],
+    retry_btn=None,
+    undo_btn="Delete Previous",
+    clear_btn="Clear",
+    submit_btn="Submit"
+)
 
 # Launch
 if __name__ == "__main__":