Create main.py

main.py

@@ -0,0 +1,68 @@
+from fastapi import FastAPI, UploadFile, File
+from fastapi.middleware.cors import CORSMiddleware
+from ultralytics import YOLO
+from PIL import Image
+import io, requests, os
+
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+MODEL_URL = "https://huggingface.co/Rahaf2001/sabiq-road-detection/resolve/main/best.pt"
+
+def download_model():
+    if not os.path.exists("best.pt"):
+        print("Downloading model...")
+        r = requests.get(MODEL_URL)
+        with open("best.pt", "wb") as f:
+            f.write(r.content)
+        print(" Model ready")
+
+download_model()
+model = YOLO("best.pt")
+
+CLASS_NAMES = {0:'crack', 1:'pothole', 2:'other'}
+
+def get_severity(conf, area):
+    if conf > 0.85 and area > 0.05: return 'high'
+    elif conf > 0.65: return 'medium'
+    else: return 'low'
+
+@app.get("/")
+def root():
+    return {"status": "SABIQ API running "}
+
+@app.post("/detect")
+async def detect(file: UploadFile = File(...)):
+    img     = Image.open(io.BytesIO(await file.read()))
+    results = model(img)
+    detections = []
+    for box in results[0].boxes:
+        cls      = int(box.cls)
+        conf     = float(box.conf)
+        xywhn    = box.xywhn[0].tolist()
+        area     = xywhn[2] * xywhn[3]
+        detections.append({
+            "damage_type": CLASS_NAMES.get(cls, 'other'),
+            "confidence" : round(conf, 3),
+            "severity"   : get_severity(conf, area),
+            "bbox"       : xywhn
+        })
+    return {"total": len(detections), "detections": detections}
+```
+
+---
+
+**`requirements.txt`**
+```
+fastapi
+uvicorn
+ultralytics
+pillow
+requests
+python-multipart