Update app.py

app.py

@@ -348,6 +348,7 @@ class ChatManager:
             if len(st.session_state.uploaded_files) == 0:
                 response, status = ChatManager._handle_product_name(user_input)
             else:
+                print("Calling handle_product_url")
                 response, status = ChatManager._handle_product_url()
                 
         return response, status
@@ -377,6 +378,7 @@ class ChatManager:
                 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)
             
             uploaded_files = st.file_uploader(
@@ -387,18 +389,17 @@ class ChatManager:
 
             print(f"DEBUG : uploaded_files after st.file_uploader() : {uploaded_files}")
             
-            if uploaded_files:
+            if len(uploaded_files) > 0:
                 st.session_state.uploaded_files = uploaded_files
                 st.session_state.awaiting_image_upload = False
                 print(f"DEBUG: User uploaded files: {uploaded_files}")
                 return f"{len(uploaded_files)} images uploaded for analysis.", "no success"
             
-            #return "Waiting for image upload...", "no success"
             else:
                 # Show a temporary message until files are uploaded
-                print(f"DEBUG: No uploaded files!")
                 st.info("Waiting for images to be uploaded.")
-                #st.stop()
+                print("Waiting for images to be uploaded!")
+                st.stop()
                 return "Waiting for images to be uploaded!", "no success"
 
     @staticmethod
@@ -456,6 +457,7 @@ def main():
         selection_in_progress = ProductSelector.handle_selection()
     
     # Only show chat input if not awaiting selection
+    print(f"Selection in progress ? : {selection_in_progress}")
     if not selection_in_progress:
         user_input = st.chat_input("Enter your message:", key="user_input")
         print(f"DEBUG - user_input : {user_input}")