Update app.py

app.py

@@ -11,8 +11,6 @@ import os
 import json
 import time
 import requests
-import tempfile
-import re
 import random
 from google_drive_upload import authenticate_google_drive, upload_to_google_drive
 from io import BytesIO
@@ -23,13 +21,25 @@ if service_account_info:
     with open('service-account.json', 'w') as f:
         json.dump(service_account_info, f)
 
-# Define la carpeta de salida
+# Define la carpeta de salida y la carpeta temporal
 output_folder = "outputs"
+temp_dir = "temp_files"
 os.makedirs(output_folder, exist_ok=True)
+os.makedirs(temp_dir, exist_ok=True)
 
 # ID de la carpeta de destino en Google Drive
 FOLDER_ID = "12S6adpanAXjf71pKKGRRPqpzbJa5XEh3"  # Reemplaza con tu ID de carpeta
 
+def cleanup_temp_files():
+    """Elimina todos los archivos temporales de la carpeta temp_files."""
+    for filename in os.listdir(temp_dir):
+        file_path = os.path.join(temp_dir, filename)
+        try:
+            if os.path.isfile(file_path):
+                os.remove(file_path)
+        except Exception as e:
+            print(f"Error deleting {file_path}: {e}")
+
 def resize_and_blur_video(clip, target_width=1920, target_height=1080):
     """Redimensiona el video al tamaño 1080p (16:9) y aplica desenfoque si es necesario."""
     try:
@@ -66,6 +76,20 @@ def resize_and_blur_video(clip, target_width=1920, target_height=1080):
         print(f"Error en resize_and_blur_video: {e}")
         return clip
 
+def download_video(link):
+    """Descarga un video desde un enlace y lo guarda en la carpeta temporal."""
+    try:
+        video_response = requests.get(link)
+        if video_response.status_code != 200:
+            return None
+        temp_video_path = os.path.join(temp_dir, f"temp_video_{int(time.time())}.mp4")
+        with open(temp_video_path, "wb") as f:
+            f.write(video_response.content)
+        return temp_video_path
+    except Exception as e:
+        print(f"Error downloading video: {e}")
+        return None
+
 def concatenate_pixabay_videos(keywords, num_videos_per_keyword=1):
     """Concatena videos de Pixabay basados en palabras clave."""
     keyword_list = [keyword.strip() for keyword in keywords.split(",") if keyword.strip()]
@@ -81,17 +105,12 @@ def concatenate_pixabay_videos(keywords, num_videos_per_keyword=1):
                 links = search_pixabay("nature", num_results=num_videos_per_keyword)
                 if not links:
                     continue
-            link = links[0]
-            video_response = requests.get(link)
-            if video_response.status_code != 200:
-                print(f"Error al descargar video desde {link}")
-                continue
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmp_video:
-                tmp_video.write(video_response.content)
-                clip = VideoFileClip(tmp_video.name)
+            temp_video_path = download_video(links[0])
+            if temp_video_path:
+                clip = VideoFileClip(temp_video_path)
                 processed_clip = resize_and_blur_video(clip)
                 video_clips.append(processed_clip)
-                os.unlink(tmp_video.name)  # Limpiamos el archivo temporal
+                os.remove(temp_video_path)  # Eliminar archivo temporal después de usarlo
         except Exception as e:
             print(f"Error procesando palabra clave '{keyword}': {e}")
             continue
@@ -189,6 +208,8 @@ def process_input(text, txt_file, mp3_file, selected_voice, rate, pitch, keyword
     except Exception as e:
         print(f"Error durante el procesamiento: {e}")
         return None
+    finally:
+        cleanup_temp_files()  # Limpiar archivos temporales al finalizar
 
 # Interfaz Gradio
 with gr.Blocks() as demo: