Update app.py

app.py

@@ -514,7 +514,9 @@ The output must be in JSON format as follows:
     #              claims_not_found_in_doc.append(key)
     #    print(f"Claims not found in the doc are {','.join(claims_not_found_in_doc)}")
     #claims_analysis = json.loads(message_content.value.replace("```", "").replace("json", "").replace("(NOT FOUND IN DOCUMENT) ", ""))
-    claims_analysis = json.loads(message_content.value.replace("```", "").replace("json", ""))
+    claims_analysis = {}
+    if message_content.value != "":
+        claims_analysis = json.loads(message_content.value.replace("```", "").replace("json", ""))
 
     claims_analysis_str = ""
     for key, value in claims_analysis.items():
@@ -604,7 +606,6 @@ def analyze_product(product_info_raw, system_prompt):
         
         if nutritional_information:
             product_type, calories, sugar, salt, serving_size = find_product_nutrients(product_info_from_db)
-            nutrient_analysis = analyze_nutrients(product_type, calories, sugar, salt, serving_size)
             if product_type is not None and serving_size is not None:                                                          
                 nutrient_analysis = analyze_nutrients(product_type, calories, sugar, salt, serving_size)                       
             else:                                                                                                              
@@ -632,8 +633,8 @@ def analyze_product(product_info_raw, system_prompt):
         final_analysis = generate_final_analysis(brand_name, product_name, nutritional_level, processing_level, harmful_ingredient_analysis, claims_analysis, system_prompt)
 
         return final_analysis
-    else:
-        return "I'm sorry, product information could not be extracted from the url."    
+    #else:
+    #    return "I'm sorry, product information could not be extracted from the url."    
 
 # Streamlit app
 # Initialize session state
@@ -758,29 +759,37 @@ class ProductSelector:
                 
                 # Confirm button
                 confirm_clicked = st.button("Confirm Selection")
-                
+                msg = ""
                 # Only process the selection when confirm is clicked
                 if confirm_clicked:
+                    st.session_state.awaiting_selection = False
                     if choice != "None of the above":
-                        st.session_state.product_selected = True
-                        st.session_state.awaiting_selection = False
-                        st.session_state.selected_product = choice
-                        _, msg = chatbot_response("", choice, data_extractor_url, 
-                                                st.session_state.system_prompt, extract_info=True)
-                        st.session_state.messages.append({"role": "assistant", "content": msg})
-
-                        # Loop through session state keys and delete all except the key_to_keep
-                        keys_to_keep = ["system_prompt", "messages", "welcome_msg"]
-                        keys_to_delete = [key for key in st.session_state.keys() if key not in keys_to_keep]
-                    
-                        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:
-                        st.session_state.awaiting_selection = False
+                        #st.session_state.selected_product = choice
+                        st.session_state.messages.append({"role": "assistant", "content": f"You selected {choice}"})
+                        _, msg = chatbot_response("", choice, data_extractor_url="", extract_info=True)
+                        #Check if analysis couldn't be done because db had incomplete information
+                        if msg != "product not found because product information in the db is corrupt":
+                            #Only when msg is acceptable
+                            st.session_state.messages.append({"role": "assistant", "content": msg})
+                            with st.chat_message("assistant"):
+                                st.markdown(msg)
+                                
+                            st.session_state.product_selected = True
+                            
+                            keys_to_keep = ["system_prompt", "messages", "welcome_msg"]
+                            keys_to_delete = [key for key in st.session_state.keys() if key not in keys_to_keep]
+                        
+                            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?"
+                            
+                    if choice == "None of the above" or msg == "product not found because product information in the db is corrupt":
                         st.session_state.messages.append(
-                            {"role": "assistant", "content": "Please provide the image URL of the product."}
+                            {"role": "assistant", "content": "Please provide the image URL of the product to analyze based on the latest information."}
                         )
+                        with st.chat_message("assistant"):
+                            st.markdown("Please provide the image URL of the product to analyze based on the latest information.")
+                        #st.session_state.selected_product = None
                         
                     st.rerun()
                 
@@ -795,10 +804,11 @@ 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:
-                return ChatManager._handle_product_name(user_input)
+                response, status = ChatManager._handle_product_name(user_input)
             else:
-                return ChatManager._handle_product_url(user_input)
-        return "Next Product"
+                response, status = ChatManager._handle_product_url(user_input)
+                
+        return response, status
 
     @staticmethod
     def _handle_product_name(user_input):
@@ -809,11 +819,12 @@ class ChatManager:
             st.session_state.system_prompt, extract_info=False
         )
         
-        if similar_products:
+        if len(similar_products) > 0:
             st.session_state.similar_products = similar_products
             st.session_state.awaiting_selection = True
-            return "Here are some similar products from our database. Please select:"
-        return "Product not found in our database. Please provide the image URL of the product."
+            return "Here are some similar products from our database. Please select:", "no success"
+            
+        return "Product not found in our database. Please provide the image URL of the product.", "no success"
 
     @staticmethod
     def _handle_product_url(user_input):
@@ -829,9 +840,19 @@ class ChatManager:
                 st.session_state.system_prompt, extract_info=True
             )
             st.session_state.product_selected = True
-            return msg
+            if msg != "product not found because image is not clear" and "Product information could not be extracted from the image" not in msg:
+                response = msg
+                status = "success"
+            elif msg == "product not found because image is not clear":
+                response = msg + ". Please share clear image URLs!"
+                status = "no success"
+            else:
+                response = msg + ".Please re-try!!"
+                status = "no success"
+                
+            return response, status
             
-        return "Please provide valid image URL of the product."
+        return "Please provide valid image URL of the product.", "no success"
 
 def main():
     # Initialize session state
@@ -876,9 +897,13 @@ def main():
                 st.markdown(user_input)
             
             # Process response
-            response = ChatManager.process_response(user_input)
+            response, status = ChatManager.process_response(user_input)
+
+            st.session_state.messages.append({"role": "assistant", "content": response})
+            with st.chat_message("assistant"):
+                st.markdown(response)
             
-            if response == "Next Product":
+            if status == "success":
                 SessionState.initialize()  # Reset states for next product
                 #st.session_state.welcome_msg = "What is the next product you would like me to analyze today?"
                 keys_to_keep = ["system_prompt", "messages", "welcome_msg"]
@@ -887,15 +912,11 @@ 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?"
-                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()
+            #elif response:  # Only add response if it's not None
+            #    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)