Update app.py

app.py

@@ -6,11 +6,11 @@ import cv2
 import base64
 from ultralytics import YOLO
 
-# ─── Load Models 
+# ─── Load Models ──────────────────────────────────────────────────────────────
 person_model = YOLO("yolov8n.pt")
 statue_model = YOLO("best.pt")
 
-# ─── App Setup 
+# ─── App Setup ────────────────────────────────────────────────────────────────
 app = FastAPI()
 
 UPLOAD_FOLDER = "/tmp/uploads"
@@ -18,26 +18,26 @@ os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
 app.mount("/uploads", StaticFiles(directory=UPLOAD_FOLDER), name="uploads")
 
-CONF_THRESHOLD = 0.30
+CONF_THRESHOLD = 0.25
 
 router = APIRouter()
 
 
 def draw_label(image, label, x1, y1, x2, y2, box_color, text_color):
-    
+    """ارسم الـ label فوق الـ box، ولو مفيش مكان فوق حطه تحت"""
     font = cv2.FONT_HERSHEY_SIMPLEX
     font_scale = 0.6
     thickness = 2
 
     (tw, th), _ = cv2.getTextSize(label, font, font_scale, thickness)
 
-    
+    # لو فيه مكان فوق الـ box
     if y1 - th - 8 >= 0:
         label_y1 = y1 - th - 8
         label_y2 = y1
         text_y = y1 - 5
     else:
-        
+        # حط الـ label جوه الـ box من فوق
         label_y1 = y1
         label_y2 = y1 + th + 8
         text_y = y1 + th + 3
@@ -71,7 +71,7 @@ async def predict_image(
     person_count = 0
     statue_count = 0
 
-    # ── Person Detection 
+    # ── Person Detection ──────────────────────────────────────────────────────
     person_results = person_model(file_path)
 
     for box in person_results[0].boxes:
@@ -97,7 +97,7 @@ async def predict_image(
         })
         person_count += 1
 
-    # ── Statue Detection 
+    # ── Statue Detection ──────────────────────────────────────────────────────
     statue_results = statue_model(file_path)
 
     for box in statue_results[0].boxes:
@@ -122,7 +122,7 @@ async def predict_image(
         })
         statue_count += 1
 
-    # ── Save Output Image 
+    # ── Save Output Image ─────────────────────────────────────────────────────
     output_filename = f"output_{safe_filename}"
     output_path = os.path.join(UPLOAD_FOLDER, output_filename)
     cv2.imwrite(output_path, image)