Add app

app.py

@@ -69,26 +69,45 @@ def concat_images(images, patch_size=256):
     full_img = np.vstack(grid)
     return Image.fromarray(full_img)
 
-def add_to_gallery(image, gallery):
-    if image is None:
-        return gallery
+def process_images(files, gallery, patch_size):
+    if not files:
+        return gallery, None
     
-    # Add the new image to the gallery
+    # Add all new images to the gallery
     if gallery is None:
         gallery = []
-    gallery.append((image, None))
-    return gallery
+    
+    for file in files:
+        try:
+            if not file.name.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp', '.webp')):
+                print(f"Skipping invalid file type: {file.name}")
+                continue
+                
+            img = Image.open(file.name)
+            gallery.append((img, None))
+        except Exception as e:
+            print(f"Error loading image {file.name}: {e}")
+    
+    # Generate the concatenated result
+    result = concat_images(gallery, patch_size)
+    
+    return gallery, result
 
 def clear_gallery():
-    return None
+    return None, None
 
 with gr.Blocks() as demo:
     gr.Markdown("# Image Concatenation")
-    gr.Markdown("Upload images to create a grid of concatenated images.")
+    gr.Markdown("Upload multiple images to create a grid of concatenated images.")
     
     with gr.Row():
         with gr.Column():
-            image_input = gr.Image(type="pil", label="Upload Image", show_label=True, height=400)
+            file_input = gr.File(
+                label="Upload Images",
+                file_count="multiple",
+                file_types=[".png", ".jpg", ".jpeg", ".gif", ".bmp", ".webp"],
+                height=400
+            )
             clear_button = gr.Button("Clear All Images")
         
         with gr.Column():
@@ -113,30 +132,22 @@ with gr.Blocks() as demo:
         label="Patch Size"
     )
     
-    show_result_button = gr.Button("Show Concatenated Result")
-    
     # Set up the event handlers
-    image_input.upload(
-        fn=add_to_gallery,
-        inputs=[image_input, image_gallery],
-        outputs=image_gallery
-    )
-    
-    show_result_button.click(
-        fn=concat_images,
-        inputs=[image_gallery, patch_size],
-        outputs=image_output
+    file_input.upload(
+        fn=process_images,
+        inputs=[file_input, image_gallery, patch_size],
+        outputs=[image_gallery, image_output]
     )
     
     patch_size.change(
-        fn=concat_images,
-        inputs=[image_gallery, patch_size],
-        outputs=image_output
+        fn=process_images,
+        inputs=[file_input, image_gallery, patch_size],
+        outputs=[image_gallery, image_output]
     )
     
     clear_button.click(
         fn=clear_gallery,
-        outputs=image_gallery
+        outputs=[image_gallery, image_output]
     )
 
-demo.launch()
+demo.launch()