Update app.py

app.py

@@ -1,23 +1,32 @@
 import gradio as gr
 from PIL import Image
 import io
+import os
 
-def compress_image(image_path):
-    img = Image.open(image_path)
-    compressed_io = io.BytesIO()
-    img.save(compressed_io, format='JPEG', quality=30)
-    compressed_io.seek(0)
-    compressed_filename = "compressed_image.jpg"
-    with open(compressed_filename, "wb") as f:
-        f.write(compressed_io.read())
-    return compressed_filename
+def compress_images(*image_paths):
+    compressed_files = []
+    for image_path in image_paths:
+        if image_path is not None:
+            img = Image.open(image_path)
+            compressed_io = io.BytesIO()
+            img.save(compressed_io, format='JPEG', quality=30)
+            compressed_io.seek(0)
+            compressed_filename = f"compressed_{os.path.basename(image_path)}"
+            with open(compressed_filename, "wb") as f:
+                f.write(compressed_io.read())
+            compressed_files.append(compressed_filename)
+        else:
+            compressed_files.append(None)
+    return compressed_files
+
+image_inputs = [gr.Image(type="filepath", label=f"Upload Image {i+1}") for i in range(5)]
 
 interface = gr.Interface(
-    fn=compress_image,
-    inputs=gr.Image(type="filepath", label="Upload Image"),
-    outputs=gr.File(label="Download Compressed Image"),
-    title="Image Compressor",
-    description="Upload an image to compress it and download the compressed version."
+    fn=compress_images,
+    inputs=image_inputs,
+    outputs=[gr.File(label=f"Download Compressed Image {i+1}") for i in range(5)],
+    title="Multi-Image Compressor",
+    description="Upload up to 5 images to compress them and download the compressed versions."
 )
 
-interface.launch()
+interface.launch()