Update app.py

app.py

@@ -9,17 +9,20 @@ from utils import process_image, process_video, generate_references
 
 loader = ModelLoader()
 
-# ===================== CHAT FUNCTION (Classic Tuple Format) =====================
+# ===================== CHAT FUNCTION =====================
 def chat(message: str, history, file=None):
-    """Works with older and current Gradio versions"""
+    """
+    Works with current HF Spaces Gradio (mixed 5/6 behavior)
+    Returns messages format (dict with role/content)
+    """
     if history is None:
         history = []
-    
+
     if not message.strip() and not file:
-        history.append([message, "Please type a message or upload a file."])
+        history.append({"role": "assistant", "content": "Please type a message or upload a file."})
         return history
 
-    # --- Process File (Image/Video) ---
+    # --- Process File ---
     context = ""
     if file:
         try:
@@ -34,9 +37,9 @@ def chat(message: str, history, file=None):
                 summary = process_video(file_path)
                 context = f"**Video Analysis**:\n{summary}\n\n"
             else:
-                context = "**File uploaded successfully.**\n\n"
+                context = "**File uploaded.**\n\n"
         except Exception as e:
-            context = f"**File processing error**: {str(e)[:100]}\n\n"
+            context = f"**File error**: {str(e)[:100]}\n\n"
 
     # --- Generate Response ---
     full_prompt = f"{context}User: {message}\n\nAssistant:"
@@ -62,12 +65,13 @@ def chat(message: str, history, file=None):
         final_response = f"{response_text.strip()}\n\n**Sources / References:**\n{generate_references(message[:100])}"
         
     except Exception as e:
-        final_response = f"⚠️ Error generating response: {str(e)[:150]}"
+        final_response = f"⚠️ Error: {str(e)[:180]}"
 
-    # Append using CLASSIC tuple format (compatible with your Gradio)
-    history.append([message, final_response])
+    # Append in MESSAGES FORMAT (what Gradio currently expects)
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": final_response})
 
-    # Memory cleanup
+    # Cleanup
     gc.collect()
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
@@ -75,7 +79,7 @@ def chat(message: str, history, file=None):
     return history
 
 
-# ===================== GRADIO UI =====================
+# ===================== UI =====================
 with gr.Blocks() as demo:
     gr.Markdown("# 🛡️ Private Personal AI Chatbot\n**Fully Local • Private • Open Source**")
     
@@ -83,11 +87,12 @@ with gr.Blocks() as demo:
         height=650,
         avatar_images=(None, "🧠"),
         label=None
+        # NO 'type' argument → avoids init error
     )
     
     with gr.Row():
         msg = gr.Textbox(
-            placeholder="Ask anything... (Image & Video supported)",
+            placeholder="Ask anything... (supports image & video)",
             scale=4,
             container=False,
             autofocus=True
@@ -109,9 +114,9 @@ with gr.Blocks() as demo:
         inputs=[msg, file_upload]
     )
 
-    gr.Markdown("**Privacy**: Everything runs in memory. No data is stored or logged.")
+    gr.Markdown("**Privacy**: No data is stored or logged.")
 
-    # ===================== EVENT BINDING =====================
+    # ===================== EVENTS =====================
     submit_btn.click(
         chat,
         inputs=[msg, chatbot, file_upload],