Update app.py

app.py

@@ -20,15 +20,15 @@ def guardar_frame(frame, index, temp_dir):
 def procesar_video(video_path):
     try:
         original_name = os.path.basename(video_path)
-        allowed_extensions = ('.mp4', '.avi', '.mov', '.mkv')
-        if not original_name.lower().endswith(allowed_extensions):
-            raise gr.Error("Solo se permiten archivos de video (mp4, avi, mov, mkv)")
 
         timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
         temp_dir = f"temp_{datetime.now().strftime('%Y%m%d%H%M%S')}"
         os.makedirs(temp_dir, exist_ok=True)
 
         cap = cv2.VideoCapture(video_path)
+        if not cap.isOpened():
+            raise gr.Error("No se pudo abrir el archivo de video. Formato no soportado o archivo dañado.")
+
         index = 0
         futures = []
         with ThreadPoolExecutor(max_workers=4) as executor:
@@ -44,7 +44,7 @@ def procesar_video(video_path):
         frame_count = len(frame_paths)
 
         if frame_count == 0:
-            raise gr.Error("No se pudieron extraer fotogramas del video")
+            raise gr.Error("No se pudieron extraer fotogramas del video.")
 
         n_seleccion = 4
         step = max(1, frame_count // (n_seleccion + 1))
@@ -93,14 +93,14 @@ def procesar_video(video_path):
                 video_hash = hashlib.md5(f.read()).hexdigest()
 
             chain_content = (
-                "=== CADENA DE CUSTODIA DIGITAL ===\r\n\r\n"
-                f"• Archivo original: {original_name}\r\n"
-                f"• Fecha de procesamiento: {timestamp}\r\n"
-                f"• Fotogramas totales: {frame_count}\r\n"
-                f"• Hash MD5 video: {video_hash}\r\n"
-                f"• Fotogramas muestra: {', '.join([f'#{i+1}' for i in selected_indices])}\r\n\r\n"
-                "Este documento certifica la integridad del proceso de extracción.\n"
-                "Sistema Certificado por Peritos Forenses Digitales de Guatemala. \n"
+                "=== CADENA DE CUSTODIA DIGITAL ===\\r\\n\\r\\n"
+                f"• Archivo original: {original_name}\\r\\n"
+                f"• Fecha de procesamiento: {timestamp}\\r\\n"
+                f"• Fotogramas totales: {frame_count}\\r\\n"
+                f"• Hash MD5 video: {video_hash}\\r\\n"
+                f"• Fotogramas muestra: {', '.join([f'#{i+1}' for i in selected_indices])}\\r\\n\\r\\n"
+                "Este documento certifica la integridad del proceso de extracción.\\n"
+                "Sistema Certificado por Peritos Forenses Digitales de Guatemala. \\n"
                 "www.forensedigital.gt"
             )
             zipf.writestr("00_CADENA_CUSTODIA.txt", chain_content)
@@ -121,7 +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**  \n(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():