Update uploader/do_spaces.py (#2)

- Update uploader/do_spaces.py (9574cce5e7496bacbd2df5e173a5b0a2d6b3302d)


Co-authored-by: Zheheng Li <[email protected]>

uploader/do_spaces.py

@@ -7,27 +7,27 @@ import io
 
 def upload_mask(image, prefix="mask"):
     """
-    将分割掩码图像上传到DigitalOcean Spaces
+    Upload the segmentation mask image to DigitalOcean Spaces
     
     Args:
-        image: PIL Image对象
-        prefix: 文件名前缀
+        image: PIL Image object
+        prefix: filename prefix
         
     Returns:
-        上传文件的URL
+        URL of the uploaded file
     """
     try:
-        # 从环境变量获取凭据
+        # Get credentials from environment variables
         do_key = os.environ.get('DO_SPACES_KEY')
         do_secret = os.environ.get('DO_SPACES_SECRET')
         do_region = os.environ.get('DO_SPACES_REGION')
         do_bucket = os.environ.get('DO_SPACES_BUCKET')
         
-        # 校验凭据是否存在
+        # Validate that credentials exist
         if not all([do_key, do_secret, do_region, do_bucket]):
-            raise ValueError("缺少DigitalOcean Spaces凭据")
+            raise ValueError("Missing DigitalOcean Spaces credentials")
         
-        # 创建S3客户端
+        # Create S3 client
         session = boto3.session.Session()
         client = session.client('s3',
                                region_name=do_region,
@@ -35,15 +35,15 @@ def upload_mask(image, prefix="mask"):
                                aws_access_key_id=do_key,
                                aws_secret_access_key=do_secret)
         
-        # 生成唯一文件名
+        # Generate a unique filename
         filename = f"{prefix}_{uuid.uuid4().hex}.png"
         
-        # 将图像转换为字节流
+        # Convert the image to a byte stream
         img_byte_arr = io.BytesIO()
         image.save(img_byte_arr, format='PNG')
         img_byte_arr.seek(0)
         
-        # 上传到Spaces
+        # Upload to Spaces
         client.upload_fileobj(
             img_byte_arr,
             do_bucket,
@@ -51,10 +51,10 @@ def upload_mask(image, prefix="mask"):
             ExtraArgs={'ACL': 'public-read', 'ContentType': 'image/png'}
         )
         
-        # 返回公共URL
+        # Return the public URL
         url = f'https://{do_bucket}.{do_region}.digitaloceanspaces.com/{filename}'
         return url
     
     except Exception as e:
-        print(f"上传失败: {str(e)}")
-        return f"上传错误: {str(e)}"
+        print(f"Upload failed: {str(e)}")
+        return f"Upload error: {str(e)}"