Update app.py

app.py

@@ -211,12 +211,38 @@ def _run_inference(image: Image.Image, max_new_tokens: int) -> str:
 
 
 
+
 def _get_nutrition_from_api(ingredients_text: str) -> dict:
-    """Use Open Food Facts search — no API key needed."""
     try:
-        # Take the first identified food item
-        food_query = ingredients_text.split(",")[0].strip()
+        # Extract the most important food word from the description
+        import re
+        # Common food words to search for
+        food_keywords = [
+            "banana", "apple", "orange", "mango", "rice", "chicken", "egg",
+            "bread", "milk", "yogurt", "cheese", "pizza", "burger", "salad",
+            "pasta", "fish", "beef", "pork", "potato", "tomato", "onion",
+            "carrot", "broccoli", "spinach", "lemon", "grape", "strawberry",
+            "chocolate", "cake", "cookie", "coffee", "tea", "juice", "soup",
+            "sandwich", "taco", "noodle", "tofu", "paneer", "dal", "roti",
+            "biryani", "curry", "samosa", "idli", "dosa"
+        ]
+        
+        text_lower = ingredients_text.lower()
+        food_query = None
+        
+        # Find first matching food keyword
+        for keyword in food_keywords:
+            if keyword in text_lower:
+                food_query = keyword
+                break
         
+        # If no keyword matched, use first 2 words of description
+        if not food_query:
+            words = re.findall(r'\b[a-zA-Z]{4,}\b', ingredients_text)
+            food_query = words[0] if words else ingredients_text[:20]
+
+        logger.info(f"Searching nutrition for: {food_query}")
+
         response = req_lib.get(
             "https://world.openfoodfacts.org/cgi/search.pl",
             params={
@@ -224,7 +250,8 @@ def _get_nutrition_from_api(ingredients_text: str) -> dict:
                 "search_simple": 1,
                 "action": "process",
                 "json": 1,
-                "page_size": 1,
+                "page_size": 3,
+                "fields": "product_name,nutriments",
             },
             timeout=10,
         )
@@ -233,21 +260,30 @@ def _get_nutrition_from_api(ingredients_text: str) -> dict:
         products = data.get("products", [])
 
         if not products:
+            logger.warning(f"No products found for: {food_query}")
             return {}
 
-        nutriments = products[0].get("nutriments", {})
-        return {
-            "calories":  round(nutriments.get("energy-kcal_100g", 0) or 0, 1),
-            "protein_g": round(nutriments.get("proteins_100g",    0) or 0, 1),
-            "carbs_g":   round(nutriments.get("carbohydrates_100g", 0) or 0, 1),
-            "fat_g":     round(nutriments.get("fat_100g",          0) or 0, 1),
-            "fibre_g":   round(nutriments.get("fiber_100g",        0) or 0, 1),
-        }
+        # Find first product with complete nutrition data
+        for product in products:
+            nutriments = product.get("nutriments", {})
+            calories   = nutriments.get("energy-kcal_100g", 0)
+            if calories and calories > 0:
+                return {
+                    "calories":  round(float(calories), 1),
+                    "protein_g": round(float(nutriments.get("proteins_100g",        0) or 0), 1),
+                    "carbs_g":   round(float(nutriments.get("carbohydrates_100g",   0) or 0), 1),
+                    "fat_g":     round(float(nutriments.get("fat_100g",             0) or 0), 1),
+                    "fibre_g":   round(float(nutriments.get("fiber_100g",           0) or 0), 1),
+                }
+
+        return {}
+
     except Exception as e:
-        logger.warning(f"Open Food Facts API failed: {e}")
+        logger.warning(f"Nutrition API failed: {e}")
         return {}
 
 
+
 def _parse_response(raw: str) -> dict:
     # FIX: pre-initialize all nutrition keys to None so KeyError never happens
     result = {