Update app.py

app.py

@@ -26,7 +26,7 @@ BLACK = (0, 0, 0)
 
 # Load font for labeling
 try:
-    font = ImageFont.truetype("arial.ttf", 10)
+    font = ImageFont.truetype("arial.ttf", 14)
 except:
     font = ImageFont.load_default()
 
@@ -76,8 +76,19 @@ def process_image(image_path):
                 total_counts[class_name] += 1
                 x_min, y_min = x1 + obj["x"] - obj["width"] // 2, y1 + obj["y"] - obj["height"] // 2
                 x_max, y_max = x1 + obj["x"] + obj["width"] // 2, y1 + obj["y"] + obj["height"] // 2
+                
+                # Draw bounding box
                 draw_final.rectangle([x_min, y_min, x_max, y_max], outline=GREEN, width=2)
-                draw_final.text((x_min, y_min - 10), class_name, fill=WHITE, font=font)
+                
+                # Draw extended label above the bounding box
+                text_size = draw_final.textbbox((0, 0), class_name, font=font)
+                text_width = text_size[2] - text_size[0]
+                text_height = text_size[3] - text_size[1]
+                text_x = x_min
+                text_y = y_min - text_height - 5 if y_min - text_height - 5 > 0 else y_max + 5
+                
+                draw_final.rectangle([text_x, text_y, text_x + text_width + 6, text_y + text_height + 2], fill=BLACK)
+                draw_final.text((text_x + 3, text_y), class_name, fill=WHITE, font=font)
     
     # Save the final processed image
     final_image_path = "processed_image.png"