Serve lightweight previews and use the Python server directly

app.py

@@ -105,12 +105,13 @@ def make_session(image: Image.Image, example: str | None = None):
         directory = RUNTIME / sid
         directory.mkdir()
         image.save(directory / "source.png")
+        image.save(directory / "preview.webp", quality=94, method=4)
         SESSIONS[sid] = {
             "directory": directory, "size": image.size, "example": example,
             "touched": time.monotonic(), "job": None,
         }
     return {"id": sid, "width": image.width, "height": image.height,
-            "image": f"/api/sessions/{sid}/source.png", "example": example}
+            "image": f"/api/sessions/{sid}/preview.webp", "example": example}
 
 
 def update_job(jid: str, **values):
@@ -258,6 +259,7 @@ def refine(session, jid, request: RefineRequest, progress_report=None):
                            sample_id="studio", progress_callback=progress)
     report(message="Preparing region comparisons", progress=0.97)
     result.image.save(directory / f"{jid}.png")
+    result.image.save(directory / f"{jid}.webp", quality=94, method=4)
     crops = []
     with Image.open(directory / "source.png") as before:
         for i, row in enumerate(rows):
@@ -266,13 +268,13 @@ def refine(session, jid, request: RefineRequest, progress_report=None):
             box = (max(0, math.floor(x0 - padding)), max(0, math.floor(y0 - padding)),
                    min(before.width, math.ceil(x1 + padding)), min(before.height, math.ceil(y1 + padding)))
             for label, image in (("before", before), ("after", result.image)):
-                image.crop(box).save(directory / f"{jid}-{i}-{label}.png")
+                image.crop(box).save(directory / f"{jid}-{i}-{label}.webp", quality=96, method=4)
             crops.append({"id": row.id, "text": row.text, "bbox": row.bbox,
-                          "before": f"{jid}-{i}-before.png", "after": f"{jid}-{i}-after.png"})
+                          "before": f"{jid}-{i}-before.webp", "after": f"{jid}-{i}-after.webp"})
     metadata = {"regions": [row.model_dump() for row in rows], "model": request.variant,
                 "steps": request.steps, "seed": request.seed, "guidance": request.guidance}
     (directory / f"{jid}.json").write_text(json.dumps(metadata, ensure_ascii=False, indent=2))
-    return {**metadata, "image": f"{jid}.png", "regions": crops, "metadata": f"{jid}.json"}
+    return {**metadata, "image": f"{jid}.png", "preview": f"{jid}.webp", "regions": crops, "metadata": f"{jid}.json"}
 
 
 @app.get("/")
@@ -367,7 +369,7 @@ def job(sid: str, jid: str):
 def session_file(sid: str, filename: str):
     session = session_for(sid)
     # File names are server-generated. A session cannot address another session's data.
-    if Path(filename).name != filename or not filename.endswith((".png", ".json")):
+    if Path(filename).name != filename or not filename.endswith((".png", ".webp", ".json")):
         raise HTTPException(404, "File not found")
     path = session["directory"] / filename
     if not path.is_file():

space.py

@@ -91,4 +91,4 @@ def refine_image(sid: str, payload: str) -> dict:
 
 if __name__ == "__main__":
     studio.app.launch(server_name=os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0"),
-                      server_port=int(os.environ.get("PIXTEXT_PORT", "7860")), show_error=False)
+                      server_port=int(os.environ.get("PIXTEXT_PORT", "7860")), show_error=False, ssr_mode=False)