Update app.py

app.py

@@ -1,36 +1,59 @@
 import torch
 import requests
 import os
-from PIL import Image
+import pandas as pd
+from PIL import Image, ImageDraw, ImageFont
 import gradio as gr
 
-# اسم الملف المحلي للموديل
+# روابط وتحميل الموديل
 model_path = "best.pt"
-
-# رابط التحميل من الريبو بتاعك على Hugging Face
 model_url = "https://huggingface.co/AlaaElsayed/yolospace/resolve/main/best.pt"
-
-# حمل الموديل لو مش موجود
 if not os.path.exists(model_path):
     r = requests.get(model_url)
     with open(model_path, "wb") as f:
         f.write(r.content)
 
-# تحميل موديل YOLO
-model = torch.hub.load('ultralytics/yolov8', 'custom', path=model_path, source='local')
+# تحميل الموديل
+model = torch.hub.load('ultralytics/yolov5', 'custom', path=model_path, source='local')
+
+# تحميل معلومات الطعام
+food_info = pd.read_csv("Food.csv")
+
+# دالة لإحضار البيانات الغذائية
+def get_nutrition(label):
+    row = food_info[food_info["Food_Name"].str.lower() == label.lower()]
+    if row.empty:
+        return "No info", "?", "?", "?"
+    cals = row["Calories_per_100g"].values[0]
+    fat = row["Fat_g"].values[0]
+    protein = row["Protein_g"].values[0]
+    carbs = row["Carbs_g"].values[0]
+    return f"{cals} kcal", f"{fat}g fat", f"{protein}g protein", f"{carbs}g carbs"
 
 # دالة الكشف
 def detect(image):
     results = model(image)
-    return Image.fromarray(results.render()[0])
+    labels = results.names
+    df = results.pandas().xyxy[0]
+
+    img = Image.fromarray(results.render()[0])
+    draw = ImageDraw.Draw(img)
+
+    for _, row in df.iterrows():
+        label = row["name"]
+        cal, fat, pro, carb = get_nutrition(label)
+        text = f"{label}: {cal}, {fat}, {pro}, {carb}"
+        draw.text((row["xmin"], row["ymin"] - 10), text, fill=(255, 0, 0))
+
+    return img
 
-# واجهة Gradio
+# Gradio app
 demo = gr.Interface(
     fn=detect,
     inputs=gr.Image(type="pil"),
     outputs=gr.Image(type="pil"),
-    title="YOLO Food Detector",
-    description="Upload a food image and yolo will detect the items!"
+    title="YOLOv5 Food Detector + Nutrition Info",
+    description="Upload an image of food and see calories and nutrients!"
 )
 
 demo.launch()