Update app.py

app.py

@@ -94,7 +94,10 @@ def process_images(image_files, enhance, scale, adjust_dpi, dpi, resize, width,
         for file_path in processed_images:
             zipf.write(file_path, os.path.basename(file_path))
     
-    return zip_path
+    # Load images for display in the gallery
+    display_images = [Image.open(img_path) for img_path in processed_images]
+    
+    return display_images, zip_path
 
 iface = gr.Interface(
     fn=process_images,
@@ -109,7 +112,8 @@ iface = gr.Interface(
         gr.Number(label="Height", value=512)
     ],
     outputs=[
-        gr.File(label="Download Final Images (ZIP)")  # Single file output for ZIP
+        gr.Gallery(label="Final Images"),  # Display the processed images
+        gr.File(label="Download Final Images (ZIP)")  # Provide a ZIP file for download
     ],
     title="Multi-Image Enhancer",
     description="Upload multiple images (.jpg, .png), enhance using AI, adjust DPI, resize, and download the final results as a ZIP file."
@@ -117,3 +121,4 @@ iface = gr.Interface(
 
 iface.launch(debug=True)
 
+