Update app.py

app.py

@@ -67,10 +67,14 @@ async def generate_image(image_bytes, text_input):
     )
     
     images = []
-    for part in response.candidates[0].content.parts:
-        if part.inline_data is not None:
-            img = Image.open(BytesIO(part.inline_data.data))
-            images.append(img)
+    if response and response.candidates and response.candidates[0].content:
+        for part in response.candidates[0].content.parts:
+            if part.inline_data is not None:
+                img = Image.open(BytesIO(part.inline_data.data))
+                images.append(img)
+    else:
+        print("⚠️ Gemini API không trả về kết quả hợp lệ!")
+    
     return images
 
 async def process_request(image, text, num_requests):
@@ -81,7 +85,7 @@ async def process_request(image, text, num_requests):
     
     # Hợp nhất danh sách ảnh từ các request
     original_images = [img for result in results for img in result]
-    resized_images = [img.resize((2560, 1440), Image.ANTIALIAS) for img in original_images]  # Resize trước khi upscale
+    resized_images = [img.resize((2560, 1440), Image.LANCZOS) for img in original_images]  # Resize trước khi upscale
     srcnn_images = [upscale_image(img) for img in resized_images]
     
     return original_images + srcnn_images  # 4 ảnh gốc + 4 ảnh đã qua SRCNN