Update app.py

app.py

@@ -394,13 +394,11 @@ class ChatManager:
                 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"
-            
             else:
                 # Show a temporary message until files are uploaded
                 st.info("Waiting for images to be uploaded.")
                 print("Waiting for images to be uploaded!")
                 st.stop()
-                return "Waiting for images to be uploaded!", "no success"
 
     @staticmethod
     def _handle_product_url():
@@ -460,9 +458,9 @@ def main():
     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}")
+        print(f"DEBUG - user_input : {user_input} and len(st.session_state.uploaded_files) is {len(st.session_state.uploaded_files)}")
         
-        if user_input or len(st.session_state.uploaded_files) > 0:
+        if user_input or st.session_state.awaiting_image_upload:
             # Add user message to chat
             st.session_state.messages.append({"role": "user", "content": user_input})
             with st.chat_message("user"):