Update app.py

app.py

@@ -405,11 +405,11 @@ class ChatManager:
     def _handle_product_url():
         #is_valid_url = (".jpeg" in user_input or ".jpg" in user_input) and \
         #               ("http:/" in user_input or "https:/" in user_input)
-        image_len = len(st.session_state.uploaded_files)
+        
         if not st.session_state.product_shared:
             return "Please provide the product name first"
         
-        if image_len > 1 and st.session_state.product_shared:
+        if len(st.session_state.uploaded_files) > 1 and st.session_state.product_shared:
             _, msg = chatbot_response(
                 st.session_state.uploaded_files, "", extract_info=True
             )
@@ -426,7 +426,7 @@ class ChatManager:
                 
             return response, status
                 
-        #return "Please provide valid image URL of the product.", "no success"
+        st.session_state.uploaded_files = []
         return "Please provide more than 1 images of the product to capture complete information.", "no success"
 
 def main():
@@ -459,7 +459,8 @@ def main():
     if not selection_in_progress:
         user_input = st.chat_input("Enter your message:", key="user_input")
         print(f"DEBUG - user_input : {user_input}")
-        if user_input:
+        
+        if user_input or len(st.session_state.uploaded_files) > 0:
             # Add user message to chat
             st.session_state.messages.append({"role": "user", "content": user_input})
             with st.chat_message("user"):
@@ -469,7 +470,7 @@ 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})
@@ -483,11 +484,9 @@ 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?"
-                
-            #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)