Update app.py

app.py

@@ -35,10 +35,9 @@ 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):
-    """Gửi request và nhận kết quả từ Gemini API (Xử lý lỗi 429 với popup thông báo)"""
-    attempt = 0  # Đếm số lần thử lại
-    while attempt < max_retries:
+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):
         try:
             image_parts = [types.Part(inline_data=types.Blob(data=img, mime_type="image/jpeg")) for img in image_bytes_list if img]
             
@@ -55,9 +54,14 @@ async def generate_image(image_bytes_list, text_input, max_retries=5):
             )
 
             if not response or not response.candidates or not response.candidates[0].content:
-                attempt += 1
-                gr.Warning(f"⚠️ API trả về dữ liệu không hợp lệ (thử {attempt}/{max_retries}). Đang thử lại...")
-                continue  # Thử lại ngay lập tức
+                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 []
 
             images = []
             for part in response.candidates[0].content.parts:
@@ -66,37 +70,31 @@ async def generate_image(image_bytes_list, text_input, max_retries=5):
                         img = Image.open(BytesIO(part.inline_data.data))
                         images.append(img)
                     except Exception as e:
-                        gr.Error(f"❌ Lỗi khi hiển thị ảnh: {e}")
-            return images  # Thành công, trả về ảnh ngay lập tức
+                        print(f"❌ Lỗi khi hiển thị ảnh: {e}")
+            return images
 
         except Exception as e:
             error_message = str(e)
-            if "RESOURCE_EXHAUSTED" in error_message:
+            if "429" in error_message and "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_delay = int(error_json["error"]["details"][-1]["retryDelay"][:-1])  # Lấy số giây từ '2s'
+                    retry_seconds = int(error_json["error"]["details"][-1]["retryDelay"][:-1])  # Lấy số giây từ '2s'
 
-                    gr.Warning(f"⚠️ API quá tải! Sẽ thử lại sau {retry_delay} giây...")
-                    
-                    # Đếm ngược thời gian chờ (có popup)
-                    for i in range(retry_delay, 0, -1):
-                        gr.Info(f"⏳ Thử lại sau {i} giây...")
+                    print(f"⚠️ Quá tải API! Chờ {retry_seconds} giây trước khi thử lại...")
+
+                    for i in range(retry_seconds, 0, -1):
+                        print(f"⏳ Thử lại sau {i} giây...", end="\r")
                         time.sleep(1)
 
-                    gr.Info("🔄 Đang thử lại...")
-                    attempt += 1
-                    continue  # Thử lại sau thời gian chờ
+                    print("\n🔄 Đang thử lại...")
+                    continue  
 
                 except Exception as parse_error:
-                    gr.Error(f"❌ Lỗi khi phân tích retryDelay: {parse_error}")
+                    print(f"❌ Lỗi khi phân tích retryDelay: {parse_error}")
 
-            gr.Error(f"❌ Lỗi khi gọi API Gemini: {e}")
+            print(f"❌ Lỗi khi gọi API Gemini: {e}")
             return []
 
-    gr.Error("❌ Đã 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]