Update app.py

app.py

@@ -37,7 +37,7 @@ def load_image_as_bytes(image_path):
         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à phản hồi không hợp lệ)"""
+    """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:
         try:
@@ -57,7 +57,7 @@ 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
-                print(f"⚠️ Phản hồi API không hợp lệ (thử {attempt}/{max_retries})")
+                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
 
             images = []
@@ -67,7 +67,7 @@ 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:
-                        print(f"❌ Lỗi khi hiển thị ảnh: {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
 
         except Exception as e:
@@ -78,24 +78,24 @@ async def generate_image(image_bytes_list, text_input, max_retries=5):
                     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'
 
-                    print(f"⚠️ API quá tải! Chờ {retry_delay} giây trước khi thử lại...")
-
-                    # Đếm ngược thời gian chờ
+                    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):
-                        print(f"⏳ Thử lại sau {i} giây...", end="\r")
+                        gr.Info(f"⏳ Thử lại sau {i} giây...")
                         time.sleep(1)
 
-                    print("\n🔄 Đang thử lại...")
+                    gr.Info("🔄 Đang thử lại...")
                     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}")
+                    gr.Error(f"❌ Lỗi khi phân tích retryDelay: {parse_error}")
 
-            print(f"❌ Lỗi khi gọi API Gemini: {e}")
+            gr.Error(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!")
+    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):