Update app.py

app.py

@@ -57,7 +57,8 @@ def load_image_as_bytes(image_path):
 async def generate_image(image_bytes_list, text_input):
     """Gửi request và nhận kết quả từ Gemini API"""
     image_parts = [types.Part(inline_data=types.Blob(data=img, mime_type="image/jpeg")) for img in image_bytes_list if img]
-    contents = [text_input, image_parts] if image_parts else [text_input]
+    contents = [text_input, image_parts] if image_parts else
+ [text_input]
     
     response = await asyncio.to_thread(
         client.models.generate_content,
@@ -74,35 +75,32 @@ async def generate_image(image_bytes_list, text_input):
                 images.append(img)
     return images
 
-async def process_request(image, text, num_requests):
+async def process_request(images, text, num_requests):
     """Chạy nhiều request song song"""
-    image_bytes = load_image_as_bytes(image) if image else None
-    tasks = [generate_image(image_bytes, text) for _ in range(num_requests)]
-    results = await asyncio.gather(*tasks)
+    image_bytes_list = [load_image_as_bytes(image) if image else None for image in images]
+    results = await generate_image(image_bytes_list, text)
     
     # 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, int(img.height * (2560 / img.width)))), Image.LANCZOS) for img in original_images]  # Resize trước khi upscale
+    resized_images = [img.resize((2560, int(img.height * (2560 / img.width))), Image.LANCZOS) for img in results]  # Resize giữ nguyên tỉ lệ chiều cao
     # srcnn_images = [upscale_image(img) for img in resized_images]
     
-    return resized_images  # 4 ảnh gốc
-    # return resized_images + srcnn_images  # 4 ảnh gốc + 4 ảnh đã qua SRCNN
+    return resized_images
 
-def gradio_interface(image, text, num_requests):
+def gradio_interface(images, text, num_requests):
     """Hàm Gradio xử lý yêu cầu và trả về ảnh"""
-    return asyncio.run(process_request(image, text, num_requests))
+    return asyncio.run(process_request(images, text, num_requests))
 
 # Tạo giao diện Gradio
 demo = gr.Interface(
     fn=gradio_interface,
     inputs=[
-        gr.Image(type='filepath', label="Upload hình ảnh"),
+        gr.File(file_types=["image"], label="Upload hình ảnh", multiple=True),
         gr.Textbox(label="Nhập yêu cầu chỉnh sửa hình ảnh"),
         gr.Slider(minimum=1, maximum=4, step=1, value=4, label="Số lượng ảnh cần tạo")
     ],
     outputs=gr.Gallery(label="Kết quả chỉnh sửa", columns=4),
     title="Chỉnh sửa ảnh bằng Gemini AI + SRCNN",
-    description="Upload ảnh và nhập yêu cầu chỉnh sửa. Hiển thị 4 ảnh gốc từ API và 4 ảnh đã qua SRCNN.",
+    description="Upload nhiều ảnh và nhập yêu cầu chỉnh sửa. Hiển thị ảnh gốc từ API và ảnh đã qua SRCNN.",
 )
 
 demo.launch()