Update app.py

app.py

@@ -2,20 +2,15 @@ import streamlit as st
 from PIL import Image
 import google.generativeai as genai
 import os
-import json
 
-# Configure API key
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Page config
 st.set_page_config(page_title="Smart Nutrition Analyzer", page_icon="🥗", layout="centered")
 
-# Title & description
 st.title("🥗 Smart Nutrition Analyzer")
 st.caption("AI-Powered Food Recognition & Nutritional Breakdown")
 st.write("Upload a food photo and receive a structured, professional nutritional analysis.")
 
-# File uploader
 uploaded_file = st.file_uploader("📸 Upload your meal photo", type=["png", "jpg", "jpeg"])
 
 if uploaded_file:
@@ -25,31 +20,21 @@ if uploaded_file:
     if st.button("Analyze Nutrition"):
         with st.spinner("Analyzing..."):
             prompt = """
-            You are a certified nutritionist. Analyze the uploaded food image and return
-            a concise nutritional report in JSON format with the following keys:
-
-            {
-              "FoodItems": [
-                {"name": "Item 1", "quantity": "e.g., 2 pieces"},
-                {"name": "Item 2", "quantity": "e.g., 1 cup"}
-              ],
-              "Calories": "Number (kcal)",
-              "Protein": "Number (g)",
-              "Carbohydrates": "Number (g)",
-              "Fats": "Number (g)",
-              "Fiber": "Number (g)",
-              "Sugar": "Number (g)",
-              "Sodium": "Number (mg)"
-            }
-
-            Guidelines:
-            - Only return data in this structure, no extra text.
-            - Be realistic with portion sizes.
-            - If unsure, provide the closest reasonable estimate.
+            You are a certified nutritionist. Analyze the uploaded food image and provide a professional nutritional breakdown.
+            Format it like this:
+
+            List of items: [Item 1] (quantity), [Item 2] (quantity)
+            Calories: [Number] kcal
+            Protein: [Number] g
+            Carbohydrates: [Number] g
+            Fats: [Number] g
+            Fiber: [Number] g
+            Sugar: [Number] g
+            Sodium: [Number] mg
+
+            Only return the information in this format, no extra text.
             """
-
             try:
-                # Generate response
                 model = genai.GenerativeModel("gemini-2.5-flash")
                 response = model.generate_content([prompt, image])
                 text = response.text.strip()
@@ -57,26 +42,29 @@ if uploaded_file:
                 if "This is not a food item." in text:
                     st.error(text)
                 else:
-                    # Parse JSON response
-                    data = json.loads(text)
-
-                    # Meal Items Section
+                    lines = [line for line in text.split("\n") if line.strip()]
+       
                     st.subheader("🍽️ Meal Items")
-                    for item in data["FoodItems"]:
-                        st.write(f"- {item['name']} ({item['quantity']})")
-
-                    # Nutritional Summary Section
+                    st.write(lines[0].replace("List of items:", "").strip())
+          
                     st.subheader("⚖️ Nutritional Summary")
-                    col1, col2, col3, col4 = st.columns(4)
-                    col1.metric("Calories", data["Calories"])
-                    col2.metric("Protein", data["Protein"])
-                    col3.metric("Carbs", data["Carbohydrates"])
-                    col4.metric("Fats", data["Fats"])
-
-                    col1, col2, col3 = st.columns(3)
-                    col1.metric("Fiber", data["Fiber"])
-                    col2.metric("Sugar", data["Sugar"])
-                    col3.metric("Sodium", data["Sodium"])
+                    nutrients = {}
+                    for line in lines[1:]:
+                        if ":" in line:
+                            key, value = line.split(":", 1)
+                            nutrients[key.strip()] = value.strip()
+          
+                    if nutrients:
+                        col1, col2, col3, col4 = st.columns(4)
+                        col1.metric("Calories", nutrients.get("Calories", "N/A"))
+                        col2.metric("Protein", nutrients.get("Protein", "N/A"))
+                        col3.metric("Carbs", nutrients.get("Carbohydrates", "N/A"))
+                        col4.metric("Fats", nutrients.get("Fats", "N/A"))
+
+                        col1, col2, col3 = st.columns(3)
+                        col1.metric("Fiber", nutrients.get("Fiber", "N/A"))
+                        col2.metric("Sugar", nutrients.get("Sugar", "N/A"))
+                        col3.metric("Sodium", nutrients.get("Sodium", "N/A"))
 
                     st.success("Nutrition Analysis Complete!")