Update app.py

app.py

@@ -11,6 +11,8 @@ load_dotenv()
 COHERE_API_KEY = os.getenv("COHERE_API_KEY")
 co = cohere.Client(COHERE_API_KEY)
 
+GIVEN_DIET = ""
+
 custom_css = """
 .gradio-container {
     font-family: 'Inter', -apple-system, BlinkMacSystemFont, sans-serif;
@@ -431,7 +433,7 @@ 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
+            2. Weekly Meal Plan - Day-by-day plan with breakfast, lunch, dinner, and snacks. Add calories and macros for each day!
             3. General Recommendations - Additional advice and nutrition tips
             
             Format in Markdown with clear headers and bullet points.
@@ -448,6 +450,7 @@ def chat_with_diet_bot(message, history, session_id=None):
                 formatted_response = format_markdown(cohere_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."
@@ -477,6 +480,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}
         Provide helpful, evidence-based advice with practical tips.
         """