From 245ecb4bbdd664d996f0bc1ef168440d13fda67a Mon Sep 17 00:00:00 2001
From: zcr <cgzhou@aidlab.hk>
Date: Tue, 13 Jan 2026 12:39:51 +0800
Subject: [PATCH] 1

---
 main.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index 0df5e42..de8fa1a 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,7 @@
 import io
 import os
 import urllib.request  # 必须这样写，不能只 import urllib
+import uuid
 from typing import Optional, List
 
 import cv2
@@ -20,6 +21,7 @@ minio_client = Minio(settings.MINIO_URL, access_key=settings.MINIO_ACCESS, secre
 
 
 class SAMRequest(BaseModel):
+    user_id: int
     image_path: str = Field(..., description="图片路径，必填字段")
     type: str = Field(..., description="推理类型，必填字段")
     points: Optional[List[List[float]]] = None
@@ -91,10 +93,12 @@ class SimpleLitAPI(ls.LitAPI):
         rgba_image = image.convert("RGBA")
         rgba_np = np.array(rgba_image)
         rgba_np[:, :, 3] = mask.astype(np.uint8) * 255
+
+        object_name = f"{request.user_id}/seg_anything/{uuid.uuid4()}"
         req = oss_upload_image(
             oss_client=minio_client,
-            bucket="test",
-            object_name=f"test.png",
+            bucket="aida-users",
+            object_name=f"{object_name}.png",
             image_bytes=cv2.imencode('.png', rgba_np)[1]
         )
         return {"output": f"{req.bucket_name}/{req.object_name}"}