Update app.py

app.py

@@ -4,27 +4,33 @@ import gradio as gr
 from moviepy.editor import *
 import requests
 from datetime import datetime
+import tempfile
 
-# 1. Configuración inicial
-PEXELS_API_KEY = os.getenv("PEXELS_API_KEY")  # Configura esto en Hugging Face
+# Configuración inicial
+PEXELS_API_KEY = os.getenv("PEXELS_API_KEY")  # Configurar en variables de entorno
 
-# 2. Lista de voces CORRECTA (formato válido para edge-tts)
+# Lista de voces válidas
 VOICES = [
-    "es-MX-DaliaNeural",    # Mujer México
-    "es-ES-ElviraNeural",   # Mujer España
-    "es-AR-ElenaNeural",    # Mujer Argentina
-    "es-CO-SalomeNeural",   # Mujer Colombia
-    "en-US-JennyNeural",    # Mujer EE.UU. (inglés)
-    "fr-FR-DeniseNeural",   # Mujer Francia (francés)
-    "de-DE-KatjaNeural",    # Mujer Alemania (alemán)
-    "pt-BR-FranciscaNeural" # Mujer Brasil (portugués)
-    # ¡Añade más si necesitas!
+    "es-MX-DaliaNeural", "es-ES-ElviraNeural", 
+    "es-AR-ElenaNeural", "en-US-JennyNeural",
+    "fr-FR-DeniseNeural", "de-DE-KatjaNeural"
 ]
 
-# 3. Función principal (a prueba de errores)
+def descargar_video(url, output_path):
+    """Descarga un video y lo guarda localmente"""
+    try:
+        response = requests.get(url, stream=True, timeout=15)
+        with open(output_path, 'wb') as f:
+            for chunk in response.iter_content(chunk_size=1024*1024):
+                f.write(chunk)
+        return True
+    except Exception as e:
+        print(f"Error descargando video: {str(e)}")
+        return False
+
 def generar_video(prompt, voz_seleccionada, musica=None):
     try:
-        # A. Generar voz
+        # 1. Generar voz
         voz_archivo = "voz.mp3"
         subprocess.run([
             'edge-tts',
@@ -33,7 +39,7 @@ def generar_video(prompt, voz_seleccionada, musica=None):
             '--write-media', voz_archivo
         ], check=True)
         
-        # B. Buscar videos en Pexels
+        # 2. Buscar videos en Pexels
         headers = {"Authorization": PEXELS_API_KEY}
         response = requests.get(
             f"https://api.pexels.com/videos/search?query={prompt[:50]}&per_page=2",
@@ -42,7 +48,19 @@ def generar_video(prompt, voz_seleccionada, musica=None):
         )
         videos = response.json().get("videos", [])
         
-        # C. Procesar música
+        # 3. Descargar y procesar videos
+        clips = []
+        for v in videos[:2]:
+            video_url = v["video_files"][0]["link"]
+            temp_video = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False)
+            if descargar_video(video_url, temp_video.name):
+                clip = VideoFileClip(temp_video.name).subclip(0, min(5, VideoFileClip(temp_video.name).duration))
+                clips.append(clip)
+        
+        if not clips:
+            raise Exception("No se pudieron cargar videos válidos")
+        
+        # 4. Procesar audio
         audio = AudioFileClip(voz_archivo)
         if musica:
             musica_clip = AudioFileClip(musica.name)
@@ -50,8 +68,7 @@ def generar_video(prompt, voz_seleccionada, musica=None):
                 musica_clip = musica_clip.loop(duration=audio.duration)
             audio = CompositeAudioClip([audio, musica_clip.volumex(0.3)])
         
-        # D. Crear video
-        clips = [VideoFileClip(v["video_files"][0]["link"]).subclip(0, 5) for v in videos[:2]]
+        # 5. Crear video final
         final_clip = concatenate_videoclips(clips).set_audio(audio)
         output_path = f"video_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4"
         final_clip.write_videofile(output_path, codec="libx264", audio_codec="aac", threads=2)
@@ -60,20 +77,24 @@ def generar_video(prompt, voz_seleccionada, musica=None):
     except Exception as e:
         print(f"ERROR: {str(e)}")
         return None
+    finally:
+        # Limpieza
+        if os.path.exists(voz_archivo):
+            os.remove(voz_archivo)
 
-# 4. Interfaz Gradio (simple pero efectiva)
+# Interfaz Gradio
 with gr.Blocks() as app:
-    gr.Markdown("# 🎬 GENERADOR DE VIDEOS (ESTABLE)")
+    gr.Markdown("# 🎬 Generador de Videos Automático")
     
     with gr.Row():
         with gr.Column():
-            prompt = gr.Textbox(label="Tema del video", placeholder="Ej: 'Lugares misteriosos de España'")
-            voz = gr.Dropdown(label="Selecciona voz", choices=VOICES, value="es-ES-ElviraNeural")
+            prompt = gr.Textbox(label="Tema del video")
+            voz = gr.Dropdown(label="Selección de voz", choices=VOICES, value="es-ES-ElviraNeural")
             musica = gr.File(label="Música de fondo (opcional)", file_types=[".mp3"])
-            btn = gr.Button("Generar Video", variant="primary")
+            btn = gr.Button("Generar Video")
         
         with gr.Column():
-            output = gr.Video(label="Resultado", autoplay=True)
+            output = gr.Video(label="Resultado")
 
     btn.click(
         fn=generar_video,
@@ -81,4 +102,5 @@ with gr.Blocks() as app:
         outputs=output
     )
 
-app.launch(server_name="0.0.0.0", server_port=7860)
+if __name__ == "__main__":
+    app.launch(server_name="0.0.0.0", server_port=7860)