Update tagger_ui_server.py

tagger_ui_server.py

@@ -113,6 +113,46 @@ async def tag_upload(
     return _run_tagger(img, max_size, floor)
 
 
+# ---------------------------------------------------------------------------
+# PCA endpoints
+# ---------------------------------------------------------------------------
+
+@app.post("/pca/url")
+async def pca_url(
+    url:      str = Query(...),
+    max_size: int = Query(default=1024),
+):
+    from fastapi.responses import Response
+    assert _tagger is not None
+    try:
+        from inference_tagger_standalone import _open_image
+        img = _open_image(url)
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=f"Could not fetch image: {e}")
+    pca_img = _tagger.embed_pca(img, max_size=max_size)
+    buf = io.BytesIO()
+    pca_img.save(buf, format="PNG")
+    return Response(content=buf.getvalue(), media_type="image/png")
+
+
+@app.post("/pca/upload")
+async def pca_upload(
+    file:     UploadFile = File(...),
+    max_size: int        = Query(default=1024),
+):
+    from fastapi.responses import Response
+    assert _tagger is not None
+    try:
+        data = await file.read()
+        img = Image.open(io.BytesIO(data)).convert("RGB")
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=f"Could not read image: {e}")
+    pca_img = _tagger.embed_pca(img, max_size=max_size)
+    buf = io.BytesIO()
+    pca_img.save(buf, format="PNG")
+    return Response(content=buf.getvalue(), media_type="image/png")
+
+
 # ---------------------------------------------------------------------------
 # Inference helper
 # ---------------------------------------------------------------------------