Update health_logic.py

health_logic.py

@@ -1,44 +1,56 @@
+def generate_advice(food_info, conditions):
+    ingredients = [i.lower() for i in food_info["ingredients"]]
+    fat = food_info["fat"]
+    calories = food_info["calories"]
+    carbs = food_info["carbs"]
+    protein = food_info["protein"]
 
-def generate_advice(food_data, conditions):
-    ingredients = [i.lower() for i in food_data.get("ingredients", [])]
     risk_score = 0
-    advice_list = []
     flags = []
+    advice_parts = []
 
-    for condition in conditions:
-        if condition == "Diabetic":
-            if "sugar" in ingredients or food_data["carbs"] > 40:
-                risk_score += 25
-                flags.append("High carbs/sugar")
-                advice_list.append("Reduce starchy content, avoid sugary drinks.")
-        if condition == "Hypertensive":
-            if "salt" in ingredients or food_data["fat"] > 25:
+    for cond in conditions:
+        if cond == "Diabetic":
+            if "sugar" in ingredients or carbs > 40:
+                risk_score += 30
+                flags.append("High carb")
+                advice_parts.append("Reduce starchy ingredients or use low-GI substitutes.")
+
+        if cond == "Hypertensive":
+            if "salt" in ingredients:
                 risk_score += 20
-                flags.append("High salt/fat")
-                advice_list.append("Use less salt and oil, consider boiling or steaming.")
-        if condition == "Weight Loss":
-            if food_data["calories"] > 350:
+                flags.append("Salt content")
+                advice_parts.append("Avoid adding salt; use natural spices.")
+            if fat > 25:
                 risk_score += 15
-                flags.append("High calories")
-                advice_list.append("Reduce portion size, avoid fried foods.")
-        if condition == "Malnourished":
-            if food_data["protein"] < 10:
-                risk_score += 10
-                flags.append("Low protein")
-                advice_list.append("Add beans, eggs, or fish to meals for better nutrition.")
-        if condition == "Pregnant/Nursing":
-            if food_data["iron"] < 10:
+                flags.append("High fat")
+
+        if cond == "Weight Loss":
+            if calories > 400:
+                risk_score += 25
+                flags.append("High calorie")
+                advice_parts.append("Consider smaller portions or reduce oil usage.")
+
+        if cond == "Malnourished":
+            if protein < 10:
                 risk_score += 15
-                flags.append("Low iron")
-                advice_list.append("Add leafy greens, beans, or iron supplements.")
-        if condition == "Cholesterol Watch":
-            if "palm oil" in ingredients or food_data["fat"] > 25:
+                flags.append("Low protein")
+                advice_parts.append("Add protein-rich sides like beans or eggs.")
+
+        if cond == "Pregnant/Nursing":
+            if protein < 10:
+                risk_score += 10
+                flags.append("Low protein for pregnancy")
+                advice_parts.append("Increase protein with fish, legumes, or eggs.")
+
+        if cond == "Cholesterol Watch":
+            if "palm oil" in ingredients or fat > 25:
                 risk_score += 20
-                flags.append("Saturated fat")
-                advice_list.append("Reduce palm oil and fried food intake.")
-        if condition == "Normal":
-            advice_list.append("Maintain a balanced diet with local ingredients.")
-
-    risk_score = min(risk_score, 100)
-    final_advice = " ".join(set(advice_list))
-    return final_advice, risk_score, flags
+                flags.append("Cholesterol risk")
+                advice_parts.append("Avoid palm oil and fried foods.")
+
+    if not advice_parts:
+        advice_parts.append("No major risks detected. Enjoy in moderation.")
+
+    advice = " ".join(set(advice_parts))
+    return advice, min(risk_score, 100), flags