Update app.py

app.py

@@ -24,7 +24,12 @@ def process_images(images, remove_bg):
         
         # Update position for next image (for simplicity stacking vertically)
         y_offset += img.height
-    
+        if y_offset + img.height > final_collage.height:
+            y_offset = 0
+            x_offset += img.width
+            if x_offset + img.width > final_collage.width:
+                break  # Stop adding images if there's no more space
+
     return final_collage
 
 # Gradio interface
@@ -32,7 +37,7 @@ with gr.Blocks() as demo:
     gr.Markdown("## Sticker Collage Creator (A4)")
     
     with gr.Row():
-        images = gr.File(label="Upload your images", file_types=["image"], multiple=True)
+        images = gr.Files(label="Upload your images", file_types=["image"])  # Use gr.Files for multiple image uploads
         remove_bg = gr.Checkbox(label="Remove background?", value=True)
     
     output = gr.Image(label="Collage", type="pil")