app.py

import gradio as gr
from transformers import pipeline

# Função para gerar vídeo a partir de texto
def gerar_video(prompt):
    # Carrega o modelo Wan 2.2 T2V diretamente do Hugging Face Hub
    pipe = pipeline("text-to-video", model="Wan-AI/Wan2.2-T2V-A14B")
    
    # Gera o vídeo com prompt do usuário
    video = pipe(prompt, max_length=60)  # ajuste o max_length conforme quiser
    return video[0]  # Retorna o arquivo de vídeo

# Interface Gradio
demo = gr.Interface(
    fn=gerar_video,
    inputs=gr.Textbox(lines=2, placeholder="Digite o prompt do vídeo aqui..."),
    outputs=gr.Video(label="Vídeo Gerado"),
    title="Agente de Vídeo Wan 2.2",
    description="Digite um texto e veja o vídeo gerado pelo modelo Wan 2.2 T2V."
)

# Lança a interface
demo.launch()