Upload 4 files

Requirements.txt

@@ -0,0 +1,5 @@
+fastapi
+uvicorn
+python-multipart
+ultralytics
+opencv-python-headless

app.py

@@ -0,0 +1,59 @@
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import JSONResponse
+import cv2
+import numpy as np
+from ultralytics import YOLO
+
+app = FastAPI(title="GiziTrace ML API")
+
+model = YOLO("best.onnx", task="detect")
+
+@app.post("/predict")
+async def predict_waste(file: UploadFile = File(...)):
+    try:
+
+        contents = await file.read()
+        nparr = np.frombuffer(contents, np.uint8)
+        img = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
+
+        results = model(img, conf=0.10)
+        
+        area_plate = 0
+        area_food = 0
+
+        for result in results:
+            if len(result.boxes) == 0:
+                continue
+                
+            boxes = result.boxes.xyxy.cpu().numpy()
+            clss = result.boxes.cls.cpu().numpy()
+            
+            for idx, box in enumerate(boxes):
+                x1, y1, x2, y2 = box
+                box_area = (x2 - x1) * (y2 - y1)
+                class_id = int(clss[idx])
+   
+                if class_id == 0:
+                    area_plate += box_area
+                elif class_id == 1:
+                    area_food += box_area
+
+        if area_plate == 0:
+            return JSONResponse(content={
+                "status": "manual_review",
+                "message": "Piring tidak terdeteksi, butuh input manual",
+                "waste_percent": 0.0,
+                "consumption_percent": 0.0
+            })
+
+        waste_percent = min(round((area_food / area_plate) * 100, 2), 100.0)
+        consumption_percent = round(100 - waste_percent, 2)
+        
+        return JSONResponse(content={
+            "status": "success",
+            "waste_percent": waste_percent,
+            "consumption_percent": consumption_percent
+        })
+        
+    except Exception as e:
+        return JSONResponse(status_code=500, content={"message": str(e)})

best.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1a486f3759075590ddbfecf99e788bec894c068b4d5719f801f2588b34b7938c
+size 12265978