Update app.py

app.py

@@ -1,11 +1,14 @@
 # =========================================
-# FLAN-T5 Chatbot (100% Stable - FINAL)
+# FLAN-T5 Dialogue System (Gradio)
 # =========================================
 
 import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
+# -----------------------------
+# 1. Load Model
+# -----------------------------
 MODEL_NAME = "google/flan-t5-base"
 
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
@@ -16,12 +19,15 @@ model = model.to(device)
 
 
 # -----------------------------
-# Chat Function (IMPORTANT)
+# 2. Chat Function
 # -----------------------------
 def chat(message, history):
-    
+    if history is None:
+        history = []
+
+    # 🔥 Better prompt for conversational response
     prompt = f"""
-You are a helpful AI assistant.
+You are a helpful and friendly AI assistant.
 Answer clearly and naturally.
 
 User: {message}
@@ -46,16 +52,43 @@ Assistant:
 
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-    return response
+    # ✅ Gradio v4 format
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": response})
+
+    return history, ""
+
+
+# -----------------------------
+# 3. Reset Function
+# -----------------------------
+def reset_chat():
+    return [], ""
 
 
 # -----------------------------
-# Gradio Chat Interface (🔥 FIX)
+# 4. UI
 # -----------------------------
-demo = gr.ChatInterface(
-    fn=chat,
-    title="🤖 AI Dialogue System (FLAN-T5)",
-    description="Chat with AI using FLAN-T5"
-)
+with gr.Blocks() as demo:
+    gr.Markdown("## 🤖 AI Dialogue System (FLAN-T5)")
 
+    chatbot = gr.Chatbot(type="messages")
+
+    msg = gr.Textbox(
+        placeholder="Type your message...",
+        label="Your Message"
+    )
+
+    with gr.Row():
+        send = gr.Button("Send")
+        clear = gr.Button("Clear Chat")
+
+    send.click(chat, [msg, chatbot], [chatbot, msg])
+    msg.submit(chat, [msg, chatbot], [chatbot, msg])
+    clear.click(reset_chat, outputs=[chatbot, msg])
+
+
+# -----------------------------
+# 5. Launch
+# -----------------------------
 demo.launch()