Update app.py

app.py

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