Update app.py

app.py

@@ -12,42 +12,50 @@ def procesar_video(video_path):
         temp_dir = f"temp_{datetime.now().strftime('%Y%m%d%H%M%S')}"
         os.makedirs(temp_dir, exist_ok=True)
         
-        # Leer el video
+        # Leer el video y extraer TODOS los frames
         cap = cv2.VideoCapture(video_path)
-        total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+        frame_count = 0
+        frame_paths = []
         
-        # Seleccionar 4 fotogramas aleatorios
-        frame_indices = sorted(random.sample(range(total_frames), 4))
-        selected_frames = []
-        
-        for idx in frame_indices:
-            cap.set(cv2.CAP_PROP_POS_FRAMES, idx)
+        while True:
             ret, frame = cap.read()
-            if ret:
-                frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-                img = Image.fromarray(frame_rgb)
-                img_path = os.path.join(temp_dir, f"frame_{idx}.jpg")
-                img.save(img_path)
-                selected_frames.append(img_path)
+            if not ret:
+                break
+            
+            # Guardar cada frame
+            frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+            img = Image.fromarray(frame_rgb)
+            img_path = os.path.join(temp_dir, f"frame_{frame_count:04d}.jpg")
+            img.save(img_path)
+            frame_paths.append(img_path)
+            frame_count += 1
         
         cap.release()
         
+        if frame_count == 0:
+            raise ValueError("No se pudieron extraer frames del video")
+        
+        # Seleccionar 4 frames aleatorios para el collage
+        selected_frames = random.sample(frame_paths, min(4, len(frame_paths)))
+        
         # Crear collage
         images = [Image.open(img) for img in selected_frames]
         collage = Image.new('RGB', (images[0].width * 2, images[0].height * 2))
         
-        collage.paste(images[0], (0, 0))
-        collage.paste(images[1], (images[0].width, 0))
-        collage.paste(images[2], (0, images[0].height))
-        collage.paste(images[3], (images[0].width, images[0].height))
+        for i, img in enumerate(images):
+            row = i // 2
+            col = i % 2
+            x = col * images[0].width
+            y = row * images[0].height
+            collage.paste(img, (x, y))
         
         collage_path = os.path.join(temp_dir, "collage.jpg")
         collage.save(collage_path)
         
-        # Crear archivo ZIP
+        # Crear archivo ZIP con TODOS los frames
         zip_path = os.path.join(temp_dir, "frames.zip")
         with zipfile.ZipFile(zip_path, 'w') as zipf:
-            for img in selected_frames:
+            for img in frame_paths:
                 zipf.write(img, os.path.basename(img))
         
         return collage_path, zip_path, temp_dir
@@ -63,7 +71,7 @@ def limpiar_cache(temp_dir):
 
 with gr.Blocks(title="Extracción de Fotogramas Forenses") as demo:
     gr.Markdown("# Herramienta de Extracción de Fotogramas Forenses")
-    gr.Markdown("**Carga un video para extraer 4 fotogramas aleatorios y generar un collage.**")
+    gr.Markdown("**Carga un video para extraer TODOS los fotogramas y generar un collage de muestra.**")
     gr.Markdown("Desarrollado por José R. Leonett para el Grupo de Peritos Forenses Digitales de Guatemala - [www.forensedigital.gt](https://www.forensedigital.gt)")
     
     with gr.Row():
@@ -72,8 +80,9 @@ with gr.Blocks(title="Extracción de Fotogramas Forenses") as demo:
             procesar_btn = gr.Button("Procesar Fotogramas", interactive=False)
         
         with gr.Column():
-            gallery_output = gr.Image(label="Collage de Fotogramas")
+            gallery_output = gr.Image(label="Collage de Muestra")
             download_btn = gr.Button("DESCARGAR FOTOGRAMAS", interactive=False)
+            download_file = gr.File(label="Archivo ZIP generado", visible=False)
     
     temp_dir_state = gr.State()
     
@@ -85,9 +94,6 @@ with gr.Blocks(title="Extracción de Fotogramas Forenses") as demo:
         collage_path, zip_path, temp_dir = procesar_video(video)
         return collage_path, zip_path, temp_dir, gr.Button(interactive=True)
     
-    def actualizar_descarga(temp_dir):
-        return gr.Button(interactive=bool(temp_dir))
-    
     video_input.change(
         fn=habilitar_procesar,
         inputs=video_input,
@@ -98,14 +104,14 @@ with gr.Blocks(title="Extracción de Fotogramas Forenses") as demo:
     procesar_btn.click(
         fn=procesar_y_mostrar,
         inputs=video_input,
-        outputs=[gallery_output, download_btn, temp_dir_state, download_btn],
+        outputs=[gallery_output, download_file, temp_dir_state, download_btn],
     )
     
     download_btn.click(
         fn=lambda temp_dir: os.path.join(temp_dir, "frames.zip"),
         inputs=temp_dir_state,
-        outputs=gr.File(label="Frames descargados"),
+        outputs=download_file,
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()