Update app.py

app.py

@@ -2,154 +2,118 @@ import os
 import uuid
 import subprocess
 import shutil
+import time
 from flask import Flask, request, jsonify, send_from_directory
 from threading import Thread
 
 # --- CONFIGURACIÓN ---
 OUTPUT_FOLDER = "processed_videos"
 TEMP_FOLDER = "temp_processing"
+MAX_FILE_AGE_SECONDS = 3600  # 1 hora en segundos
 
-# ✅ Creamos las carpetas necesarias aquí, al inicio del script.
-# Se ejecutarán una vez cuando Gunicorn inicie la aplicación.
+# Crea las carpetas necesarias al iniciar la app
 os.makedirs(OUTPUT_FOLDER, exist_ok=True)
 os.makedirs(TEMP_FOLDER, exist_ok=True)
 
 app = Flask(__name__, static_folder='static', static_url_path='')
-
-# Diccionario en memoria para rastrear el estado de las tareas.
 TASKS_STATUS = {}
 
-@app.route('/healthz')
-def health_check():
-    """Ruta súper ligera para el chequeo de salud de Hugging Face."""
-    return "OK", 200
-
-@app.route('/download/<filename>')
-def download_file(filename):
-
+# --- FUNCIÓN DE AUTOLIMPIEZA ---
+def cleanup_old_files():
+    """Borra archivos en OUTPUT_FOLDER más viejos que MAX_FILE_AGE_SECONDS."""
+    while True:
+        try:
+            for filename in os.listdir(OUTPUT_FOLDER):
+                file_path = os.path.join(OUTPUT_FOLDER, filename)
+                if os.path.isfile(file_path):
+                    file_age = time.time() - os.path.getmtime(file_path)
+                    if file_age > MAX_FILE_AGE_SECONDS:
+                        os.remove(file_path)
+                        print(f"Limpiado archivo antiguo: {filename}")
+        except Exception as e:
+            print(f"Error durante la limpieza: {e}")
+        time.sleep(600) # Revisa cada 10 minutos
+
+# --- LÓGICA DE PROCESAMIENTO DE VIDEO ---
 def video_processing_worker(task_id, urls):
-    """
-    Función que se ejecuta en un hilo separado para no bloquear la API.
-    Descarga, une, convierte y limpia los videos.
-    """
     task_dir = os.path.join(TEMP_FOLDER, task_id)
-    final_video_name = f"video_{task_id}.webm"
-    final_video_path = os.path.join(OUTPUT_FOLDER, final_video_name)
+    final_video_path = os.path.join(OUTPUT_FOLDER, f"video_{task_id}.webm")
     log = []
 
+    def update_status(status, progress, new_log=None):
+        if new_log:
+            log.append(new_log)
+        TASKS_STATUS[task_id] = {'status': status, 'progress': progress, 'log': "\n".join(log)}
+
     try:
-        # --- PASO 0: PREPARACIÓN ---
         os.makedirs(task_dir, exist_ok=True)
-        TASKS_STATUS[task_id] = {'status': 'processing', 'progress': 5, 'log': "Creando entorno de trabajo..."}
+        update_status('processing', 5, "Entorno de trabajo creado.")
 
-        # --- PASO 1: DESCARGA DE VIDEOS ---
         downloaded_files = []
-        total_videos = len(urls)
         for i, url in enumerate(urls):
-            progress = 5 + int(((i + 1) / total_videos) * 70) # La descarga ocupa del 5% al 75%
-            log_msg = f"({i+1}/{total_videos}) Descargando: {url}"
-            log.append(log_msg)
-            TASKS_STATUS[task_id].update({'progress': progress, 'log': "\n".join(log)})
-
+            progress = 5 + int(((i + 1) / len(urls)) * 70)
+            update_status('processing', progress, f"({i+1}/{len(urls)}) Descargando: {url}")
+            
             output_template = os.path.join(task_dir, f"{i:03d}.%(ext)s")
-            command = [
-                'yt-dlp',
-                '--no-check-certificate',
-                '-f', 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',
-                '--merge-output-format', 'mp4',
-                '-o', output_template,
-                url
-            ]
+            command = ['yt-dlp','-f','bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best','--merge-output-format','mp4','-o',output_template,url]
             subprocess.run(command, check=True, capture_output=True, text=True, encoding='utf-8')
             
             created_file = next(f for f in os.listdir(task_dir) if f.startswith(f"{i:03d}"))
             downloaded_files.append(os.path.join(task_dir, created_file))
 
