Update app.py

app.py

@@ -24,13 +24,12 @@ if uploaded_video:
     video_path = tfile.name
 
     st.video(video_path)
-
     st.markdown("### ⏳ Processing video...")
 
     cap = cv2.VideoCapture(video_path)
-    width  = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
-    fps    = int(cap.get(cv2.CAP_PROP_FPS))
+    fps = int(cap.get(cv2.CAP_PROP_FPS))
 
     output_path = os.path.join(tempfile.gettempdir(), "output_segmentation.mp4")
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
@@ -56,33 +55,12 @@ 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
+                # Resize and apply mask
                 resized_mask = cv2.resize(mask, (frame.shape[1], frame.shape[0]))
                 mask_bool = resized_mask > 0.5
 
-                # Overlay color
-                color = (0, 255, 0)  # green
+                # Overlay color (green for now)
+                color = (0, 255, 0)
                 colored_mask = np.zeros_like(frame, dtype=np.uint8)
                 colored_mask[mask_bool] = color
                 frame = cv2.addWeighted(frame, 1.0, colored_mask, 0.5, 0)
@@ -91,20 +69,21 @@ if uploaded_video:
                 x1, y1, x2, y2 = box.astype(int)
                 label_text = label
                 font = cv2.FONT_HERSHEY_SIMPLEX
-                font_scale = 0.6
+                font_scale = 0.7
                 thickness = 2
 
-                # Get text size to create background rectangle
+                # Get text size for background box
                 (text_width, text_height), _ = cv2.getTextSize(label_text, font, font_scale, thickness)
-                text_x, text_y = x1, y1 - 10 if y1 - 10 > 10 else y1 + 20
+                text_x = x1
+                text_y = y1 - 10 if y1 - 10 > 10 else y1 + text_height + 10
 
-                # Background rectangle
-                cv2.rectangle(frame, (text_x, text_y - text_height - 4),
-                              (text_x + text_width + 4, text_y + 4), color, -1)
+                # Draw background rectangle
+                cv2.rectangle(frame, (text_x - 2, text_y - text_height - 4),
+                              (text_x + text_width + 2, text_y + 4), (0, 0, 0), -1)
 
-                # Text
-                cv2.putText(frame, label_text, (text_x + 2, text_y),
-                            font, font_scale, (0, 0, 0), thickness=1, lineType=cv2.LINE_AA)
+                # Put text on top of rectangle
+                cv2.putText(frame, label_text, (text_x, text_y),
+                            font, font_scale, (255, 255, 255), thickness=2, lineType=cv2.LINE_AA)
 
         out.write(frame)
         frame_index += 1