Update app.py

app.py

@@ -515,8 +515,8 @@ def show_welcome_message():
         st.session_state.welcome_shown = True
 
 def handle_product_selection():
-    print(f"DEBUG : st.session_state.similar_products : {st.session_state.similar_products}")
-    print(f"DEBUG : st.session_state.awaiting_selection : {st.session_state.awaiting_selection}")
+    print(f"DEBUG Inside handle_product_selection: st.session_state.similar_products : {st.session_state.similar_products}")
+    print(f"DEBUG Inside handle_product_selection: st.session_state.awaiting_selection : {st.session_state.awaiting_selection}")
     
     if st.session_state.similar_products and st.session_state.awaiting_selection:
         choice = st.radio(
@@ -528,10 +528,10 @@ def handle_product_selection():
         # Add a confirm button
         if st.button("Confirm Selection"):
             if choice != "None of the above":
+                print(f"Selected choice : {choice}")
                 st.session_state.product_selected = True
                 st.session_state.awaiting_selection = False
                 st.session_state.selected_product = choice
-                print(f"Selected choice : {choice}")
                 _, msg = chatbot_response("", choice, data_extractor_url, system_prompt, extract_info=True)
                 st.session_state.messages.append({"role": "assistant", "content": msg})
             else:
@@ -539,8 +539,10 @@ def handle_product_selection():
                 st.session_state.messages.append(
                     {"role": "assistant", "content": "Please provide the image URL of the product."}
                 )
+            
             st.session_state.similar_products = []  # Clear the products list
-            st.rerun()
+            #print("DEBUG : Re-running the code")
+            #st.rerun()
 
 def chat_response(user_input, system_prompt):
     if not st.session_state.product_selected:
@@ -583,11 +585,6 @@ else:
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
     
-    # Handle product selection if awaiting
-    #print(f"Are you awaiting_selection ??  : {st.session_state.awaiting_selection}")
-    #if st.session_state.awaiting_selection:
-    #    handle_product_selection()
-    
     # Input Box for User Interaction
     user_input = st.chat_input("Enter your message:", key="user_input")
 
@@ -607,6 +604,7 @@ else:
             print(f"Are you awaiting_selection ??  : {st.session_state.awaiting_selection}")
             if st.session_state.awaiting_selection:
                 handle_product_selection()
+                print("After executing handle_product_selection")
         
         else:
             st.session_state.product_selected = False