import gradio as gr
from transformers import pipeline

# Загрузка модели Text2video
text2video = pipeline("text-to-video-generation", model="hotshotco/Hotshot-XL")

def generate_video(text):
    # Генерация видео из текста с использованием модели
    video = text2video(text)
    # Возвращаем результат в видео-файла
    return video['video_file']

# Создание интерфейса с использованием gr.Blocks
with gr.Blocks() as demo:
    with gr.Row():
        with gr.Column(scale=2):
            text_input = gr.Textbox(label="Введите текст", placeholder="Введите описание видео здесь...")
        with gr.Column():
            video_output = gr.Video(label="Сгенерированное видео")
    
    generate_button = gr.Button("Сгенерировать видео")
    generate_button.click(fn=generate_video, inputs=text_input, outputs=video_output)

# Запуск интерфейса
demo.launch()