Update labeling augmentation

main.py

@@ -463,5 +463,56 @@ async def proxy_preview_session_clear(payload: dict = Body(...)):
     except Exception as e:
         return {"success": False, "message": f"Proxy Error: {str(e)}"}
 
+
+@app.api_route("/proxy_adjustments{rest:path}", methods=["GET", "POST", "PUT"])
+async def proxy_adjustments(rest: str, request: Request):
+    """Forward preview-mask and traditional-feature adjustment requests."""
+    username = request.query_params.get("username", "")
+    content_type = request.headers.get("content-type", "")
+    files = []
+    data = {}
+    json_body = None
+    if request.method in {"POST", "PUT"}:
+        if "multipart/form-data" in content_type:
+            form = await request.form()
+            for key, value in form.multi_items():
+                if hasattr(value, "filename"):
+                    files.append((key, (value.filename, await value.read(), value.content_type)))
+                else:
+                    data[key] = str(value)
+            username = username or data.get("username", "")
+        elif "application/json" in content_type:
+            json_body = await request.json()
+            if isinstance(json_body, dict):
+                username = username or str(json_body.get("username", ""))
+    base_url = DEV_URL if username.strip().lower() == "devtest" else PROD_URL
+    target_url = f"{base_url}/adjustments{rest}"
+    if request.url.query:
+        target_url += f"?{request.url.query}"
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+
+    def make_request():
+        return request_with_hf_backoff(
+            getattr(requests, request.method.lower()),
+            target_url,
+            max_retries=0,
+            headers=headers,
+            files=files or None,
+            data=data or None,
+            json=json_body,
+            timeout=DATASET_PROXY_TIMEOUT_SECONDS,
+        )
+
+    try:
+        response = await run_in_threadpool(make_request)
+        return Response(
+            content=response.content,
+            status_code=response.status_code,
+            media_type=response.headers.get("content-type", "application/json"),
+        )
+    except requests.exceptions.RequestException as exc:
+        return JSONResponse(status_code=502, content=proxy_error_payload("Proxy Error (Hugging Face)", exc))
+
+
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", 7860)))