Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from PIL import Image
-import io
+import os
 
 # Function to process and display the images in grid format
 def create_collage(uploaded_files, grid_layout):
@@ -36,11 +36,12 @@ def create_collage(uploaded_files, grid_layout):
         col = i % cols
         collage.paste(img, (col * img.width, row * img.height))
 
-    # Save the collage as a PNG file in memory (using BytesIO)
-    output_io = io.BytesIO()
-    collage.save(output_io, format="PNG")
-    output_io.seek(0)  # Reset the pointer to the start of the file
-    return output_io  # Return the BytesIO object for download
+    # Save the collage as a PNG file in a temporary directory
+    output_path = "temp_collage.png"
+    collage.save(output_path, format="PNG")
+
+    # Return the file path for download
+    return output_path
 
 # Gradio Interface
 with gr.Blocks() as app: