Update app.py

app.py

@@ -406,6 +406,7 @@ def chatbot_response(image_urls_str, product_name_by_user, data_extractor_url, s
     image_urls = []
     if product_name_by_user != "":
         similar_product_list_json = get_product_list(product_name_by_user, data_extractor_url)
+        
         if similar_product_list_json and extract_info == False:
             with st.spinner("Fetching product information from our database... This may take a moment."):
                 print(f"similar_product_list_json : {similar_product_list_json}")
@@ -460,6 +461,12 @@ if "welcome_msg" not in st.session_state:
     st.session_state.welcome_msg = "Welcome to ConsumeWise! What product would you like me to analyze today?"
 if "system_prompt" not in st.session_state:
     st.session_state.system_prompt = ""
+if "similar_products" not in st.session_state:
+    st.session_state.similar_products = []
+if "awaiting_selection" not in st.session_state:
+    st.session_state.awaiting_selection = False
+if "current_user_input" not in st.session_state:
+    st.session_state.current_user_input = ""
 
 # Sidebar input for system prompt with an 'Apply' button
 st.sidebar.header("System Prompt")
@@ -474,7 +481,6 @@ system_prompt = st.sidebar.text_area(
 if st.sidebar.button("Submit Prompt"):
     if system_prompt.strip():
         st.session_state.system_prompt = system_prompt
-        
         # Reset session state when system prompt is changed
         st.session_state.product_selected = False
         st.session_state.analyze_more = True
@@ -482,8 +488,11 @@ if st.sidebar.button("Submit Prompt"):
         st.session_state.yes_no_choice = None
         st.session_state.product_shared = False
         st.session_state.welcome_msg = "Welcome to ConsumeWise! What product would you like me to analyze today?"
-        st.session_state.messages = []  # Clear chat history when prompt changes
-        st.rerun()  # Rerun the app to apply the changes
+        st.session_state.messages = []
+        st.session_state.similar_products = []
+        st.session_state.awaiting_selection = False
+        st.session_state.current_user_input = ""
+        st.rerun()
     else:
         st.sidebar.error("Please enter a valid system prompt.")
 
@@ -492,7 +501,7 @@ st.title("ConsumeWise - Your Food Product Analysis Assistant")
 
 # Show warning if system prompt is empty and disable chat input
 if not st.session_state.system_prompt.strip():
-    st.sidebar.error("Please enter a system prompt before starting the conversation.")
+    st.sidebar.error("Please enter a system prompt in the sidebar before starting the conversation.")
     st.warning("⚠️ Please enter a system prompt in the sidebar before proceeding.")
     st.chat_input("Enter your message:", disabled=True)
 else:
@@ -504,81 +513,77 @@ else:
                 st.markdown(message["content"])
         st.markdown("</div>", unsafe_allow_html=True)
 
-    # Function to display the welcome message
     def show_welcome_message():
-        """Display the welcome message at the start of the conversation."""
         if not st.session_state.welcome_shown:
             st.session_state.messages.append({"role": "assistant", "content": st.session_state.welcome_msg})
             with st.chat_message("assistant"):
                 st.markdown(st.session_state.welcome_msg)
             st.session_state.welcome_shown = True
 
-    # Function to handle user input from chat box
-    def handle_user_input():
-        """Handle user input from the chat box."""
-        user_input = st.session_state.user_input.strip()
-        if user_input:
-            st.session_state.messages.append({"role": "user", "content": user_input})
-            with st.chat_message("user"):
-                st.markdown(user_input)
-            return user_input
-        return None
-
-    # Simulate product selection and analysis logic
+    def handle_radio_selection():
+        if st.session_state.awaiting_selection and st.session_state.similar_products:
+            choice = st.radio(
+                "Select a product:",
+                st.session_state.similar_products + ["None of the above"],
+                key="product_choice"
+            )
+            
+            if choice:
+                if choice != "None of the above":
+                    st.session_state.product_selected = True
+                    st.session_state.awaiting_selection = False
+                    _, msg = chatbot_response("", st.session_state.current_user_input, data_extractor_url, system_prompt, extract_info=True)
+                    st.session_state.messages.append({"role": "assistant", "content": msg})
+                    st.rerun()
+                else:
+                    st.session_state.awaiting_selection = False
+                    st.session_state.messages.append(
+                        {"role": "assistant", "content": "Please provide the image URL of the product."}
+                    )
+                    st.rerun()
+
     def chat_response(user_input, system_prompt):
-        """Simulate product selection and analysis logic."""
         if not st.session_state.product_selected:
-            # Fetch similar products
             if "http:/" not in user_input and "https:/" not in user_input:
                 st.session_state.product_shared = True
+                st.session_state.current_user_input = user_input
                 similar_products, msg = chatbot_response("", user_input, data_extractor_url, system_prompt, extract_info=False)
+                
                 if len(similar_products) > 0:
-                    st.session_state.messages.append(
-                        {"role": "assistant", "content": "Here are some similar products. Please select one or enter 'None':"}
-                    )
-                    with st.chat_message("assistant"):
-                        st.markdown("Here are some similar products. Please select one or enter 'None':")
-            
-                    # Show product selection radio only once
-                    choice = st.radio("Select a product:", similar_products + ["None of the above"], key="product_choice")
-                    if choice and choice != "None of the above":
-                        st.session_state.product_selected = True
-                        _, msg = chatbot_response("", user_input, data_extractor_url, system_prompt, extract_info=True)
-                        return msg
-                        
-                    elif choice == "None of the above":
-                        return "Please provide the image URL of the product."
+                    st.session_state.similar_products = similar_products
+                    st.session_state.awaiting_selection = True
+                    return "Here are some similar products. Please select:"
                 else:
-                    return "Product not found in our database. Please provide the image URL of the product (Example : https://m.media-amazon.com/images/I/71043tKaomL._SX679_.jpg)."
+                    return "Product not found in our database. Please provide the image URL of the product."
             
             else:
                 if ("http:/" in user_input or "https:/" in user_input) and (".jpeg" in user_input or ".jpg" in user_input) and st.session_state.product_shared:
                     _, msg = chatbot_response(user_input, "", data_extractor_url, system_prompt, extract_info=True)
-                        
                     st.session_state.product_selected = True
-                    
-                    st.session_state.messages.append({"role": "assistant", "content": msg})
-                    with st.chat_message("assistant"):
-                        st.markdown(msg)
-                        
-                    return "Next Product"
-
+                    return msg
                 else:
                     if not st.session_state.product_shared:
-                        return("Please provide the product name first")
+                        return "Please provide the product name first"
                     else:
-                        return "Please provide valid image URL of the product (Example : https://m.media-amazon.com/images/I/71043tKaomL._SX679_.jpg)."
+                        return "Please provide valid image URL of the product."
         else:
             return "Next Product"
 
     # Show the welcome message at the start
     show_welcome_message()
 
-    # Input Box for User Interaction - Only enabled if system prompt is not empty
+    # Handle radio button selection
+    handle_radio_selection()
+
+    # Input Box for User Interaction
     user_input = st.chat_input("Enter your message:", key="user_input")
 
     if user_input:
-        response = chat_response(handle_user_input(), system_prompt)
+        st.session_state.messages.append({"role": "user", "content": user_input})
+        with st.chat_message("user"):
+            st.markdown(user_input)
+
+        response = chat_response(user_input, system_prompt)
         
         if response != "Next Product":
             st.session_state.messages.append({"role": "assistant", "content": response})
@@ -591,6 +596,9 @@ else:
             st.session_state.yes_no_choice = None
             st.session_state.product_shared = False
             st.session_state.welcome_msg = "What is the next product you would like me to analyze today?"
+            st.session_state.similar_products = []
+            st.session_state.awaiting_selection = False
+            st.session_state.current_user_input = ""
             st.rerun()
 
 # Option to Clear Chat History