Update app.py

multi-image uploads

app.py

@@ -60,27 +60,34 @@ def resize_image(input_image, width, height):
     temp_file.close()
     return Image.open(temp_file.name)
 
-def process_image(input_image, enhance, scale, adjust_dpi, dpi, resize, width, height):
-    original_image = Image.fromarray(input_image.astype('uint8'), 'RGB')
+def process_images(input_images, enhance, scale, adjust_dpi, dpi, resize, width, height):
+    processed_images = []
+    file_paths = []
     
-    if enhance:
-        original_image = enhance_image(original_image, scale)
-    
-    if adjust_dpi:
-        original_image = muda_dpi(np.array(original_image), dpi)
+    for input_image in input_images:
+        original_image = Image.fromarray(input_image.astype('uint8'), 'RGB')
         
-    if resize:
-        original_image = resize_image(np.array(original_image), width, height)
+        if enhance:
+            original_image = enhance_image(original_image, scale)
         
-    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
-    original_image.save(temp_file.name)
-    return original_image, temp_file.name
+        if adjust_dpi:
+            original_image = muda_dpi(np.array(original_image), dpi)
+            
+        if resize:
+            original_image = resize_image(np.array(original_image), width, height)
+            
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
+        original_image.save(temp_file.name)
+        processed_images.append(original_image)
+        file_paths.append(temp_file.name)
+    
+    return processed_images, file_paths
 
 iface = gr.Interface(
-    fn=process_image,
+    fn=process_images,
     inputs=[
-        gr.Image(label="Upload"),
-        gr.Checkbox(label="Enhance Image (ESRGAN)"),
+        gr.Images(label="Upload Images"),  # Use gr.Images for multi-image upload
+        gr.Checkbox(label="Enhance Images (ESRGAN)"),
         gr.Radio(['2x', '4x', '8x'], type="value", value='2x', label='Resolution model'),
         gr.Checkbox(label="Adjust DPI"),
         gr.Number(label="DPI", value=300),
@@ -89,14 +96,15 @@ iface = gr.Interface(
         gr.Number(label="Height", value=512)
     ],
     outputs=[
-        gr.Image(label="Final Image"),
-        gr.File(label="Download Final Image")
+        gr.Gallery(label="Final Images"),  # Use gr.Gallery to display multiple images
+        gr.Files(label="Download Final Images")
     ],
-    title="Image Enhancer",
-    description="Upload an image (.jpg, .png), enhance using AI, adjust DPI, resize and download the final result.",
+    title="Multi-Image Enhancer",
+    description="Upload multiple images (.jpg, .png), enhance using AI, adjust DPI, resize, and download the final results.",
     examples=[
         ["gatuno.JPG"]
     ]
 )
 
 iface.launch(debug=True)
+