import gradio as gr

def generate(prompt):
    # Replace this with WAN 2.1 / ComfyUI API call later
    return "Video generation started for: " + prompt

with gr.Blocks() as demo:
    gr.Markdown("# 🎬 WAN 2.1 Video Generator")

    prompt = gr.Textbox(label="Enter Prompt")
    btn = gr.Button("Generate")

    output = gr.Textbox()

    btn.click(generate, inputs=prompt, outputs=output)

demo.queue()
demo.launch()