Update app.py

app.py

@@ -83,23 +83,27 @@ async def process_request(images, text, num_requests):
     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
+    return results + resized_images
+    # return results + srcnn_images  # Ảnh gốc + ảnh đã qua SRCNN
 
-def gradio_interface(images, text, num_requests):
+def gradio_interface(image1, image2, image3, text, num_requests):
     """Hàm Gradio xử lý yêu cầu và trả về ảnh"""
+    images = [img for img in [image1, image2, image3] if img]
     return asyncio.run(process_request(images, text, num_requests))
 
 # Tạo giao diện Gradio
 demo = gr.Interface(
     fn=gradio_interface,
     inputs=[
-        gr.File(file_types=["image"], label="Upload hình ảnh"),
+        gr.Image(type='filepath', label="Upload hình ảnh 1"),
+        gr.Image(type='filepath', label="Upload hình ảnh 2"),
+        gr.Image(type='filepath', label="Upload hình ảnh 3"),
         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 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.",
+    description="Upload tối đa 3 ả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()