Update app.py

app.py

@@ -19,34 +19,28 @@ llm = Llama(
 )
 
 # =========================
-# SAFE CHAT FUNCTION
+# SAFE CHAT FUNCTION (NO MIXING)
 # =========================
-def chat(prompt, history):
+def chat(message, history):
     history = history or []
-    messages = []
-
-    # Convert Gradio history safely (tuple-based)
-    for item in history:
-        if isinstance(item, tuple) and len(item) == 2:
-            user_msg, bot_msg = item
 
-            messages.append({
-                "role": "user",
-                "content": str(user_msg)
-            })
+    messages = []
 
+    # STRICT: only dict format allowed
+    for msg in history:
+        if isinstance(msg, dict):
             messages.append({
-                "role": "assistant",
-                "content": str(bot_msg)
+                "role": msg.get("role", "user"),
+                "content": str(msg.get("content", ""))
             })
 
-    # Add current message
+    # add new user message
     messages.append({
         "role": "user",
-        "content": str(prompt)
+        "content": str(message)
     })
 
-    # Call model
+    # run model
     output = llm.create_chat_completion(
         messages=messages,
         max_tokens=300,
@@ -55,17 +49,21 @@ def chat(prompt, history):
 
     response = output["choices"][0]["message"]["content"]
 
-    history.append((prompt, response))
+    # append assistant reply
+    messages.append({
+        "role": "assistant",
+        "content": response
+    })
 
-    return "", history
+    return "", messages
 
 # =========================
-# UI (GRADIO SAFE MODE)
+# UI (GRADIO 4 SAFE MODE)
 # =========================
 with gr.Blocks() as demo:
-    gr.Markdown("# 🧠 Qwen GGUF AI (Living Legend System)")
+    gr.Markdown("# 🧠 Qwen GGUF AI (Living Legend Stable Build)")
 
-    chatbot = gr.Chatbot()  # Gradio 3 safe mode
+    chatbot = gr.Chatbot(type="messages")
     msg = gr.Textbox(label="Ask your AI")
     clear = gr.Button("Clear")
 
@@ -74,7 +72,7 @@ with gr.Blocks() as demo:
     clear.click(lambda: [], None, chatbot, queue=False)
 
 # =========================
-# LAUNCH SAFE
+# LAUNCH
 # =========================
 demo.queue()
 demo.launch(server_name="0.0.0.0", server_port=7860)