Update app.py

app.py

@@ -370,7 +370,7 @@ class ChatManager:
         st.session_state.awaiting_image_upload = True
     
         # Only show message and uploader if waiting for image upload
-        while st.session_state.awaiting_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
@@ -386,8 +386,11 @@ 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:
+                # Show a temporary message until files are uploaded
+                st.info("Waiting for images to be uploaded.")
+                st.stop()  # Prevent further execution until user interacts
+                #return "Waiting for images!", "no success"
 
     @staticmethod
     def _handle_product_url():