Update app.py

app.py

@@ -12,15 +12,15 @@ DOBBY_API_KEY = "fw_3ZjtsywUGddwa1wGY4VvB3eW"
 DOBBY_LEASHED_MODEL = "accounts/sentientfoundation/models/dobby-mini-leashed-llama-3-1-8b#accounts/sentientfoundation/deployments/22e7b3fd"
 DOBBY_UNHINGED_MODEL = "accounts/sentientfoundation/models/dobby-mini-unhinged-llama-3-1-8b#accounts/sentientfoundation/deployments/81e155fc"
 
-# Initialize session state for tracking meals and user info
+# Initialize session state for tracking meals, workouts, and user info
 if "meal_data" not in st.session_state:
     st.session_state.meal_data = {"Breakfast": {}, "Lunch": {}, "Dinner": {}}
 if "daily_totals" not in st.session_state:
-    st.session_state.daily_totals = {"Calories": 0, "Protein": 0, "Carbs": 0, "Fat": 0}
+    st.session_state.daily_totals = {"Calories": 0, "Protein": 0, "Carbs": 0, "Fat": 0, "Workout Calories Burnt": 0}
 if "user_info" not in st.session_state:
     st.session_state.user_info = {}
 
-
+# Custom CSS for modern look
 st.markdown("""
     <style>
         .stApp {
@@ -48,18 +48,9 @@ st.markdown("""
             color: #FFFFFF; /* White text for headers */
             margin-bottom: 10px;
         }
