Update app.py

app.py

@@ -1,17 +1,16 @@
 import gradio as gr
 from transformers import pipeline
 
-# Better model for conversation
+# Better chatbot model
 chatbot = pipeline(
-    "text-generation",
+    "text2text-generation",
     model="google/flan-t5-small"
 )
 
 def reply(message, history):
     if not message.strip():
-        return history
+        return "Please enter a message."
 
-    # Instruction-based prompt
     prompt = f"User: {message}\nAssistant:"
 
     result = chatbot(
@@ -23,16 +22,19 @@ def reply(message, history):
 
     output = result[0]["generated_text"]
 
-    # Extract only assistant reply
-    response = output.split("Assistant:")[-1].strip()
+    # Extract only assistant response
+    if "Assistant:" in output:
+        response = output.split("Assistant:")[-1].strip()
+    else:
+        response = output.strip()
 
-    history.append((message, response))
-    return history
+    return response  # ✅ ONLY RETURN STRING
 
+# Chat UI
 demo = gr.ChatInterface(
     fn=reply,
     title="💬 Smart Dialogue System",
-    description="Instruction-based chatbot using FLAN-T5"
+    description="Chatbot using FLAN-T5"
 )
 
 demo.launch(server_name="0.0.0.0", server_port=7860)