Update app.py

app.py

@@ -560,9 +560,9 @@ for message in st.session_state.messages:
 
 if product_name_by_user != "":
     # Add user message to chat history
-    st.session_state.messages.append({"role": "user", "content": image_urls_str})
+    st.session_state.messages.append({"role": "user", "content": product_name_by_user})
     with st.chat_message("user"):
-        st.markdown(image_urls_str)
+        st.markdown(product_name_by_user)
 
     # Prevent further input while generating response
     st.session_state["response_pending"] = True
@@ -577,6 +577,12 @@ if product_name_by_user != "":
             
     if len(similar_product_list) == 0 or choice == "None of the above":
         image_urls_str = st.text_input("Enter image URLs separated by commas")
+        
+        # Add user message to chat history
+        st.session_state.messages.append({"role": "user", "content": image_urls_str})
+        with st.chat_message("user"):
+            st.markdown(image_urls_str)
+            
         response = chatbot_response(image_urls_str, "", data_extractor_url, extract_info = True)
         
     st.session_state.messages.append({"role": "assistant", "content": response})