Update app.py

app.py

@@ -358,23 +358,24 @@ class ChatManager:
             [], user_input, extract_info=False
         )
         
-        
+        st.session_state.similar_products = similar_products
+
         if len(similar_products) > 0:
-            st.session_state.similar_products = similar_products
             st.session_state.awaiting_selection = True
             return "Here are some similar products from our database. Please select:", "no success"
 
-        st.session_state.messages.append({"role": "assistant", "content": f"Please provide images of the product since {len(similar_products)} similar products found in our database"})
-        with st.chat_message("assistant"):
-            st.markdown(f"Please provide images of the product since {len(similar_products)} similar products found in our database")
+        
         # 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:
-            st.write(f"Please provide images of the product since {len(st.session_state.similar_products)} similar products found in our database")
-        
+        while 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"})
+
             uploaded_files = st.file_uploader(
                     "Upload product images here:",
                     type=["jpg", "jpeg", "png"],
@@ -385,8 +386,8 @@ class ChatManager:
                 st.session_state.uploaded_files = uploaded_files
                 st.session_state.awaiting_image_upload = False
                 return f"{len(uploaded_files)} images uploaded for analysis.", "no success"
-            else:          
-                return "Waiting for images!", "no success"
+            #else:          
+            #    return "Waiting for images!", "no success"
 
     @staticmethod
     def _handle_product_url():