From 7c906a3ff0f42c1bc315faf738173070da8179bc Mon Sep 17 00:00:00 2001
From: Misha4ca228 <tvvas1977@gmail.com>
Date: Wed, 30 Apr 2025 21:53:01 +0300
Subject: [PATCH] dev

---
 modules/api/api.py | 32 ++++++++++++++++++++++++--------
 1 file changed, 24 insertions(+), 8 deletions(-)

diff --git a/modules/api/api.py b/modules/api/api.py
index ed0ae4046..05ea214bb 100644
--- a/modules/api/api.py
+++ b/modules/api/api.py
@@ -577,15 +577,31 @@ class Api:
         filename = str(uid_int)[0:18]
         folder = "ddcn_results"
 
-        s3_client.put_object(
-            Bucket=get_env_var(key="S3_BUCKET"),
-            Key=f"{folder}/{filename}.jpg",
-            Body=image_bytes,
-            ContentType="image/jpeg"
-        )
-        s3_url = f"{get_env_var('S3_ENDPOINT')}/{get_env_var('S3_BUCKET')}/{folder}/{filename}.jpg"
+        retry = 0
+
+        while retry <= 5:
+
+            try:
+                s3_client.put_object(
+                    Bucket=get_env_var(key="S3_BUCKET"),
+                    Key=f"{folder}/{filename}.jpg",
+                    Body=image_bytes,
+                    ContentType="image/jpeg"
+                )
+                s3_url = f"{get_env_var('S3_ENDPOINT')}/{get_env_var('S3_BUCKET')}/{folder}/{filename}.jpg"
+                return models.ImageToImageResponse(parameters=vars(img2imgreq), info=processed.js(), s3_url=s3_url)
+
+            except Exception as e:
+                print(f"Error save on s3: {e}")
+                retry =+ 1
+                continue
+
+        else:
+            raise HTTPException(status_code=500, detail="Error save on s3:")
+
+
+
 
-        return models.ImageToImageResponse(parameters=vars(img2imgreq), info=processed.js(), s3_url=s3_url)
 
     def extras_single_image_api(self, req: models.ExtrasSingleImageRequest):
         reqDict = setUpscalers(req)