Update app.py

app.py

@@ -6,8 +6,16 @@ import zipfile
 from PIL import Image
 from datetime import datetime
 
-def procesar_video(video_path):
+def procesar_video(video):
     try:
+        # Determinar la ruta del video y el nombre original
+        if isinstance(video, dict):
+            original_name = video.get("name", "video")
+            video_path = video.get("file", video.get("data"))
+        else:
+            original_name = os.path.basename(video)
+            video_path = video
+
         # Configurar directorio temporal
         temp_dir = f"temp_{datetime.now().strftime('%Y%m%d%H%M%S')}"
         os.makedirs(temp_dir, exist_ok=True)
@@ -52,8 +60,10 @@ def procesar_video(video_path):
         collage_path = os.path.join(temp_dir, "collage.jpg")
         collage.save(collage_path)
         
-        # Crear archivo ZIP con TODOS los frames
-        zip_path = os.path.join(temp_dir, "frames.zip")
+        # Crear archivo ZIP con TODOS los frames usando el nombre original del video
+        base_name = os.path.splitext(original_name)[0]
+        zip_filename = f"{base_name}.zip"
+        zip_path = os.path.join(temp_dir, zip_filename)
         with zipfile.ZipFile(zip_path, 'w') as zipf:
             for img in frame_paths:
                 zipf.write(img, os.path.basename(img))
@@ -84,15 +94,20 @@ with gr.Blocks(title="Extracción de Fotogramas Forenses") as demo:
             download_btn = gr.Button("DESCARGAR FOTOGRAMAS", interactive=False)
             download_file = gr.File(label="Archivo ZIP generado", visible=False)
     
+    # Estados para guardar el directorio temporal y la ruta del ZIP
     temp_dir_state = gr.State()
+    zip_path_state = gr.State()
     
+    # Habilita el botón de procesar solo si se ha subido un video
     def habilitar_procesar(video):
-        return gr.Button(interactive=True) if video else gr.Button(interactive=False)
+        return gr.Button.update(interactive=True) if video else gr.Button.update(interactive=False)
     
+    # Procesa el video, genera el collage y el ZIP y actualiza los estados
     def procesar_y_mostrar(video):
-        limpiar_cache(temp_dir_state.value)
+        if temp_dir_state.value:
+            limpiar_cache(temp_dir_state.value)
         collage_path, zip_path, temp_dir = procesar_video(video)
-        return collage_path, zip_path, temp_dir, gr.Button(interactive=True)
+        return collage_path, zip_path, temp_dir, zip_path, gr.Button.update(interactive=True)
     
     video_input.change(
         fn=habilitar_procesar,
@@ -104,14 +119,15 @@ 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_file, temp_dir_state, download_btn],
+        outputs=[gallery_output, download_file, temp_dir_state, zip_path_state, download_btn],
     )
     
+    # Al pulsar "DESCARGAR FOTOGRAMAS", se envía la ruta del ZIP almacenada en zip_path_state
     download_btn.click(
-        fn=lambda temp_dir: os.path.join(temp_dir, "frames.zip"),
-        inputs=temp_dir_state,
+        fn=lambda zip_path: zip_path,
+        inputs=zip_path_state,
         outputs=download_file,
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()