-        .button {
-            background-color: #4CAF50; /* Green button */
-            color: white;
-            font-size: 16px;
-            padding: 10px 20px;
-            border-radius: 5px;
-        }
     </style>
 """, unsafe_allow_html=True)
 
-
-
 st.markdown('<div class="title">Calorie Bro</div>', unsafe_allow_html=True)
 st.markdown('<div class="subtitle">Your AI image calorie tracker, powered by Dobby. Dare to get feedback!</div>', unsafe_allow_html=True)
 
@@ -92,13 +83,8 @@ if st.session_state.user_info:
         bytes_data = uploaded_file.getvalue()
         return [{"mime_type": uploaded_file.type, "data": bytes_data}]
 
-    def get_dobby_comment(food_name, total_calories, total_protein, total_carbs, total_fat, meal, goal, tone):
+    def get_dobby_comment(prompt):
         selected_model = DOBBY_UNHINGED_MODEL  # Default tone is sarcastic
-        dobby_prompt = f"""
-        The user had {meal}: {food_name}.
-        Nutritional data: {total_calories} kcal, {total_protein} g protein, {total_carbs} g carbs, {total_fat} g fat.
-        The user's goal is to {goal.lower()}. Provide feedback for this meal.
-        """
         response = requests.post(
             DOBBY_API_URL,
             headers={
@@ -107,7 +93,7 @@ if st.session_state.user_info:
             },
             json={
                 "model": selected_model,
-                "messages": [{"role": "user", "content": dobby_prompt}],
+                "messages": [{"role": "user", "content": prompt}],
             },
         )
         return response.json().get("choices", [{}])[0].get("message", {}).get("content", "No response")
@@ -155,43 +141,58 @@ if st.session_state.user_info:
                 st.session_state.daily_totals["Carbs"] += total_carbs
                 st.session_state.daily_totals["Fat"] += total_fat
 
-                # Get Dobby's feedback
-                dobby_comment = get_dobby_comment(
-                    food_name, total_calories, total_protein, total_carbs, total_fat, meal,
-                    st.session_state.user_info["Goal"], st.session_state.user_info["Tone"]
-                )
-
                 # Display results in the desired format
                 st.write(
                     f"**{food_name}: {total_calories} kcal, {total_protein}g of Protein, {total_carbs}g of Carbs, {total_fat}g of Fat**"
                 )
-                st.write(dobby_comment)
 
             except Exception as e:
                 st.error(f"Error processing the image: {e}")
 
-    # Step 3: Daily Summary
-    if all(st.session_state.meal_data[meal] for meal in ["Breakfast", "Lunch", "Dinner"]):
+# Step 3: Workout tracking
+if all(st.session_state.meal_data[meal] for meal in ["Breakfast", "Lunch", "Dinner"]):
+    st.markdown('<div class="header">Step 3: Log Your Workout</div>', unsafe_allow_html=True)
+    workout_input = st.text_area("Describe your workout for the day:", help="E.g., 'I walked for 30 minutes and biked for 1 hour.'")
+    if st.button("Analyze Workout"):
+        with st.spinner("Calculating calories burnt..."):
+            workout_prompt = f"""
+            The user provided this workout description: "{workout_input}".
+            Estimate the calories burned and summarize the activities. Provide the output in this format:
+            Calories Burned: X kcal
+            Activities: [List of activities]
+            """
+            try:
+                workout_analysis = analyze_food_with_gemini(workout_prompt, [""])
+                lines = workout_analysis.split("\n")
+                calories_burned = int(lines[0].replace("Calories Burned: ", "").replace("kcal", "").strip())
+                activities = lines[1].replace("Activities: ", "").strip()
+
+                # Update daily totals
+                st.session_state.daily_totals["Workout Calories Burnt"] = calories_burned
+
+                # Dobby's comment on workout
+                dobby_workout_comment = get_dobby_comment(f"The user did this workout: {workout_input}. Provide feedback.")
+                st.write(f"**Calories Burned:** {calories_burned} kcal")
+                st.write(f"**Activities:** {activities}")
+                st.write(f"**Dobby's Comment:** {dobby_workout_comment}")
+            except Exception as e:
+                st.error(f"Error processing workout: {e}")
+
+    # Step 4: Daily summary
+    if st.session_state.daily_totals["Workout Calories Burnt"]:
         st.subheader("Daily Summary")
-        st.write(f"**Total Calories:** {st.session_state.daily_totals['Calories']} kcal")
+        st.write(f"**Total Calories Consumed:** {st.session_state.daily_totals['Calories']} kcal")
         st.write(f"**Total Protein:** {st.session_state.daily_totals['Protein']} g")
         st.write(f"**Total Carbs:** {st.session_state.daily_totals['Carbs']} g")
         st.write(f"**Total Fat:** {st.session_state.daily_totals['Fat']} g")
+        st.write(f"**Calories Burned from Workout:** {st.session_state.daily_totals['Workout Calories Burnt']} kcal")
 
         # Final Dobby comment
-        final_prompt = f"""
-        The user had the following meals today:
-        Breakfast: {st.session_state.meal_data['Breakfast']['Food Name']}
-        Lunch: {st.session_state.meal_data['Lunch']['Food Name']}
-        Dinner: {st.session_state.meal_data['Dinner']['Food Name']}
-        Total daily intake: {st.session_state.daily_totals['Calories']} kcal, 
-        {st.session_state.daily_totals['Protein']} g protein, {st.session_state.daily_totals['Carbs']} g carbs, 
-        {st.session_state.daily_totals['Fat']} g fat.
-        The user's goal is to {st.session_state.user_info['Goal'].lower()}. Provide a summary and feedback for the day.
+        total_calorie_balance = st.session_state.daily_totals['Calories'] - st.session_state.daily_totals['Workout Calories Burnt']
+        final_summary_prompt = f"""
+        The user consumed {st.session_state.daily_totals['Calories']} kcal and burned {st.session_state.daily_totals['Workout Calories Burnt']} kcal through workouts.
+        Total calorie balance: {total_calorie_balance} kcal.
+        Provide a summary and feedback for the day.
         """
-        final_comment = get_dobby_comment(
-            "", st.session_state.daily_totals["Calories"], st.session_state.daily_totals["Protein"],
-            st.session_state.daily_totals["Carbs"], st.session_state.daily_totals["Fat"], "Daily Summary",
-            st.session_state.user_info["Goal"], st.session_state.user_info["Tone"]
-        )
-        st.write(final_comment)
+        final_comment = get_dobby_comment(final_summary_prompt)
+        st.write(f"**Dobby's Daily Summary:** {final_comment}")