Update app.py

app.py

@@ -215,9 +215,28 @@ with gr.Blocks(title="SonicMaster – Text-Guided Restoration & Mastering", fill
         concurrency_limit=1,
     )
 
-# Warm up cache & repo, then launch
+# ---------- FastAPI mount with ClientDisconnect handler (Spaces-friendly) ----------
+from fastapi import FastAPI, Request
+from starlette.responses import PlainTextResponse
+from starlette.requests import ClientDisconnect
+import gradio as gr
+
+# Warm up cache & repo (optional but nice)
 _ = get_weights_path()
 _ = ensure_repo()
-demo.queue(max_size=16).launch()
-# Or, a global default for all events:
-# demo.queue(max_size=16, default_concurrency_limit=1).launch()
+
+app = FastAPI()
+
+@app.exception_handler(ClientDisconnect)
+async def client_disconnect_handler(request: Request, exc: ClientDisconnect):
+    # Treat as benign client cancel; avoids noisy tracebacks on Spaces
+    return PlainTextResponse("Client disconnected", status_code=499)
+
+# Mount Gradio at root. Use queue if you were using it before.
+app = gr.mount_gradio_app(app, demo.queue(max_size=16), path="/")
+
+# Optional: allow local dev with `python app.py`
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)
+