Upload 2 files

app.py

@@ -133,9 +133,23 @@ def load_profile():
         return "Profile not found."
 
 def respond(message, chat_history):
-    answer = query_rag(message, vector_store, hf_client)
+    if not message:
+        return "", chat_history
+    
+    # Initialize history if None (though Gradio usually handles this)
+    if chat_history is None:
+        chat_history = []
+        
+    # Append user message immediately for responsiveness
     chat_history.append({"role": "user", "content": message})
-    chat_history.append({"role": "assistant", "content": answer})
+    
+    try:
+        # Get AI response
+        answer = query_rag(message, vector_store, hf_client)
+        chat_history.append({"role": "assistant", "content": answer})
+    except Exception as e:
+        chat_history.append({"role": "assistant", "content": f"Error: {str(e)}"})
+        
     return "", chat_history
 
 with gr.Blocks(title="My AI Twin", theme=gr.themes.Soft()) as demo: