Update app.py

app.py

@@ -355,31 +355,33 @@ class ChatManager:
 
     @staticmethod
     def _handle_product_name(user_input):
-        st.session_state.product_shared = True
-        st.session_state.current_user_input = user_input
-        similar_products, _ = chatbot_response(
-            [], user_input, extract_info=False
-        )
-        
-        st.session_state.similar_products = similar_products
-
-        if len(st.session_state.similar_products) > 0:
-            st.session_state.awaiting_selection = True
-            return "Here are some similar products from our database. Please select:", "no success"
-
-        
-        # Add a file uploader to allow users to upload multiple images
-        # No similar products found
-        st.session_state.awaiting_image_upload = True
+        if not st.session_state.awaiting_image_upload:
+            st.session_state.product_shared = True
+            st.session_state.current_user_input = user_input
+            similar_products, _ = chatbot_response(
+                [], user_input, extract_info=False
+            )
+            
+            st.session_state.similar_products = similar_products
+    
+            if len(st.session_state.similar_products) > 0:
+                st.session_state.awaiting_selection = True
+                return "Here are some similar products from our database. Please select:", "no success"
+    
+            
+            # Add a file uploader to allow users to upload multiple images
+            # No similar products found
+            st.session_state.awaiting_image_upload = True
     
         # Only show message and uploader if waiting for image upload
         if st.session_state.awaiting_image_upload:
-            with st.chat_message("assistant"):
-                st.markdown(f"Please provide images of the product since {len(st.session_state.similar_products)} similar products found in our database")
-            # Append the message to session state for chat history
-            st.session_state.messages.append({"role": "assistant", "content": f"Please provide images of the product since {len(st.session_state.similar_products)} similar products found in our database"})
+            if user_input:
+                with st.chat_message("assistant"):
+                    st.markdown(f"Please provide images of the product since {len(st.session_state.similar_products)} similar products found in our database")
+                # Append the message to session state for chat history
+                st.session_state.messages.append({"role": "assistant", "content": f"Please provide images of the product since {len(st.session_state.similar_products)} similar products found in our database"})
             
-            st.chat_input("Please upload files ...", disabled=True)
+                st.chat_input("Please upload files ...", disabled=True)
             
             uploaded_files = st.file_uploader(
                     "Upload product images here:",