| import os | |
| os.system('pip install dashscope') | |
| import gradio as gr | |
| from http import HTTPStatus | |
| import dashscope | |
| from dashscope import VideoSynthesis | |
| DASHSCOPE_API_KEY = os.getenv('DASHSCOPE_API_KEY') | |
| dashscope.api_key = DASHSCOPE_API_KEY | |
| def video_generation(prompt): | |
| try: | |
| rsp = VideoSynthesis.call(model="wanx2.1-t2v-plus", prompt=prompt, watermark_wanx=True) | |
| video_url = rsp.output.video_url | |
| return video_url | |
| except Exception as e: | |
| gr.Warning(f"Warning: {e}") | |
| return None | |
| examples = [ | |
| 'a tiny astronaut hatching from an egg on the moon', | |
| 'A close-up of a panda wearing red clothes and holding a green sign with the words "Wanx" in both hands. The background is a busy city street with cars driving fast on the road.', | |
| '远景拍摄，塞纳河畔，绚烂的烟花在空中绽放，烟花形成了粉色数字“2025”时镜头拉近特写，然后逐渐消散', | |
| ] | |
| css = """ | |
| #col-container { | |
| margin: 0 auto; | |
| max-width: 800px; | |
| } | |
| """ | |
| with gr.Blocks(css=css) as demo: | |
| with gr.Column(elem_id="col-container"): | |
| gr.Markdown(f"""# WanX 2.1 (Tongyi Wanxiang 2.1), the latest powerful text to video generation model developed by the WanX Team from Tongyi Lab, Alibaba Group. | |
| [[YouTube Videos]](https://www.youtube.com/watch?v=bq4nAVbYQQU) | |
| [[WanX WEB]](https://tongyi.aliyun.com/wanxiang/videoCreation) | |
| """) | |
| with gr.Row(): | |
| prompt = gr.Text( | |
| label="Prompt", | |
| show_label=False, | |
| max_lines=100, | |
| min_width=320, | |
| placeholder="Enter your prompt", | |
| container=False, | |
| ) | |
| run_button = gr.Button("Run", scale=0) | |
| result = gr.Video(label="Generated video", show_label=False) | |
| gr.Examples( | |
| examples=examples, | |
| inputs=[prompt] | |
| ) | |
| gr.on( | |
| triggers=[run_button.click], | |
| fn=video_generation, | |
| inputs=[prompt], | |
| outputs=[result] | |
| ) | |
| demo.queue(max_size=10) | |
| demo.launch() |