Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import gradio as gr
 import cv2
 import os
@@ -7,9 +6,17 @@ from PIL import Image, ImageOps
 from datetime import datetime
 import hashlib
 import shutil
+from concurrent.futures import ThreadPoolExecutor
 
 TEMP_CACHE = None
 
+def guardar_frame(frame, index, temp_dir):
+    frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+    img = Image.fromarray(frame_rgb)
+    img_path = os.path.join(temp_dir, f"frame_{index:04d}.jpg")
+    img.save(img_path)
+    return img_path
+
 def procesar_video(video_path):
     try:
         original_name = os.path.basename(video_path)
@@ -22,20 +29,20 @@ def procesar_video(video_path):
         os.makedirs(temp_dir, exist_ok=True)
 
         cap = cv2.VideoCapture(video_path)
-        frame_count = 0
-        frame_paths = []
-        while True:
-            ret, frame = cap.read()
-            if not ret:
-                break
-            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
+        index = 0
+        futures = []
+        with ThreadPoolExecutor(max_workers=4) as executor:
+            while True:
+                ret, frame = cap.read()
+                if not ret:
+                    break
+                futures.append(executor.submit(guardar_frame, frame, index, temp_dir))
+                index += 1
         cap.release()
 
+        frame_paths = [f.result() for f in futures]
+        frame_count = len(frame_paths)
+
         if frame_count == 0:
             raise gr.Error("No se pudieron extraer fotogramas del video")
 
@@ -114,9 +121,7 @@ def limpiar_cache():
 
 with gr.Blocks(title="Extractor Forense de Fotogramas") as demo:
     gr.Markdown("# 📷 Extractor Forense de Fotogramas de Videos")
-    gr.Markdown("""
-**Herramienta certificada para extracción forense de fotogramas de videos**  (No se guarda ninguna información).
-""")
+    gr.Markdown("""**Herramienta certificada para extracción forense de fotogramas de videos**  (No se guarda ninguna información).""")
     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():
@@ -125,7 +130,7 @@ with gr.Blocks(title="Extractor Forense de Fotogramas") as demo:
                 label="🎞️ VIDEO CARGADO",
                 format="mp4",
                 interactive=True,
-                height=380,
+                height=480,
                 sources=["upload"]
             )
             procesar_btn = gr.Button("🔍 INICIAR ANÁLISIS", interactive=False)