Update app.py

app.py

@@ -11,29 +11,27 @@ def identify_dish(image_bytes):
     encoded_image = base64.b64encode(image_bytes).decode("utf-8")
     dish_name = ""
 
-    try:
-        for message in client.chat_completion(
-            model="meta-llama/Llama-3.2-11B-Vision-Instruct",
-            messages=[
-                {
-                    "role": "You are a food identification expert who identifies dishes from images. Your task is to strictly return the names of the dishes present in the image. Only return the dish names if you have high Confidence Level and without additional explanation or description.",
-                    "content": [
-                        {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{encoded_image}" }},
-                        {"type": "text", "text": "Identify the dishes in the image and return only the names of the dishes."},
-                    ],
-                }
-            ],
-            max_tokens=70,
-            stream=True,
-        ):
-            if message.choices[0].delta.content:
-                dish_name += message.choices[0].delta.content
-    except:
-        st.error("Error processing the image. Please upload a valid image.")
+    for message in client.chat_completion(
+        model="meta-llama/Llama-3.2-11B-Vision-Instruct",
+        messages=[
+            {
+                "role": "You are a food identification expert who identifies dishes from images. Your task is to strictly return the names of the dishes present in the image. Only return the dish names if you have high Confidence Level and without additional explanation or description.",
+                "content": [
+                    {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{encoded_image}" }},
+                    {"type": "text", "text": "Identify the dishes in the image and return only the names of the dishes."},
+                ],
+            }
+        ],
+        max_tokens=70,
+        stream=True,
+    ):
+        if message.choices[0].delta.content:
+            dish_name += message.choices[0].delta.content
+
     return dish_name.strip()
 
-# 2. Function to calculate metrics
-def calculate_metrics(age, gender, height_cm, weight_kg, weight_goal, activity_level):
+# 2. Function to get user inputs and calculate daily caloric needs
+def calculate_metrics(age, gender, height_cm, weight_kg, weight_goal, activity_level, time_frame_months):
     bmi = weight_kg / ((height_cm / 100) ** 2)
 
     if gender == "male":
@@ -86,6 +84,7 @@ def generate_diet_plan(dish_name, calorie_intake_per_day, goal):
     - Dish Name: {dish_name}
     - Caloric Intake per Day: {calorie_intake_per_day} calories
     - Goal: {goal}
+
     """
     response = client.chat_completion(
         model="meta-llama/Meta-Llama-3-8B-Instruct",
@@ -95,7 +94,10 @@ def generate_diet_plan(dish_name, calorie_intake_per_day, goal):
 
     return response.choices[0].message.content
 
-# Streamlit Sidebar for user input
+# Streamlit App Title
+st.title("AI Diet Planner")
+
+# Sidebar for user input
 st.sidebar.title("User Input")
 image_file = st.sidebar.file_uploader("Upload an image of the dish", type=["jpeg", "png"])
 age = st.sidebar.number_input("Enter your age", min_value=1)
@@ -106,47 +108,49 @@ weight_goal = st.sidebar.selectbox("Weight goal", ["loss", "gain", "maintain"])
 activity_level = st.sidebar.selectbox("Activity level", ["sedentary", "light", "moderate", "active", "very active"])
 time_frame = st.sidebar.number_input("Time frame to achieve goal (months)", min_value=1)
 
-# Submit Button
-if st.sidebar.button("Submit"):
+# Submit button
+submit = st.sidebar.button("Submit")
+
+# Process the image and calculate metrics upon submission
+if submit:
     if image_file:
-        # Step 1: Identify the dish
+        st.write("### Results")
         image_bytes = image_file.read()
+
+        # Step 1: Identify the dish
         dish_name = identify_dish(image_bytes)
-        if dish_name:
-            st.write("### Results")
-            st.divider()
-            st.write("**Dish Name Identified:**")
-            st.success(dish_name)
-
-            # Step 2: Perform Calculations
-            metrics = calculate_metrics(age, gender, height_cm, weight_kg, weight_goal, activity_level)
-            st.divider()
-            st.write("**Calculated Metrics:**")
-            st.write(f"**BMI:** {metrics['BMI']:.2f}")
-            st.write(f"**BMR:** {metrics['BMR']:.2f} calories")
-            st.write(f"**TDEE:** {metrics['TDEE']:.2f} calories")
-            st.write(f"**Ideal Body Weight:** {metrics['IBW']:.2f} kg")
-            st.write(f"**Daily Caloric Needs:** {metrics['Daily Caloric Needs']:.2f} calories")
-
-            # Step 3: Generate Diet Plan
-            st.divider()
-            st.write("**Diet Plan Based on Dish & Goal:**")
-            diet_plan = generate_diet_plan(dish_name, metrics["Daily Caloric Needs"], weight_goal)
-            st.info(diet_plan)
-        else:
-            st.error("Unable to identify the dish. Please try again with a different image.")
+        st.markdown("<hr>", unsafe_allow_html=True)
+        st.write("#### Dish Name Identified:")
+        st.markdown(f"<div style='background-color: #d4edda; color: #155724; padding: 10px; border-radius: 10px;'>{dish_name}</div>", unsafe_allow_html=True)
+
+        # Step 2: Perform Calculations
+        metrics = calculate_metrics(age, gender, height_cm, weight_kg, weight_goal, activity_level, time_frame)
+        st.markdown("<hr>", unsafe_allow_html=True)
+        st.write("#### Metrics Calculated:")
+        st.markdown(f"""
+            <div style='background-color: #f8d7da; color: #721c24; padding: 10px; border-radius: 10px;'>
+                <p><b>Your BMI:</b> {metrics['BMI']:.2f}</p>
+                <p><b>Your BMR:</b> {metrics['BMR']:.2f} calories</p>
+                <p><b>Your TDEE:</b> {metrics['TDEE']:.2f} calories</p>
+                <p><b>Ideal Body Weight (IBW):</b> {metrics['IBW']:.2f} kg</p>
+                <p><b>Daily Caloric Needs:</b> {metrics['Daily Caloric Needs']:.2f} calories</p>
+            </div>
+        """, unsafe_allow_html=True)
+
+        # Step 3: Generate diet plan
+        diet_plan = generate_diet_plan(dish_name, metrics["Daily Caloric Needs"], weight_goal)
+        st.markdown("<hr>", unsafe_allow_html=True)
+        st.write("#### Diet Plan Based on Dish & Goal:")
+        st.markdown(f"<div style='background-color: #d1ecf1; color: #0c5460; padding: 10px; border-radius: 10px;'>{diet_plan}</div>", unsafe_allow_html=True)
+
     else:
-        st.error("Please upload an image of the dish.")
+        st.error("Please upload a valid image in JPEG or PNG format.")
 
-# CSS for styling
+# CSS styling
 st.markdown("""
     <style>
     .stButton button { background-color: #4CAF50; color: white; }
-    .stAlert { background-color: #f9f9f9; border: 1px solid #ddd; padding: 10px; }
     .stContainer { border: 1px solid #ddd; padding: 20px; margin-bottom: 20px; }
-    .stSuccess { background-color: #e8f5e9; }
-    .stInfo { background-color: #e3f2fd; }
-    .stError { background-color: #ffebee; }
+    hr { border: 1px solid #e9ecef; margin: 20px 0; }
     </style>
 """, unsafe_allow_html=True)
-