Update app.py

app.py

@@ -60,26 +60,33 @@ 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_image(input_images, enhance, scale, adjust_dpi, dpi, resize, width, height):
+    processed_images = []
+    temp_files = []
     
-    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)
+        temp_files.append(temp_file.name)
+    
+    return processed_images, temp_files
 
 iface = gr.Interface(
     fn=process_image,
     inputs=[
-        gr.Image(label="Upload"),
+        gr.File(label="Upload", type="file", file_count="multiple", optional=False),
         gr.Checkbox(label="Enhance Image (ESRGAN)"),
         gr.Radio(['2x', '4x', '8x'], type="value", value='2x', label='Resolution model'),
         gr.Checkbox(label="Adjust DPI"),
@@ -89,11 +96,11 @@ 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"),
+        gr.File(label="Download Final Images", file_count="multiple")
     ],
     title="Image Enhancer",
-    description="Upload an image (.jpg, .png), enhance using AI, adjust DPI, resize and download the final result.",
+    description="Upload images (.jpg, .png), enhance using AI, adjust DPI, resize and download the final results.",
     examples=[
         ["gatuno.JPG"]
     ]