Update models/detection/detector.py

models/detection/detector.py

@@ -70,9 +70,9 @@ class ObjectDetector:
         overlay = image.copy()
         draw = ImageDraw.Draw(overlay)
     
-        # Use a basic font (optional: can use truetype for better styling)
+        # Use a font (try arial, fallback to default)
         try:
-            font = ImageFont.truetype("arial.ttf", 16)  # Try a common system font
+            font = ImageFont.truetype("arial.ttf", 16)
         except:
             font = ImageFont.load_default()
     
@@ -80,17 +80,18 @@ class ObjectDetector:
             bbox = det["bbox"]
             label = f'{det["class_name"]} {det["confidence"]:.2f}'
     
-            # Thicker rectangle (3–4 px)
-            for offset in range(3):  # creates a thicker border
+            # Draw thicker bounding box
+            for offset in range(3):
                 draw.rectangle(
-                    [bbox[0]-offset, bbox[1]-offset, bbox[2]+offset, bbox[3]+offset],
+                    [bbox[0] - offset, bbox[1] - offset, bbox[2] + offset, bbox[3] + offset],
                     outline="red"
                 )
     
-            # Draw label with background box
-            text_size = draw.textsize(label, font=font)
-            text_bg = [bbox[0], bbox[1] - text_size[1], bbox[0] + text_size[0] + 4, bbox[1]]
+            # 🧠 Use font.getsize instead of draw.textsize
+            text_width, text_height = font.getsize(label)
+            text_bg = [bbox[0], bbox[1] - text_height, bbox[0] + text_width + 4, bbox[1]]
+    
             draw.rectangle(text_bg, fill="red")
-            draw.text((bbox[0] + 2, bbox[1] - text_size[1]), label, fill="white", font=font)
+            draw.text((bbox[0] + 2, bbox[1] - text_height), label, fill="white", font=font)
     
         return Image.blend(image, overlay, alpha)