updated

app.py

@@ -1,21 +1,25 @@
 import gradio as gr
 from transformers import pipeline
 
-# -------- TEXT MODEL (LIGHT + SAFE) --------
+# -------- MODEL --------
 text_model = pipeline("text-generation", model="distilgpt2")
 
-# -------- SIMPLE FUNCTION --------
+# -------- FUNCTION --------
 def respond(user_input, history):
-    # text response only (no crashes)
     result = text_model(user_input, max_length=80)[0]["generated_text"]
-    history.append((user_input, result))
+
+    history = history + [
+        {"role": "user", "content": user_input},
+        {"role": "assistant", "content": result}
+    ]
+
     return history, history
 
 # -------- UI --------
 with gr.Blocks() as demo:
-    gr.Markdown("# Simple AI Chat (Stable Version)")
+    gr.Markdown("# Simple AI Chat (Fixed)")
 
-    chatbot = gr.Chatbot()
+    chatbot = gr.Chatbot(type="messages")  # ✅ IMPORTANT
     state = gr.State([])
 
     with gr.Row():