Update models/detection/detector.py

models/detection/detector.py

@@ -70,7 +70,7 @@ class ObjectDetector:
         overlay = image.copy()
         draw = ImageDraw.Draw(overlay)
     
-        # Use a font (try arial, fallback to default)
+        # Load font
         try:
             font = ImageFont.truetype("arial.ttf", 16)
         except:
@@ -87,11 +87,16 @@ class ObjectDetector:
                     outline="red"
                 )
     
-            # 🧠 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]]
+            # Calculate text size using textbbox
+            text_bbox = draw.textbbox((bbox[0], bbox[1]), label, font=font)
+            text_width = text_bbox[2] - text_bbox[0]
+            text_height = text_bbox[3] - text_bbox[1]
     
+            # Define background rectangle for text
+            text_bg = [bbox[0], bbox[1] - text_height, bbox[0] + text_width + 4, bbox[1]]
             draw.rectangle(text_bg, fill="red")
+    
+            # Draw text
             draw.text((bbox[0] + 2, bbox[1] - text_height), label, fill="white", font=font)
     
         return Image.blend(image, overlay, alpha)