Update app.py

app.py

@@ -197,7 +197,8 @@ def format_markdown(text):
 class ConversationState:
     def __init__(self):
         self.reset()
-    
+        self.given_diet = ""
+
     def reset(self):
         self.current_step = "greeting"
         self.user_profile = {
@@ -211,6 +212,7 @@ class ConversationState:
             "restrictions": [],
             "allergies": []
         }
+        self.given_diet = ""
 
 
 session_states = {}
@@ -433,9 +435,11 @@ def chat_with_diet_bot(message, history, session_id=None):
             
             Generate a complete meal plan with:
             1. Diet Overview - Brief summary of the diet approach
-            2. Weekly Meal Plan - Day-by-day plan with breakfast, lunch, dinner, and snacks. Add calories and macros for each day!
+            2. Weekly Meal Plan - Day-by-day plan with breakfast, lunch, dinner, and snacks (sometimes).
             3. General Recommendations - Additional advice and nutrition tips
             
+            For every day, give an estimated calories and protein value!
+            
             Format in Markdown with clear headers and bullet points.
             """
             
@@ -448,9 +452,9 @@ def chat_with_diet_bot(message, history, session_id=None):
                 ).generations[0].text
                 
                 formatted_response = format_markdown(cohere_response)
+                state.given_diet = formatted_response
                 response = formatted_response + "\n\nI hope this meal plan helps you achieve your goals! Feel free to ask if you have any nutrition questions."
                 state.current_step = "diet_questions"
-                GIVEN_DIET = response
                 
             except Exception:
                 response = "I'm sorry, I couldn't generate a diet plan at the moment. Please try again later."
@@ -480,7 +484,7 @@ def chat_with_diet_bot(message, history, session_id=None):
         prompt = f"""
         Answer this nutrition question clearly and accurately:
         Question: {message}
-        Previous given diet: {GIVEN_DIET}
+        Previous given diet: {state.given_diet}
         Provide helpful, evidence-based advice with practical tips.
         """