Update app.py

app.py

@@ -37,6 +37,53 @@ def check_camera_availability():
     
     return available_cameras
 
+def preview_polygons(poly1, poly2):
+    """Crée une prévisualisation des polygones sur une image vide"""
+    # Créer une image noire de taille standard
+    preview_img = np.zeros((640, 1200, 3), dtype=np.uint8)
+    
+    # Dessiner le polygone 1 en vert
+    if len(poly1) >= 3:
+        cv2.polylines(preview_img, [np.array(poly1, np.int32)], isClosed=True, color=(0, 255, 0), thickness=2)
+        cv2.fillPoly(preview_img, [np.array(poly1, np.int32)], color=(0, 255, 0, 0.3))
+        
+        # Ajouter des points et des annotations
+        for i, point in enumerate(poly1):
+            cv2.circle(preview_img, point, 5, (255, 255, 255), -1)
+            draw_text_with_background(preview_img, f"P1-{i+1}: {point}", 
+                                    (point[0] + 10, point[1]), font_scale=0.5, bg_color=(0, 100, 0))
+    
+    # Dessiner le polygone 2 en rouge
+    if len(poly2) >= 3:
+        cv2.polylines(preview_img, [np.array(poly2, np.int32)], isClosed=True, color=(0, 0, 255), thickness=2)
+        cv2.fillPoly(preview_img, [np.array(poly2, np.int32)], color=(0, 0, 255, 0.3))
+        
+        # Ajouter des points et des annotations
+        for i, point in enumerate(poly2):
+            cv2.circle(preview_img, point, 5, (255, 255, 255), -1)
+            draw_text_with_background(preview_img, f"P2-{i+1}: {point}", 
+                                    (point[0] + 10, point[1]), font_scale=0.5, bg_color=(100, 0, 0))
+    
+    # Ajouter une légende
+    draw_text_with_background(preview_img, "Zone 1 (Vert)", (10, 30), font_scale=0.7, bg_color=(0, 100, 0))
+    draw_text_with_background(preview_img, "Zone 2 (Rouge)", (10, 60), font_scale=0.7, bg_color=(100, 0, 0))
+    
+    # Dessiner une grille pour aider à positionner
+    grid_spacing = 100
+    grid_color = (50, 50, 50)
+    
+    for x in range(0, preview_img.shape[1], grid_spacing):
+        cv2.line(preview_img, (x, 0), (x, preview_img.shape[0]), grid_color, 1)
+        # Ajouter le numéro de coordonnée X
+        draw_text_with_background(preview_img, str(x), (x, 20), font_scale=0.5, bg_color=(30, 30, 30))
+    
+    for y in range(0, preview_img.shape[0], grid_spacing):
+        cv2.line(preview_img, (0, y), (preview_img.shape[1], y), grid_color, 1)
+        # Ajouter le numéro de coordonnée Y
+        draw_text_with_background(preview_img, str(y), (5, y), font_scale=0.5, bg_color=(30, 30, 30))
+    
+    return preview_img
+
 # --- CLASSE YOLO OPTIMISÉE ---
 class YOLOVideoProcessor:
     def __init__(self, model_path, poly1, poly2, tracker_method="bot"):
@@ -405,6 +452,17 @@ def main():
 
     # Vérifier que les polygones sont valides
     valid_polygons = len(poly1) == 4 and len(poly2) == 4
+    
+    # --- PRÉVISUALISATION DES MASQUES ---
+    st.header("🖼️ Prévisualisation des masques")
+    
+    if valid_polygons:
+        preview_image = preview_polygons(poly1, poly2)
+        preview_image_rgb = cv2.cvtColor(preview_image, cv2.COLOR_BGR2RGB)
+        st.image(preview_image_rgb, use_column_width=True, caption="Prévisualisation des masques de détection")
+        st.success("✅ Les polygones sont correctement définis.")
+    else:
+        st.warning("⚠️ Veuillez définir des polygones valides avec exactement 4 points chacun.")
 
     # Section principale de la détection en temps réel
     st.header("Détection en Temps Réel avec Webcam")