Update app.py

app.py

@@ -1201,6 +1201,125 @@ def init_database():
     finally:
         if conn:
             conn.close()
+def process_video_frame(frame):
+    """Traiter un frame vidéo avec toutes les détections"""
+    # Charger le frame
+    shared_results["img_rgb"] = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+    shared_results["img_draw"] = shared_results["img_rgb"].copy()
+
+    # Exécuter toutes les détections
+    detect_vehicle()
+    detect_color()
+    detect_orientation()
+    detect_logo_and_model()
+    detect_plate()
+
+    # Retourner le frame annoté
+    return shared_results["img_draw"]
+
+def save_modified_video():
+    """Sauvegarder la vidéo annotée avec toutes les détections"""
+    if not shared_results.get("video_path"):
+        raise gr.Error("Aucune vidéo chargée")
+
+    # Préparer le writer vidéo
+    cap = cv2.VideoCapture(shared_results["video_path"])
+    if not cap.isOpened():
+        raise gr.Error("Impossible de lire la vidéo source")
+
+    fps = cap.get(cv2.CAP_PROP_FPS)
+    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+
+    # Créer un fichier temporaire pour la sortie
+    temp_dir = tempfile.gettempdir()
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    output_path = os.path.join(temp_dir, f"annotated_{timestamp}.mp4")
+
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
+
+    frame_count = 0
+    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+    progress = gr.Progress()
+
+    try:
+        while True:
+            ret, frame = cap.read()
+            if not ret:
+                break
+
+            progress(frame_count / total_frames, f"Traitement du frame {frame_count}/{total_frames}")
+
+            # Utiliser le frame pré-annoté si disponible
+            if frame_count in shared_results.get("modified_frames", {}):
+                annotated_frame = np.array(shared_results["modified_frames"][frame_count])
+            else:
+                # Traiter le frame en temps réel si non déjà annoté
+                shared_results["img_rgb"] = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+                shared_results["img_draw"] = shared_results["img_rgb"].copy()
+                shared_results["frame_count"] = frame_count
+
+                # Exécuter toutes les détections
+                detect_vehicle()
+                detect_color()
+                detect_orientation()
+                detect_logo_and_model()
+                detect_plate()
+
+                annotated_frame = shared_results["img_draw"]
+
+            # Convertir et écrire le frame
+            out.write(cv2.cvtColor(annotated_frame, cv2.COLOR_RGB2BGR))
+            frame_count += 1
+
+    except Exception as e:
+        raise gr.Error(f"Erreur lors de la sauvegarde: {str(e)}")
+    finally:
+        cap.release()
+        out.release()
+
+    # Vérifier que la vidéo a bien été créée
+    if not os.path.exists(output_path):
+        raise gr.Error("Échec de la création de la vidéo")
+
+    return output_path
+
+def process_and_save_video():
+    """Traiter et sauvegarder la vidéo annotée"""
+    if not shared_results.get("video_path"):
+        raise gr.Error("Aucune vidéo chargée")
+
+    # Créer un fichier temporaire pour la sortie
+    output_path = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False).name
+
+    cap = cv2.VideoCapture(shared_results["video_path"])
+    fps = cap.get(cv2.CAP_PROP_FPS)
+    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
+
+    frame_count = 0
+    while True:
+        ret, frame = cap.read()
+        if not ret:
+            break
+
+        # Si le frame a été modifié, utiliser la version annotée
+        if frame_count in shared_results.get("modified_frames", {}):
+            annotated_frame = np.array(shared_results["modified_frames"][frame_count])
+            out.write(cv2.cvtColor(annotated_frame, cv2.COLOR_RGB2BGR))
+        else:
+            out.write(frame)
+
+        frame_count += 1
+
+    cap.release()
+    out.release()
+
+    return output_path
 
 # ------------------------------
 # 6. INTERFACE GRADIO
@@ -1252,7 +1371,7 @@ with gr.Blocks(title="🚗 Système de Reconnaissance de Véhicules Algériens",
 
                 # Ajout du lecteur vidéo compact (initialement caché)
                 video_player = gr.Video(
-                    visible=False,
+                    visible=True,
                     label="Aperçu vidéo",
                     interactive=False,
                     height=150  # Hauteur réduite pour un espace compact
@@ -1276,6 +1395,8 @@ with gr.Blocks(title="🚗 Système de Reconnaissance de Véhicules Algériens",
 
                 with gr.Row():
                     next_frame_btn = gr.Button("Next Frame", visible=False)
+                    save_video_btn = gr.Button("Save Video", visible=True, variant="primary")
+                    saved_video = gr.Video(label="annotated video saved", visible= True, interactive=False)
 
 
             with gr.Column():
@@ -1312,6 +1433,7 @@ with gr.Blocks(title="🚗 Système de Reconnaissance de Véhicules Algériens",
                             algerian_check_output = gr.Textbox(label="Origine", scale=2)
                             action_output = gr.Textbox(label="Action recommandée", scale=3)
 
+
     # Page de gestion d'accès
     with gr.Tab("Access Management", id="access"):
         with gr.Column():
@@ -1342,7 +1464,7 @@ with gr.Blocks(title="🚗 Système de Reconnaissance de Véhicules Algériens",
         with gr.Column():
             with gr.Row():
                 refresh_db_btn = gr.Button("🔄 Refresh", variant="secondary")
-                export_csv_btn = gr.Button("📤 Exporte CSV", variant="secondary")
+                export_csv_btn = gr.Button("📤 Export CSV", variant="secondary")
                 export_db_btn = gr.Button("💾 Exporter DB", variant="secondary")
 
             db_table = gr.Dataframe(
@@ -1635,6 +1757,10 @@ with gr.Blocks(title="🚗 Système de Reconnaissance de Véhicules Algériens",
                 logo_output, model_output,
                 plate_classification, vehicle_type_output]
     )
+    save_video_btn.click(
+        fn=process_and_save_video,
+        outputs=saved_video
+    )
 
      # Connecter les nouveaux composants
     time_range.change(