Update app.py

app.py

@@ -56,6 +56,27 @@ if uploaded_video:
             for mask, class_id, box in zip(masks, class_ids, boxes):
                 label = names[class_id]
 
+                # Resize mask to match original frame
+                resized_mask = cv2.resize(mask, (frame.shape[1], frame.shape[0]))
+                mask_bool = resized_mask > 0.5
+
+                # Create colored overlay
+                colored_mask = np.zeros_like(frame, dtype=np.uint8)
+                colored_mask[mask_bool] = [0, 255, 0]  # green
+                frame = cv2.addWeighted(frame, 1.0, colored_mask, 0.5, 0)
+
+                # Get contours to find where to place the label
+                contours, _ = cv2.findContours(mask_bool.astype(np.uint8), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+                    if contours:
+                        x, y, w, h = cv2.boundingRect(contours[0])
+
+                # Draw background rectangle for better visibility
+                        cv2.rectangle(frame, (x, y - 25), (x + len(label) * 12, y), (0, 0, 0), -1)
+
+                # Put label text
+                        cv2.putText(frame, label, (x, y - 5), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (255, 255, 255), 2, cv2.LINE_AA)
+
+
                 # Resize mask
                 resized_mask = cv2.resize(mask, (frame.shape[1], frame.shape[0]))
                 mask_bool = resized_mask > 0.5