Update app.py

app.py

@@ -54,7 +54,11 @@ def handle_chat(user_message):
             visual_query = None    
         # Get AI response
         ai_response, visual_response = chat_handler.answer_question(user_message.strip(),visual_query)
-        user_message = f"<b style='color:#6366f1;'>user:</b> {user_message.strip()}"
+        if visual_query is not None:
+            user_message = f"<b style='color:#6366f1;'>user:</b> {visual_query.strip()}"
+        else:
+            user_message = f"<b style='color:#6366f1;'>user:</b> {user_message.strip()}"
+        
         if visual_response:
             from PIL import Image
             import base64