Update app.py

app.py

@@ -19,18 +19,28 @@ llm = Llama(
 )
 
 # =========================
-# CHAT FUNCTION (GRADIO 3 SAFE)
+# CHAT FUNCTION (STRICT MESSAGES FORMAT)
 # =========================
-def chat(prompt, history):
+def chat(message, history):
     history = history or []
+
     messages = []
 
-    for user_msg, bot_msg in history:
-        messages.append({"role": "user", "content": str(user_msg)})
-        messages.append({"role": "assistant", "content": str(bot_msg)})
+    # SAFE: ensure correct format
+    for msg in history:
+        if isinstance(msg, dict):
+            messages.append({
+                "role": msg.get("role", ""),
+                "content": str(msg.get("content", ""))
+            })
 
-    messages.append({"role": "user", "content": str(prompt)})
+    # add user message
+    messages.append({
+        "role": "user",
+        "content": str(message)
+    })
 
+    # inference
     output = llm.create_chat_completion(
         messages=messages,
         max_tokens=300,
@@ -39,25 +49,27 @@ 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 (NO TYPE PARAMETER)
+# UI (GRADIO 4 SAFE STYLE BUT BACKWARD COMPATIBLE)
 # =========================
 with gr.Blocks() as demo:
     gr.Markdown("# 🧠 Living Legend AI Chatbot")
 
-    chatbot = gr.Chatbot()  # 🔥 FIXED (NO type="messages")
+    chatbot = gr.Chatbot(type="messages")  # IMPORTANT FIX
     msg = gr.Textbox()
     clear = gr.Button("Clear")
 
     msg.submit(chat, [msg, chatbot], [msg, chatbot])
+
     clear.click(lambda: [], None, chatbot)
 
-# =========================
-# LAUNCH
-# =========================
 demo.queue()
 demo.launch(server_name="0.0.0.0", server_port=7860)