Update app.py

the output images into a ZIP file

app.py

@@ -1,3 +1,4 @@
+
 import torch
 from PIL import Image
 from RealESRGAN import RealESRGAN
@@ -5,12 +6,12 @@ import gradio as gr
 import numpy as np
 import tempfile
 import time
+import zipfile
 import os
 
-# Set device to GPU if available, otherwise use CPU
+# Set the device to CUDA if available, otherwise CPU
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
-# Load the Real-ESRGAN model with specified scale
 def load_model(scale):
     model = RealESRGAN(device, scale=scale)
     weights_path = f'weights/RealESRGAN_x{scale}.pth'
@@ -22,12 +23,11 @@ def load_model(scale):
         model.load_weights(weights_path, download=False)
     return model
 
-# Load different scales of the Real-ESRGAN model
+# Load models for different scales
 model2 = load_model(2)
 model4 = load_model(4)
 model8 = load_model(8)
 
-# Enhance the image using the specified scale
 def enhance_image(image, scale):
     try:
         print(f"Enhancing image with scale {scale}...")
@@ -49,7 +49,6 @@ def enhance_image(image, scale):
         print(f"Error enhancing image: {e}")
         return image
 
-# Adjust the DPI of the image
 def muda_dpi(input_image, dpi):
     dpi_tuple = (dpi, dpi)
     image = Image.fromarray(input_image.astype('uint8'), 'RGB')
@@ -58,7 +57,6 @@ def muda_dpi(input_image, dpi):
     temp_file.close()
     return Image.open(temp_file.name)
 
-# Resize the image to the specified width and height
 def resize_image(input_image, width, height):
     image = Image.fromarray(input_image.astype('uint8'), 'RGB')
     resized_image = image.resize((width, height))
@@ -67,12 +65,11 @@ def resize_image(input_image, width, height):
     temp_file.close()
     return Image.open(temp_file.name)
 
-# Process the images: enhance, adjust DPI, resize, and save
 def process_images(image_files, enhance, scale, adjust_dpi, dpi, resize, width, height):
     processed_images = []
-    file_paths = []
+    temp_dir = tempfile.mkdtemp()
     
-    for i, image_file in enumerate(image_files):
+    for image_file in image_files:
         input_image = np.array(Image.open(image_file).convert('RGB'))
         original_image = Image.fromarray(input_image.astype('uint8'), 'RGB')
         
@@ -84,24 +81,21 @@ def process_images(image_files, enhance, scale, adjust_dpi, dpi, resize, width,
             
         if resize:
             original_image = resize_image(np.array(original_image), width, height)
-            
-        # Create a custom filename
-        custom_filename = f"Image_Captioning_with_BLIP_{i+1}.jpg"
-        
-        # Save the image with the custom filename
-        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.jpg')
-        original_image.save(temp_file.name, format='JPEG')
         
-        # Rename the file with the custom name
-        final_path = temp_file.name.replace(temp_file.name.split('/')[-1], custom_filename)
-        os.rename(temp_file.name, final_path)
-        
-        processed_images.append(original_image)
-        file_paths.append(final_path)
+        # Save each image as JPEG
+        file_name = os.path.basename(image_file.name)
+        output_path = os.path.join(temp_dir, f"processed_{file_name}")
+        original_image.save(output_path, format='JPEG')
+        processed_images.append(output_path)
+    
+    # Create a ZIP file with all processed images
+    zip_path = os.path.join(temp_dir, 'processed_images.zip')
+    with zipfile.ZipFile(zip_path, 'w') as zipf:
+        for file_path in processed_images:
+            zipf.write(file_path, os.path.basename(file_path))
     
-    return processed_images, file_paths
+    return zip_path
 
-# Gradio interface setup
 iface = gr.Interface(
     fn=process_images,
     inputs=[
@@ -115,13 +109,11 @@ iface = gr.Interface(
         gr.Number(label="Height", value=512)
     ],
     outputs=[
-        gr.Gallery(label="Final Images"),  # Use gr.Gallery to display multiple images
-        gr.Files(label="Download Final Images")
+        gr.File(label="Download Final Images (ZIP)")  # Single file output for ZIP
     ],
     title="Multi-Image Enhancer",
-    description="Upload multiple images (.jpg, .png), enhance using AI, adjust DPI, resize, and download the final results."
+    description="Upload multiple images (.jpg, .png), enhance using AI, adjust DPI, resize, and download the final results as a ZIP file."
 )
 
-# Launch the Gradio interface
 iface.launch(debug=True)