Update app.py

app.py

@@ -209,29 +209,22 @@ def _run_inference(image: Image.Image, max_new_tokens: int) -> str:
 
 
 def _parse_response(raw: str) -> dict:
-    """
-    Parse the CaLoRAify reasoning loop output:
-    "Ingredients detected: X. Portion Analysis: Y. JSON Summary: {...}"
-    """
     result = {"ingredients": "", "portion_notes": "", "raw_text": raw}
 
-    # Extract ingredients
+    # Try the CaLoRAify reasoning loop format first
     if "Ingredients detected:" in raw:
         ing_start = raw.index("Ingredients detected:") + len("Ingredients detected:")
         ing_end   = raw.index(".", ing_start) if "." in raw[ing_start:] else len(raw)
         result["ingredients"] = raw[ing_start:ing_end].strip()
 
-    # Extract portion notes
     if "Portion Analysis:" in raw:
         pa_start = raw.index("Portion Analysis:") + len("Portion Analysis:")
         pa_end   = raw.index(".", pa_start) if "." in raw[pa_start:] else len(raw)
         result["portion_notes"] = raw[pa_start:pa_end].strip()
 
-    # Extract JSON
     if "JSON Summary:" in raw:
         json_start = raw.index("JSON Summary:") + len("JSON Summary:")
         json_str   = raw[json_start:].strip()
-        # Find the JSON object boundaries
         brace_start = json_str.find("{")
         brace_end   = json_str.rfind("}") + 1
         if brace_start != -1 and brace_end > brace_start:
@@ -245,7 +238,11 @@ def _parse_response(raw: str) -> dict:
                     "fibre_g":   nutrition.get("fibre_g"),
                 })
             except json.JSONDecodeError:
-                pass  # raw_text still available for debugging
+                pass
+
+    # ── FALLBACK: if model didn't follow format, use raw text as ingredients
+    if not result["ingredients"] and raw.strip():
+        result["ingredients"] = raw.strip()[:300]   # show raw output so you can see what model said
 
     return result
 
@@ -273,3 +270,9 @@ def analyze(req: AnalyzeRequest, _=Depends(verify_api_key)):
 
     parsed = _parse_response(raw)
     return NutritionResponse(**parsed, latency_ms=ms)
+
+@app.post("/debug")
+def debug(req: AnalyzeRequest, _=Depends(verify_api_key)):
+    image = _decode_image(req.image_b64)
+    raw   = _run_inference(image, req.max_new_tokens)
+    return {"raw_text": raw}