Update app.py

app.py

@@ -5,7 +5,6 @@ import random
 import zipfile
 from PIL import Image
 from datetime import datetime
-import io
 import hashlib
 
 def procesar_video(video):
@@ -103,21 +102,20 @@ 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 Muestra")
-            download_btn = gr.Button("DESCARGAR FOTOGRAMAS", interactive=False)
-            download_file = gr.File(label="Archivo ZIP generado", visible=True)  # Cambiado a visible
+            download_file = gr.File(label="Archivo ZIP generado", visible=True)  # Componente visible
     
     temp_dir_state = gr.State(None)
     zip_path_state = gr.State(None)
     
     def on_video_change(video):
         if video:
-            return gr.update(interactive=True), gr.update(interactive=False)
-        return gr.update(interactive=False), gr.update(interactive=False)
+            return gr.update(interactive=True)
+        return gr.update(interactive=False)
     
     video_input.change(
         fn=on_video_change,
         inputs=video_input,
-        outputs=[procesar_btn, download_btn],
+        outputs=[procesar_btn],
         queue=False
     )
     
@@ -125,26 +123,12 @@ with gr.Blocks(title="Extracción de Fotogramas Forenses") as demo:
         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, zip_path, gr.update(interactive=True), gr.update(interactive=False)
+        return collage_path, zip_path, temp_dir, zip_path
     
     procesar_btn.click(
         fn=procesar_y_mostrar,
         inputs=video_input,
-        outputs=[gallery_output, download_file, temp_dir_state, zip_path_state, download_btn, procesar_btn],
-    )
-    
-    def trigger_download(zip_path):
-        try:
-            with open(zip_path, "rb") as f:
-                file_bytes = f.read()
-            return os.path.basename(zip_path), file_bytes
-        except Exception as e:
-            raise gr.Error(f"Error al generar descarga: {str(e)}")
-    
-    download_btn.click(
-        fn=trigger_download,
-        inputs=zip_path_state,
-        outputs=download_file,
+        outputs=[gallery_output, download_file, temp_dir_state, zip_path_state],
     )
 
 if __name__ == "__main__":