Update app.py

app.py

@@ -397,8 +397,8 @@ class ChatManager:
             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
+                st.info("Waiting for images to be uploaded.")
+                #st.stop()
                 return "Waiting for images to be uploaded!", "no success"
 
     @staticmethod
@@ -470,11 +470,12 @@ def main():
             print(f"DEBUG - Calling process_response with user_input : {user_input}")
             response, status = ChatManager.process_response(user_input)
             print(f"DEBUG - response from process_response is {response}")
-                
-            with st.chat_message("assistant"):
-                st.markdown(response)
-            st.session_state.messages.append({"role": "assistant", "content": response})
 
+            if "Waiting for images to be uploaded" not in response:
+                with st.chat_message("assistant"):
+                    st.markdown(response)
+                st.session_state.messages.append({"role": "assistant", "content": response})
+                
             if status == "success":               
                 SessionState.initialize()  # Reset states for next product
 
@@ -484,7 +485,8 @@ def main():
                 for key in keys_to_delete:
                     del st.session_state[key]
                 st.session_state.welcome_msg = "What product would you like me to analyze next?"
-
+                
+                
             print("Re-running...")
             st.experimental_rerun()
     else: