Update app.py

app.py

@@ -61,17 +61,15 @@ def procesar_video(video):
         collage_path = os.path.join(temp_dir, "collage.jpg")
         collage.save(collage_path)
 
-        # Generación del ZIP con formato mejorado
+        # 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 frames
             for img_path in frame_paths:
                 zipf.write(img_path, os.path.basename(img_path))
             
-            # Cadena de custodia con formato mejorado
             with open(video_path, "rb") as f:
                 video_hash = hashlib.md5(f.read()).hexdigest()
             
@@ -82,7 +80,7 @@ def procesar_video(video):
                 f"• Fotogramas extraídos: {frame_count}\r\n"
                 f"• Hash MD5 del video original: {video_hash}\r\n\r\n"
                 "Firmado digitalmente por:\n"
-                "Sistema de Extracción Forense - Grupo de Peritos Forenses Digitales de Guatemala"
+                "Sistema de Extracción Forense - GPFD Guatemala"
             )
             zipf.writestr("cadena_custodia.txt", chain_content)
 
@@ -91,24 +89,54 @@ def procesar_video(video):
     except Exception as e:
         raise gr.Error(f"Error al procesar el video: {str(e)}")
 
-# ... (El resto del código permanece igual hasta la interfaz)
+def limpiar_cache(temp_dir):
+    if temp_dir and os.path.exists(temp_dir):
+        for file in os.listdir(temp_dir):
+            os.remove(os.path.join(temp_dir, file))
+        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)")
     
     with gr.Row():
         with gr.Column():
-            # Video input sin webcam
             video_input = gr.Video(
                 label="Subir Video", 
                 interactive=True,
-                sources=["upload"]  # Solo permite subir archivos
+                sources=["upload"],
+                format="mp4"
             )
             procesar_btn = gr.Button("Procesar Fotogramas", interactive=False)
         with gr.Column():
             gallery_output = gr.Image(label="Collage de Muestra")
             download_file = gr.File(label="Archivo ZIP generado", visible=True)
+    
+    temp_dir_state = gr.State()
+    zip_path_state = gr.State()
+    
+    def on_video_change(video):
+        return gr.update(interactive=bool(video))
+    
+    video_input.change(
+        fn=on_video_change,
+        inputs=video_input,
+        outputs=procesar_btn,
+        queue=False
+    )
+    
+    def procesar_y_mostrar(video):
+        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  # 4 outputs
+    
+    procesar_btn.click(
+        fn=procesar_y_mostrar,
+        inputs=video_input,
+        outputs=[gallery_output, download_file, temp_dir_state, zip_path_state]
+    )
 
-    # ... (El resto del código permanece igual)
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)