Update app.py

app.py

@@ -329,6 +329,7 @@ class ProductSelector:
                             )
                             with st.chat_message("assistant"):
                                 st.markdown(f"{len(uploaded_files)} images uploaded for analysis.")
+                                
                             st.session_state.uploaded_files = uploaded_files
                             st.experimental_rerun()
                 
@@ -343,7 +344,7 @@ class ChatManager:
     def process_response(user_input):
         if not st.session_state.product_selected:
             #if "http:/" not in user_input and "https:/" not in user_input:
-            print(f"DEBUG : st.session_state.uploaded_files : {st.session_state.uploaded_files}")
+            print(f"DEBUG : st.session_state.uploaded_files inside process_response : {st.session_state.uploaded_files}")
             if len(st.session_state.uploaded_files) == 0:
                 response, status = ChatManager._handle_product_name(user_input)
             else:
@@ -361,7 +362,7 @@ class ChatManager:
         
         st.session_state.similar_products = similar_products
 
-        if len(similar_products) > 0:
+        if len(st.session_state.similar_products) > 0:
             st.session_state.awaiting_selection = True
             return "Here are some similar products from our database. Please select:", "no success"
 
@@ -382,7 +383,9 @@ class ChatManager:
                     type=["jpg", "jpeg", "png"],
                     accept_multiple_files=True
                 )
-                        
+
+            print(f"DEBUG : uploaded_files : {uploaded_files}")
+            
             if uploaded_files:
                 st.session_state.uploaded_files = uploaded_files
                 st.session_state.awaiting_image_upload = False
@@ -458,11 +461,12 @@ def main():
             
             # Process response
             response, status = ChatManager.process_response(user_input)
+            print(f"DEBUG - response from process_response is {response}")
             
-            st.session_state.messages.append({"role": "assistant", "content": 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