Update app.py

app.py

@@ -62,19 +62,22 @@ def procesar_video(video):
         collage_path = os.path.join(temp_dir, "collage.jpg")
         collage.save(collage_path)
 
-        # Generación del ZIP con todos los fotogramas
+        # Generación del ZIP
         base_name = os.path.splitext(original_name)[0]
         zip_filename = f"{base_name}.zip"
         final_zip_path = os.path.join(temp_dir, zip_filename)
         
         with zipfile.ZipFile(final_zip_path, mode="w") as zipf:
+            # Añadir todos los frames
             for img_path in frame_paths:
                 zipf.write(img_path, os.path.basename(img_path))
             
+            # Añadir TXT de cadena de custodia
             chain_content = (
                 f"Nombre del archivo: {original_name}\n"
                 f"Fecha de carga y extracción: {timestamp}\n"
                 f"Número de fotogramas: {frame_count}\n"
+                f"Hash MD5 del video: {hashlib.md5(open(video_path, 'rb').read()).hexdigest()}\n"
             )
             zipf.writestr("cadena_custodia.txt", chain_content)
 
@@ -90,7 +93,7 @@ def limpiar_cache(temp_dir):
         os.rmdir(temp_dir)
 
 with gr.Blocks(title="Extracción de Fotogramas Forenses") as demo:
-    gr.Markdown("# Herramienta de Extracción de Fotogramas Forenses")
+    gr.Markdown("# 📷 Herramienta de Extracción de Fotogramas Forenses")
     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)")
     
@@ -101,7 +104,7 @@ with gr.Blocks(title="Extracción de Fotogramas Forenses") as demo:
         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=False)
+            download_file = gr.File(label="Archivo ZIP generado", visible=True)  # Cambiado a visible
     
     temp_dir_state = gr.State(None)
     zip_path_state = gr.State(None)
@@ -131,9 +134,12 @@ with gr.Blocks(title="Extracción de Fotogramas Forenses") as demo:
     )
     
     def trigger_download(zip_path):
-        with open(zip_path, "rb") as f:
-            file_bytes = f.read()
-        return os.path.basename(zip_path), file_bytes
+        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,