Update app.py

app.py

@@ -377,7 +377,8 @@ 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(
                     "Upload product images here:",
                     type=["jpg", "jpeg", "png"],
@@ -392,13 +393,13 @@ class ChatManager:
                 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()  # Prevent further execution until user interacts
-            #    return "Waiting for images to be uploaded!", "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()  # Prevent further execution until user interacts
+                # return "Waiting for images to be uploaded!", "no success"
 
     @staticmethod
     def _handle_product_url():
@@ -485,8 +486,8 @@ def main():
                 
             #else:
             #    print(f"DEBUG : st.session_state.awaiting_selection : {st.session_state.awaiting_selection}")
-            print("Re-running...")
-            st.experimental_rerun()
+        print("Re-running...")
+        st.experimental_rerun()
     else:
         # Disable chat input while selection is in progress
         st.chat_input("Please confirm your selection above first...", disabled=True)