app.py

import os
import asyncio
import edge_tts
import gradio as gr
from moviepy.editor import *
from transformers import pipeline
import requests
from datetime import datetime

# 1. Configuración inicial
VOICES = asyncio.run(edge_tts.list_voices())
PEXELS_API_KEY = os.getenv("PEXELS_API_KEY")

# 2. Generador de guión
def generar_guion(prompt):
    try:
        generator = pipeline("text-generation", model="facebook/mbart-large-50")
        return generator(
            f"Genera un guion sobre '{prompt}':",
            max_length=200,
            num_return_sequences=1
        )[0]['generated_text']
    except:
        return prompt  # Fallback

# 3. Función principal (sincrónica)
def crear_video(prompt, script_personalizado, voz_seleccionada, musica=None):
    try:
        # A. Generar contenido
        guion = script_personalizado if script_personalizado else generar_guion(prompt)
        
        # B. Generar voz (solución sincrónica)
        os.system(f'edge-tts --voice "{voz_seleccionada}" --text "{guion}" --write-media "voz.mp3"')
        
        # C. Buscar videos en Pexels
        headers = {"Authorization": PEXELS_API_KEY}
        query = prompt[:50].replace(" ", "+")
        videos = requests.get(
            f"https://api.pexels.com/videos/search?query={query}&per_page=2",
            headers=headers,
            timeout=10
        ).json().get("videos", [])
        
        # D. Procesar música
        audio = AudioFileClip("voz.mp3")
        if musica:
            musica_clip = AudioFileClip(musica.name)
            if musica_clip.duration < audio.duration:
                musica_clip = musica_clip.loop(duration=audio.duration)
            audio = CompositeAudioClip([audio, musica_clip.volumex(0.3)])
        
        # E. Renderizar video
        clips = [VideoFileClip(v["video_files"][0]["link"]).subclip(0, 5) for v in videos[:2]]
        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)
        
        return output_path
    except Exception as e:
        print(f"ERROR: {str(e)}")
        return None

# 4. Interfaz Gradio (versión estable)
with gr.Blocks() as app:
    gr.Markdown("# 🎬 GENERADOR DE VIDEOS")
    
    with gr.Row():
        with gr.Column():
            prompt = gr.Textbox(label="Tema del video")
            script = gr.TextArea(label="Guion personalizado (opcional)", lines=5)
            voz = gr.Dropdown(
                label="Selecciona voz",
                choices=[v["Name"] for v in VOICES],
                value="es-ES-ElviraNeural"
            )
            musica = gr.File(label="Música (opcional)", file_types=[".mp3"])
            btn = gr.Button("Generar Video", variant="primary")
        
        with gr.Column():
            output = gr.Video(label="Resultado")

    btn.click(
        fn=crear_video,
        inputs=[prompt, script, voz, musica],
        outputs=output
    )

# 5. Lanzamiento SIN parámetros conflictivos
app.launch(server_name="0.0.0.0", server_port=7860)