Update app.py

app.py

@@ -322,7 +322,7 @@ def analyze_claims(claims, ingredients, product_name, assistant_id):
     
     return claims_analysis_str
 
-def generate_final_analysis(brand_name, product_name, processing_level, harmful_ingredient_analysis, claims_analysis, system_prompt):
+def generate_final_analysis(brand_name, product_name, nutrient_analysis, processing_level, harmful_ingredient_analysis, claims_analysis, system_prompt):
     global debug_mode, client
     system_prompt_orig = """You are provided with a detailed analysis of a food product. Your task is to generate actionable insights to help the user decide whether to consume the product, at what frequency, and identify any potential harms or benefits. Consider the context of consumption to ensure the advice is personalized and practical.
 
@@ -354,6 +354,9 @@ Additionally, consider the following while generating insights:
     user_prompt = f"""
 Product Name: {brand_name} {product_name}
 
+Nutrition Analysis :
+{nutrient_analysis}
+
 Processing Level:
 {processing_level}
 
@@ -390,13 +393,13 @@ def analyze_product(product_info_raw, system_prompt):
         if len(ingredients_list) > 0:
             product_type, calories, sugar, salt, serving_size = find_product_type(product_info_from_db)
             nutrient_analysis = analyze_nutrients(product_type, calories, sugar, salt, serving_size)
-            print(f"nutrient analysis is {nutrient_analysis}")
+            print(f"DEBUG ! nutrient analysis is {nutrient_analysis}")
             processing_level = analyze_processing_level(ingredients_list, brand_name, product_name, assistant1.id) if ingredients_list else ""
             harmful_ingredient_analysis = analyze_harmful_ingredients(ingredients_list, brand_name, product_name, assistant2.id) if ingredients_list else ""
         if len(claims_list) > 0:                    
             claims_analysis = analyze_claims(claims_list, ingredients_list, product_name, assistant3.id) if claims_list else ""
                 
-        final_analysis = generate_final_analysis(brand_name,product_name,processing_level,harmful_ingredient_analysis,claims_analysis, system_prompt)
+        final_analysis = generate_final_analysis(brand_name, product_name, nutrient_analysis, 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."    
@@ -534,11 +537,12 @@ class ProductSelector:
                         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"]
+                        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.messages.append(
@@ -644,11 +648,12 @@ def main():
             if response == "Next Product":
                 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"]
+                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?"
                 st.rerun()
                 
             elif response:  # Only add response if it's not None