Upload app.py

app.py

@@ -61,23 +61,26 @@ if not mirror_examples and os.path.exists("car.jpeg"):
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 CONF   = 0.45
 
-def apply_mask_overlay(img_rgb, mask_bool, color=(0, 215, 255), alpha=0.5):
-    # Dim the background slightly to make the segmented object pop out
-    background = cv2.addWeighted(img_rgb, 0.6, np.zeros_like(img_rgb), 0.4, 0)
+def apply_mask_overlay(img_rgb, mask_bool, color=(0, 215, 255), alpha=0.4):
+    # 1. Cinematic Background (Bokeh/Blur effect + darkened)
+    blurred_bg = cv2.GaussianBlur(img_rgb, (35, 35), 0)
+    background = cv2.addWeighted(blurred_bg, 0.6, np.zeros_like(img_rgb), 0.4, 0)
     
-    # Create the overlay with color
-    overlay = background.copy()
-    overlay[mask_bool] = color
-    blended = cv2.addWeighted(overlay, alpha, background, 1 - alpha, 0)
+    # 2. Keep the segmented mask area perfectly sharp and tint it
+    tinted_sharp = img_rgb.copy()
+    tinted_sharp[mask_bool] = color
+    tinted_sharp = cv2.addWeighted(tinted_sharp, alpha, img_rgb, 1 - alpha, 0)
     
-    # Draw a fancy glowing neon edge around the mask
+    # 3. Combine: Blurred background + Sharp highlighted object
+    blended = np.where(mask_bool[:, :, None], tinted_sharp, background)
+    
+    # 4. Multi-layered Glowing Edge for the mask
     mask_img = (mask_bool * 255).astype(np.uint8)
     contours, _ = cv2.findContours(mask_img, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
     
-    # Outer glow (thinner, same color)
-    cv2.drawContours(blended, contours, -1, color, 2, cv2.LINE_AA)
-    # Inner bright core (very thin, white)
-    cv2.drawContours(blended, contours, -1, (255, 255, 255), 1, cv2.LINE_AA)
+    cv2.drawContours(blended, contours, -1, (0, 100, 150), 6, cv2.LINE_AA) # Deep outer shadow/glow
+    cv2.drawContours(blended, contours, -1, color, 2, cv2.LINE_AA)         # Neon main line
+    cv2.drawContours(blended, contours, -1, (255, 255, 255), 1, cv2.LINE_AA) # Bright inner core
     
     return blended
 
@@ -86,16 +89,33 @@ def draw_boxes(img_rgb, boxes, labels, color=(0, 215, 255)):
     for box, label in zip(boxes, labels):
         x1, y1, x2, y2 = map(int, box)
         
-        # Solid continuous bounding box
-        cv2.rectangle(out, (x1, y1), (x2, y2), color, 2)
+        # Faint inner bounding box line
+        cv2.rectangle(out, (x1, y1), (x2, y2), color, 1)
+        
+        # HUD-Style Corner Brackets
+        length = int(min(x2 - x1, y2 - y1) * 0.15)
+        thick = 3
         
-        # Draw a solid background for the text label for better readability
-        (w, h), _ = cv2.getTextSize(label, cv2.FONT_HERSHEY_SIMPLEX, 0.5, 1)
-        cv2.rectangle(out, (x1, max(y1 - 25, 0)), (x1 + w + 10, max(y1 - 25, 0) + h + 10), color, -1)
+        # Top-Left
+        cv2.line(out, (x1, y1), (x1 + length, y1), color, thick, cv2.LINE_AA)
+        cv2.line(out, (x1, y1), (x1, y1 + length), color, thick, cv2.LINE_AA)
+        # Top-Right
+        cv2.line(out, (x2, y1), (x2 - length, y1), color, thick, cv2.LINE_AA)
+        cv2.line(out, (x2, y1), (x2, y1 + length), color, thick, cv2.LINE_AA)
+        # Bottom-Left
+        cv2.line(out, (x1, y2), (x1 + length, y2), color, thick, cv2.LINE_AA)
+        cv2.line(out, (x1, y2), (x1, y2 - length), color, thick, cv2.LINE_AA)
+        # Bottom-Right
+        cv2.line(out, (x2, y2), (x2 - length, y2), color, thick, cv2.LINE_AA)
+        cv2.line(out, (x2, y2), (x2, y2 - length), color, thick, cv2.LINE_AA)
+
+        # Sleek Floating Badge for text (Tesla/Palantir UI style)
+        (w, h), _ = cv2.getTextSize(label, cv2.FONT_HERSHEY_DUPLEX, 0.5, 1)
+        cv2.rectangle(out, (x1, max(y1 - 32, 0)), (x1 + w + 16, max(y1 - 32, 0) + h + 12), color, -1)
         
-        # Draw the text in white
-        cv2.putText(out, label, (x1 + 5, max(y1 - 25, 0) + h + 5),
-                    cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255, 255, 255), 1, cv2.LINE_AA)
+        # Black text on Cyan badge gives a very premium, high-contrast look
+        cv2.putText(out, label, (x1 + 8, max(y1 - 32, 0) + h + 8),
+                    cv2.FONT_HERSHEY_DUPLEX, 0.5, (0, 0, 0), 1, cv2.LINE_AA)
     return out
 
 # ═══════════════════════════════════════════════════════════════════════════════