Update app.py

app.py

@@ -631,10 +631,14 @@ def main():
                 SessionState.initialize()  # Reset states for next product
                 st.session_state.welcome_msg = "What is the next product you would like me to analyze today?"
                 st.rerun()
+                
             elif response:  # Only add response if it's not None
                 st.session_state.messages.append({"role": "assistant", "content": response})
                 with st.chat_message("assistant"):
                     st.markdown(response)
+                print(f"DEBUG : st.session_state.awaiting_selection : {st.session_state.awaiting_selection}")
+                print(f"response : {response}")
+                st.rerun()
     else:
         # Disable chat input while selection is in progress
         st.chat_input("Please confirm your selection above first...", disabled=True)