From 72c9d8c4854567ace0d826dd51da95cb242eefa3 Mon Sep 17 00:00:00 2001
From: Mikei386 <44135113+Mikei386@users.noreply.github.com>
Date: Thu, 10 Sep 2026 21:52:21 +0200
Subject: [PATCH] Fix YuE2 playground output path

---
 experiments/yue2-3b/ui/server.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/experiments/yue2-3b/ui/server.py b/experiments/yue2-3b/ui/server.py
index 37b718a..c790b57 100644
--- a/experiments/yue2-3b/ui/server.py
+++ b/experiments/yue2-3b/ui/server.py
@@ -71,7 +71,8 @@ def run_job(job_id: str, request: dict) -> None:
     log_file = LOGS / f"{job_id}.log"
     command = [
         "yue2", "generate", "--offline", "--device", "cuda:0", "--budget", "16",
-        "--request", str(request_file), "--output", str(output),
+        # YuE2 creates a child directory from request["id"] itself.
+        "--request", str(request_file), "--output", str(ROOT),
     ]
     started = time.monotonic()
     try:
@@ -245,7 +246,12 @@ class Handler(BaseHTTPRequestHandler):
                 chunk = source.read(min(1024 * 1024, remaining))
                 if not chunk:
                     break
-                self.wfile.write(chunk)
+                try:
+                    self.wfile.write(chunk)
+                except (BrokenPipeError, ConnectionResetError):
+                    # Browsers routinely close an old range request after a
+                    # seek or metadata probe. This is not a server failure.
+                    break
                 remaining -= len(chunk)