init

util.py

@@ -13,6 +13,7 @@ import gradio as gr
 import boto3
 import tempfile
 import io
+import uuid
 from botocore.client import Config
 from PIL import Image
 
@@ -183,7 +184,9 @@ def upload_user_img_r2(clientIp, timeId, pil_image):
     Returns:
         str: Uploaded URL
     """
-    fileName = clientIp.replace(".", "")+str(timeId)+".jpg"
+    # Generate unique filename using UUID to prevent file conflicts in concurrent environment
+    unique_id = str(uuid.uuid4())
+    fileName = f"user_img_{unique_id}_{timeId}.jpg"
     
     # Convert PIL Image to bytes
     img_buffer = io.BytesIO()
@@ -270,7 +273,9 @@ def upload_mask_image_r2(client_ip, time_id, mask_image):
     Returns:
         str: Uploaded URL
     """
-    file_name = f"{client_ip.replace('.', '')}{time_id}_mask.png"
+    # Generate unique filename using UUID to prevent file conflicts in concurrent environment
+    unique_id = str(uuid.uuid4())
+    file_name = f"mask_img_{unique_id}_{time_id}.png"
     
     try:
         # Convert mask image to bytes