Update app.py

app.py

@@ -313,13 +313,16 @@ class ProductSelector:
                         st.experimental_rerun()
                         
                     if (choice == "None of the above" or msg == "product not found because product information in the db is corrupt") and len(st.session_state.uploaded_files) == 0:
-                        st.session_state.messages.append(
-                            {"role": "assistant", "content": "Please provide the images of the product to analyze based on the latest information."}
-                        )
-                        with st.chat_message("assistant"):
-                            st.markdown("Please provide the images of the product to analyze based on the latest information.")
-                        #st.session_state.selected_product = None
+                        if not st.session_state.awaiting_image_upload:
+                            st.session_state.messages.append(
+                                {"role": "assistant", "content": "Please provide the images of the product to analyze based on the latest information."}
+                            )
+                            with st.chat_message("assistant"):
+                                st.markdown("Please provide the images of the product to analyze based on the latest information.")
+                        
                         # Add a file uploader to allow users to upload multiple images
+                        st.session_state.awaiting_image_upload = True
+                        
                         uploaded_files = st.file_uploader(
                             "Upload product images here:",
                             type=["jpg", "jpeg", "png"],
@@ -334,6 +337,7 @@ class ProductSelector:
                                 st.markdown(f"{len(uploaded_files)} images uploaded for analysis.")
                                 
                             st.session_state.uploaded_files = uploaded_files
+                            st.session_state.awaiting_image_upload = False
                             st.experimental_rerun()
                 
                 # Prevent further chat input while awaiting selection
@@ -377,7 +381,7 @@ class ChatManager:
             st.session_state.awaiting_image_upload = True
     
         # Only show message and uploader if waiting for image upload
-        if st.session_state.awaiting_image_upload:
+        if st.session_state.awaiting_image_upload and len(st.session_state.uploaded_files) == 0:
             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")