Update Space

app.py

@@ -53,14 +53,6 @@ def draw_boxes(image: Image.Image, outputs: dict):
 
     # Dynamic font size based on image dimensions
     image_width, image_height = image.size
-    font_size = max(20, image_width // 50)  # Increased minimum font size
-    try:
-        # Attempt to load a truetype font; adjust the path if necessary
-        font = ImageFont.truetype("arial.ttf", size=font_size)
-    except IOError:
-        # Fallback to default font if truetype font is not found
-        font = ImageFont.load_default()
-
     boxes = outputs.get('detected_boxes', [])
     classes = outputs.get('detected_classes', [])
     scores = outputs.get('detected_scores', [])
@@ -88,12 +80,8 @@ def draw_boxes(image: Image.Image, outputs: dict):
         #And width of the entire bounding box
 
         text_width = right - left
-        text_height = (bottom - top) // 10
+        text_height = (bottom - top) // 20 # 5% of the bounding box height
 
-        # Calculate text size using textbbox
-        #text_bbox = draw.textbbox((0, 0), text, font=font)
-        #text_width = text_bbox[2] - text_bbox[0]
-        #text_height = text_bbox[3] - text_bbox[1]
 
         # Calculate label background position
         # Ensure the label box does not go above the image
@@ -106,11 +94,18 @@ def draw_boxes(image: Image.Image, outputs: dict):
             fill=(255, 0, 0, 160)  # Semi-transparent red
         )
 
+        #Font Size should fill the text box we just drew vertically
+        font_size = 1
+        font = ImageFont.load_default()
+        while font.getsize(text)[1] < text_height:
+            font_size += 1
+            font = ImageFont.load_default().font_variant(size=font_size)
+
         # Draw text
         draw.text(
             (label_left + 5, label_top + 5),
             text,
-            fill="white",
+            fill="black",
             font=font
         )