Update app.py

app.py

@@ -19,28 +19,18 @@ llm = Llama(
 )
 
 # =========================
-# SAFE CHAT FUNCTION (NO MIXING)
+# CHAT FUNCTION (GRADIO 3 SAFE)
 # =========================
-def chat(message, history):
+def chat(prompt, history):
     history = history or []
-
     messages = []
 
-    # STRICT: only dict format allowed
-    for msg in history:
-        if isinstance(msg, dict):
-            messages.append({
-                "role": msg.get("role", "user"),
-                "content": str(msg.get("content", ""))
-            })
+    for user_msg, bot_msg in history:
+        messages.append({"role": "user", "content": str(user_msg)})
+        messages.append({"role": "assistant", "content": str(bot_msg)})
 
-    # add new user message
-    messages.append({
-        "role": "user",
-        "content": str(message)
-    })
+    messages.append({"role": "user", "content": str(prompt)})
 
-    # run model
     output = llm.create_chat_completion(
         messages=messages,
         max_tokens=300,
@@ -49,27 +39,22 @@ def chat(message, history):
 
     response = output["choices"][0]["message"]["content"]
 
-    # append assistant reply
-    messages.append({
-        "role": "assistant",
-        "content": response
-    })
+    history.append((prompt, response))
 
-    return "", messages
+    return "", history
 
 # =========================
-# UI (GRADIO 4 SAFE MODE)
+# UI (NO TYPE PARAMETER)
 # =========================
 with gr.Blocks() as demo:
-    gr.Markdown("# 🧠 Qwen GGUF AI (Living Legend Stable Build)")
+    gr.Markdown("# 🧠 Living Legend AI Chatbot")
 
-    chatbot = gr.Chatbot(type="messages")
-    msg = gr.Textbox(label="Ask your AI")
+    chatbot = gr.Chatbot()  # 🔥 FIXED (NO type="messages")
+    msg = gr.Textbox()
     clear = gr.Button("Clear")
 
     msg.submit(chat, [msg, chatbot], [msg, chatbot])
-
-    clear.click(lambda: [], None, chatbot, queue=False)
+    clear.click(lambda: [], None, chatbot)
 
 # =========================
 # LAUNCH