-        # --- PASO 2: UNIÓN Y CONVERSIÓN CON FFMPEG ---
-        log.append("\nUniendo videos y convirtiendo a WebM...")
-        TASKS_STATUS[task_id].update({'progress': 85, 'log': "\n".join(log)})
-
+        update_status('processing', 85, "\nUniendo videos...")
         file_list_path = os.path.join(task_dir, "filelist.txt")
         with open(file_list_path, 'w', encoding='utf-8') as f:
             for file_path in downloaded_files:
                 f.write(f"file '{os.path.abspath(file_path)}'\n")
 
-        ffmpeg_command = [
-            'ffmpeg',
-            '-f', 'concat',
-            '-safe', '0',
-            '-i', file_list_path,
-            '-c:v', 'libvpx-vp9',
-            '-crf', '30',
-            '-b:v', '0',
-            '-c:a', 'libopus',
-            '-b:a', '128k',
-            '-y',
-            final_video_path
-        ]
+        ffmpeg_command = ['ffmpeg','-f','concat','-safe','0','-i',file_list_path,'-c:v','libvpx-vp9','-crf','30','-b:v','0','-c:a','libopus','-b:a','128k','-y',final_video_path]
         subprocess.run(ffmpeg_command, check=True, capture_output=True, text=True, encoding='utf-8')
         
-        log.append("\n¡Proceso completado con éxito!")
-        TASKS_STATUS[task_id] = {
-            'status': 'completed',
-            'progress': 100,
-            'log': "\n".join(log),
-            'fileUrl': f"/download/{final_video_name}"
-        }
+        update_status('completed', 100, "\n¡Proceso completado con éxito!")
+        TASKS_STATUS[task_id]['fileUrl'] = f"/download/{os.path.basename(final_video_path)}"
 
     except subprocess.CalledProcessError as e:
-        error_details = f"Error en el comando:\nSTDOUT: {e.stdout}\nSTDERR: {e.stderr}"
-        log.append(f"\n¡ERROR! El proceso falló.\n{error_details}")
-        TASKS_STATUS[task_id] = {
-            'status': 'failed',
-            'log': "\n".join(log),
-            'error': error_details
-        }
+        error_details = f"Error en comando:\nSTDOUT: {e.stdout}\nSTDERR: {e.stderr}"
+        update_status('failed', 100, f"\n¡ERROR! El proceso falló.\n{error_details}")
     except Exception as e:
-        log.append(f"\n¡ERROR INESPERADO! {str(e)}")
-        TASKS_STATUS[task_id] = {
-            'status': 'failed',
-            'log': "\n".join(log),
-            'error': str(e)
-        }
+        update_status('failed', 100, f"\n¡ERROR INESPERADO! {str(e)}")
     finally:
-        # --- PASO 3: LIMPIEZA ---
         if os.path.exists(task_dir):
             shutil.rmtree(task_dir)
 
-# --- ENDPOINTS DE LA API ---
-
+# --- RUTAS DE LA API (ENDPOINTS) ---
 @app.route('/')
 def serve_index():
-    """Sirve la página principal de la aplicación."""
     return app.send_static_file('index.html')
 
+@app.route('/healthz')
+def health_check():
+    return "OK", 200
+
 @app.route('/api/process', methods=['POST'])
 def start_processing():
-    """Inicia un nuevo trabajo de procesamiento de video."""
     urls = request.json.get('urls')
     if not urls or not isinstance(urls, list):
         return jsonify({'error': 'La lista de URLs es inválida.'}), 400
-
+    
     task_id = str(uuid.uuid4())
     TASKS_STATUS[task_id] = {'status': 'queued', 'progress': 0, 'log': 'Tarea en cola...'}
     
     thread = Thread(target=video_processing_worker, args=(task_id, urls))
     thread.start()
-    
     return jsonify({'taskId': task_id})
 
 @app.route('/api/status/<task_id>')
 def get_status(task_id):
-    """Consulta el estado de un trabajo."""
-    task = TASKS_STATUS.get(task_id)
-    if not task:
-        return jsonify({'error': 'Tarea no encontrada.'}), 404
-    return jsonify(task)
+    return jsonify(TASKS_STATUS.get(task_id, {'error': 'Tarea no encontrada.'}))
 
 @app.route('/download/<filename>')
 def download_file(filename):
-    """Permite la descarga del video final."""
     if ".." in filename or filename.startswith("/"):
         return "Nombre de archivo no válido", 400
-    return send_from_directory(OUTPUT_FOLDER, filename, as_attachment=True)
+    return send_from_directory(OUTPUT_FOLDER, filename, as_attachment=True)
+
+# --- INICIA EL HILO DE LIMPIEZA ---
+cleanup_thread = Thread(target=cleanup_old_files, daemon=True)
+cleanup_thread.start()