import gradio as gr
import torch
from diffusers import DiffusionPipeline

model_id = "cerspense/zeroscope_v2_576w"
pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe.to("cuda" if torch.cuda.is_available() else "cpu")

def text_to_video(prompt):
    output = pipe(prompt, num_inference_steps=25)
    video_frames = output.frames[0]
    output_path = "output.mp4"

    import imageio
    imageio.mimsave(output_path, video_frames, fps=8)
    return output_path

gr.Interface(
    fn=text_to_video,
    inputs=gr.Textbox(label="أدخل وصف الفيديو (بالإنجليزية)"),
    outputs=gr.Video(label="الفيديو الناتج"),
    title="تحويل نص إلى فيديو باستخدام ZeroScope"
).launch()