Stream subprocess output line-by-line; increase GPU duration to 360s

app.py

@@ -105,9 +105,13 @@ setup()
 RESOLUTIONS = ["832x480", "480x832", "640x640", "1024x576", "576x1024"]
 
 
-@spaces.GPU(duration=180)
-def _generate(image: str, prompt: str, model: str, steps: int,
-              guidance_scale: float, resolution: str, frames: int, seed: int):
+@spaces.GPU(duration=360)
+def generate_video(image, prompt, resolution, steps, guidance_scale, frames, seed):
+    if image is None:
+        raise gr.Error("Please upload an image.")
+    if not prompt.strip():
+        raise gr.Error("Please enter a prompt.")
+
     out_file = Path(tempfile.mkdtemp()) / "output.mp4"
     env = {**os.environ, "WAN2GP_ROOT": str(WAN2GP_ROOT)}
 
@@ -116,46 +120,34 @@ def _generate(image: str, prompt: str, model: str, steps: int,
         "--image",  image,
         "--prompt", prompt,
         "--output", str(out_file),
-        "--model",  model,
-        "--seed",   str(seed),
+        "--model",  "sulphur-2",
+        "--seed",   str(int(seed)),
         "--resolution", resolution,
-        "--steps",  str(steps),
-        "--guidance_scale", str(guidance_scale),
-        "--frames", str(frames),
+        "--steps",  str(int(steps)),
+        "--guidance_scale", str(float(guidance_scale)),
+        "--frames", str(int(frames)),
     ]
 
     log_lines = []
-    proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE,
+    proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT,
                             text=True, bufsize=1, env=env)
+
     for line in proc.stdout:
-        log_lines.append(line.rstrip())
-        print(line, end="")
-    stderr_out = proc.stderr.read()
-    proc.wait()
-    if stderr_out:
-        log_lines.append("--- stderr ---")
-        log_lines.extend(stderr_out.splitlines())
-        print(stderr_out)
+        stripped = line.rstrip()
+        log_lines.append(stripped)
+        print(stripped)
+        yield None, "\n".join(log_lines[-30:])
 
+    proc.wait()
     log = "\n".join(log_lines)
+
     if proc.returncode != 0 or not out_file.exists():
-        return None, log + "\n[ERROR]"
+        yield None, log + "\n\n[ERROR] Generation failed."
+        return
 
     final = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False)
     shutil.copy2(out_file, final.name)
-    return final.name, log + "\n[DONE]"
-
-
-def generate_video(image, prompt, resolution, steps, guidance_scale, frames, seed):
-    if image is None:
-        raise gr.Error("Please upload an image.")
-    if not prompt.strip():
-        raise gr.Error("Please enter a prompt.")
-    yield None, f"Generating {resolution}, {steps} steps…"
-    video, log = _generate(image, prompt, "sulphur-2",
-                           int(steps), float(guidance_scale),
-                           resolution, int(frames), int(seed))
-    yield video, log
+    yield final.name, log + "\n\n[DONE]"
 
 
 with gr.Blocks(title="Sulphur — Image to Video") as demo: