bug fix

app.py

@@ -19,6 +19,7 @@ from diffusers.utils import load_image
 from gradio_imageslider import ImageSlider
 import boto3
 from io import BytesIO
+from datetime import datetime
 
 device = "cuda"
 base_model_id = "SG161222/RealVisXL_V4.0"
@@ -84,8 +85,8 @@ def get_depth_map(image):
     return image
 
 
-def upload_to_s3(image, account_id, access_key, secret_key, bucket_name):
-
+def upload_image_to_s3(image, account_id, access_key, secret_key, bucket_name):
+    print("upload_image_to_s3", account_id, access_key, secret_key, bucket_name)
     connectionUrl = f"https://{account_id}.r2.cloudflarestorage.com"
 
     s3 = boto3.client(
@@ -95,7 +96,9 @@ def upload_to_s3(image, account_id, access_key, secret_key, bucket_name):
         aws_access_key_id=access_key,
         aws_secret_access_key=secret_key
     )
-    image_file = f"generated_images/{random.randint(0, MAX_SEED)}.png"
+
+    current_time = datetime.now().strftime("%Y%m%d_%H%M%S")
+    image_file = f"generated_images/{current_time}_{random.randint(0, MAX_SEED)}.png"
     buffer = BytesIO()
     image.save(buffer, "PNG")
     buffer.seek(0)
@@ -131,7 +134,7 @@ def process(image, image_url, prompt, n_prompt, num_steps, guidance_scale, contr
     ).images[0]
 
     if upload_to_s3:
-        url = upload_to_s3(generated_image, account_id, access_key, secret_key, bucket)
+        url = upload_image_to_s3(generated_image, account_id, access_key, secret_key, bucket)
         result = {"status": "success", "url": url}
     else:
         result = {"status": "success", "message": "Image generated but not uploaded"}