Update app.py

app.py

@@ -77,22 +77,39 @@ def _extract_video_from_history(history: Dict[str, Any]) -> Dict[str, str]:
                     return {"filename": it["filename"], "subfolder": it["subfolder"], "type": it["type"]}
     raise RuntimeError("No video file found in history outputs")
 
-with gr.Blocks(title="Wan 2.2 T2V UI running on AMD MI300x") as demo:
+sample_prompts = [
+    "A golden retriever running across a beach at sunset, cinematic, 24fps",
+    "A cyberpunk city street at night with neon lights, light rain, slow pan",
+    "Aerial drone shot over snowy mountains, 5 seconds, dramatic lighting",
+    "Cartoon-style cat riding a skateboard in a park, vibrant colors"
+]
+
+with gr.Blocks(
+    title="T2V UI",
+    theme=gr.themes.Soft(primary_hue="blue", secondary_hue="blue", neutral_hue="slate"),
+) as demo:
     st_token = gr.State()
-    with gr.Row():
-        text = gr.Textbox(label="Prompt", placeholder="Text to generate", lines=3)
-    with gr.Row():
-        width = gr.Number(label="Width", value=1280, precision=0)
-        height = gr.Number(label="Height", value=704, precision=0)
-        length = gr.Number(label="FPS", value=121, precision=0)
-        fps = gr.Number(label="FPS", value=24, precision=0)
-    with gr.Row():
-        steps = gr.Number(label="Steps", value=20, precision=0)
-        cfg = gr.Number(label="CFG", value=5.0)
-        seed = gr.Number(label="Seed", value=None)
-    filename_prefix = gr.Textbox(label="Prefix of video", value="video/ComfyUI")
-    run_btn = gr.Button("Generate")
-    prog_bar = gr.Slider(label="Step", minimum=0, maximum=100, value=0, step=1, interactive=False)
+    gr.Markdown("# Wan2.2 T2V running on AMD MI300x")
+    gr.Markdown("### Prompt")
+    text = gr.Textbox(label="Prompt", placeholder="Describe the video you want", lines=3)
+    
+    gr.Examples(examples=sample_prompts, inputs=text)
+
+    with gr.Accordion("Advanced Settings", open=False):
+        with gr.Row():
+            width = gr.Number(label="Width", value=1280, precision=0)
+            height = gr.Number(label="Height", value=704, precision=0)
+        with gr.Row():
+            length = gr.Number(label="Frames", value=121, precision=0)
+            fps = gr.Number(label="FPS", value=24, precision=0)
+        with gr.Row():
+            steps = gr.Number(label="Steps", value=20, precision=0)
+            cfg = gr.Number(label="CFG", value=5.0)
+            seed = gr.Number(label="Seed (optional)", value=None)
+        filename_prefix = gr.Textbox(label="Filename prefix", value="video/ComfyUI")
+
+    run_btn = gr.Button("Generate", variant="primary")
+    prog_bar = gr.Slider(label="Progress", minimum=0, maximum=100, value=0, step=1, interactive=False)
     out_video = gr.Video(label="Result")
 
     def _init_token():
@@ -101,62 +118,63 @@ with gr.Blocks(title="Wan 2.2 T2V UI running on AMD MI300x") as demo:
     demo.load(_init_token, outputs=st_token)
 
     def generate_fn(text, width, height, length, fps, steps, cfg, seed, filename_prefix, token):
-        def _runner():
-            req = T2VReq(
-                token=token,
-                text=text,
-                seed=int(seed) if seed is not None else None,
-                steps=int(steps) if steps is not None else None,
-                cfg=float(cfg) if cfg is not None else None,
-                width=int(width) if width is not None else None,
-                height=int(height) if height is not None else None,
-                length=int(length) if length is not None else None,
-                fps=int(fps) if fps is not None else None,
-                filename_prefix=filename_prefix if filename_prefix else None,
-            )
-            prompt = _inject_params(WORKFLOW_TEMPLATE, req)
-            client_id = str(uuid.uuid4())
-            ws = _open_ws(client_id, req.token)
-            prompt_id = _queue_prompt(prompt, client_id, req.token)
-            total_nodes = max(1, len(prompt))
-            seen = set()
-            p = 0
-            last_emit = -1
-            start = time.time()
-            ws.settimeout(60)
-            while True:
-                out = ws.recv()
-                if isinstance(out, (bytes, bytearray)):
-                    if p < 95 and time.time() - start > 2:
-                        p = min(95, p + 1)
+        req = T2VReq(
+            token=token,
+            text=text,
+            seed=int(seed) if seed is not None else None,
+            steps=int(steps) if steps is not None else None,
+            cfg=float(cfg) if cfg is not None else None,
+            width=int(width) if width is not None else None,
+            height=int(height) if height is not None else None,
+            length=int(length) if length is not None else None,
+            fps=int(fps) if fps is not None else None,
+            filename_prefix=filename_prefix if filename_prefix else None,
+        )
+        prompt = _inject_params(WORKFLOW_TEMPLATE, req)
+        client_id = str(uuid.uuid4())
+        ws = _open_ws(client_id, req.token)
+        prompt_id = _queue_prompt(prompt, client_id, req.token)
+        total_nodes = max(1, len(prompt))
+        seen = set()
+        p = 0
+        last_emit = -1
+        start = time.time()
+        ws.settimeout(60)
+        while True:
+            out = ws.recv()
+            if isinstance(out, (bytes, bytearray)):
+                if p < 95 and time.time() - start > 2:
+                    p = min(95, p + 1)
+                if p != last_emit:
+                    last_emit = p
+                    yield p, None
+                continue
+            msg = json.loads(out)
+            if msg.get("type") == "executing":
+                data = msg.get("data", {})
+                if data.get("prompt_id") != prompt_id:
+                    continue
+                node = data.get("node")
+                if node is None:
+                    break
+                if node not in seen:
+                    seen.add(node)
+                    p = min(99, int(len(seen) / total_nodes * 100))
                     if p != last_emit:
                         last_emit = p
                         yield p, None
-                    continue
-                msg = json.loads(out)
-                if msg.get("type") == "executing":
-                    data = msg.get("data", {})
-                    if data.get("prompt_id") != prompt_id:
-                        continue
-                    node = data.get("node")
-                    if node is None:
-                        break
-                    if node not in seen:
-                        seen.add(node)
-                        p = min(99, int(len(seen) / total_nodes * 100))
-                        if p != last_emit:
-                            last_emit = p
-                            yield p, None
-            ws.close()
-            hist = _get_history(prompt_id, req.token)
-            info = _extract_video_from_history(hist)
-            q = urlencode(info)
-            video_url = f"http://{COMFY_HOST}/view?{q}"
-            yield 100, video_url
-        return _runner()
+        ws.close()
+        hist = _get_history(prompt_id, req.token)
+        info = _extract_video_from_history(hist)
+        q = urlencode(info)
+        video_url = f"http://{COMFY_HOST}/view?{q}&token={req.token}"
+        yield 100, video_url
 
     run_btn.click(
         generate_fn,
         inputs=[text, width, height, length, fps, steps, cfg, seed, filename_prefix, st_token],
         outputs=[prog_bar, out_video]
-    )
+    )
+
+if __name__ == "__main__":
+    demo.queue().launch(server_name="0.0.0.0", server_port=9001)