Update app.py

app.py

@@ -26,7 +26,8 @@ def load_image_as_bytes(image_path):
             img.save(img_bytes, format="JPEG")  # Lưu ảnh vào buffer
             return img_bytes.getvalue()  # Lấy dữ liệu nhị phân
     except FileNotFoundError:
-        print(f"❌ Lỗi: Không tìm thấy file {image_path}")
+        print(f"❌ Lỗi: Không tìm thấy file 
+{image_path}")
         return None
     except UnidentifiedImageError:
         print(f"❌ Lỗi: Không thể mở file {image_path} (định dạng không hợp lệ)")
@@ -36,9 +37,10 @@ def load_image_as_bytes(image_path):
         print(f"❌ Lỗi khi đọc ảnh {image_path}: {e}")
         return None
 
-async def generate_image(image_bytes_list, text_input, max_retries=5, retry_delay=2):
-    """Gửi request và nhận kết quả từ Gemini API (Xử lý lỗi 429 và lỗi phản hồi không hợp lệ)"""
-    for attempt in range(1, max_retries + 1):
+async def generate_image(image_bytes_list, text_input, max_retries=5):
+    """Gửi request và nhận kết quả từ Gemini API (Xử lý lỗi 429 và phản hồi không hợp lệ)"""
+    attempt = 0  # Đếm số lần thử lại
+    while attempt < max_retries:
         try:
             image_parts = [types.Part(inline_data=types.Blob(data=img, mime_type="image/jpeg")) for img in image_bytes_list if img]
             
@@ -55,14 +57,9 @@ async def generate_image(image_bytes_list, text_input, max_retries=5, retry_dela
             )
 
             if not response or not response.candidates or not response.candidates[0].content:
+                attempt += 1
                 print(f"⚠️ Phản hồi API không hợp lệ (thử {attempt}/{max_retries})")
-                if attempt < max_retries:
-                    print(f"🔄 Đang thử lại sau {retry_delay} giây...")
-                    time.sleep(retry_delay)
-                    continue
-                else:
-                    print("❌ Đã thử lại nhiều lần nhưng vẫn thất bại!")
-                    return []
+                continue  # Thử lại ngay lập tức
 
             images = []
             for part in response.candidates[0].content.parts:
@@ -72,23 +69,26 @@ async def generate_image(image_bytes_list, text_input, max_retries=5, retry_dela
                         images.append(img)
                     except Exception as e:
                         print(f"❌ Lỗi khi hiển thị ảnh: {e}")
-            return images
+            return images  # Thành công, trả về ảnh ngay lập tức
 
         except Exception as e:
             error_message = str(e)
-            if "429" in error_message and "RESOURCE_EXHAUSTED" in error_message:
+            if "RESOURCE_EXHAUSTED" in error_message:
                 try:
+                    # Trích xuất retryDelay từ JSON lỗi
                     error_json = json.loads(error_message.split("RESOURCE_EXHAUSTED. ")[1])
-                    retry_seconds = int(error_json["error"]["details"][-1]["retryDelay"][:-1])  # Lấy số giây từ '2s'
+                    retry_delay = int(error_json["error"]["details"][-1]["retryDelay"][:-1])  # Lấy số giây từ '2s'
 
-                    print(f"⚠️ Quá tải API! Chờ {retry_seconds} giây trước khi thử lại...")
+                    print(f"⚠️ API quá tải! Chờ {retry_delay} giây trước khi thử lại...")
 
-                    for i in range(retry_seconds, 0, -1):
+                    # Đếm ngược thời gian chờ
+                    for i in range(retry_delay, 0, -1):
                         print(f"⏳ Thử lại sau {i} giây...", end="\r")
                         time.sleep(1)
 
                     print("\n🔄 Đang thử lại...")
-                    continue  
+                    attempt += 1
+                    continue  # Thử lại sau thời gian chờ
 
                 except Exception as parse_error:
                     print(f"❌ Lỗi khi phân tích retryDelay: {parse_error}")
@@ -96,6 +96,9 @@ async def generate_image(image_bytes_list, text_input, max_retries=5, retry_dela
             print(f"❌ Lỗi khi gọi API Gemini: {e}")
             return []
 
+    print("❌ Đã thử lại nhiều lần nhưng vẫn thất bại!")
+    return []
+
 async def process_request(images, text, num_requests):
     """Chạy nhiều request song song"""
     image_bytes_list = [load_image_as_bytes(image) if image else None for image in images]