Update app.py

app.py

@@ -56,9 +56,9 @@ def handle_chat(user_message):
         # Get AI response
         ai_response, visual_response = chat_handler.answer_question(user_message.strip(),visual_query)
         if visual_query is not None:
-            user_message = f"<b style='color:#6366f1;'>user:</b> {visual_query.strip()}"
+            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()}"
+            user_message = f"<b style='color:#6366f1;'>USER:</b> {user_message.strip()}"
         
         if visual_response:
             from PIL import Image
@@ -76,7 +76,7 @@ def handle_chat(user_message):
 
             ai_response =   img + ai_response  
             
-            user_message = visual_query.strip()
+            user_message = f"<b style='color:#6366f1;'>USER:</b> {visual_query.strip()}"
         
         ai_response = f"<b style='color:#6366f1;'>AI:</b> {ai_response}"
         # Add AI response to chat history