Update app.py

app.py

@@ -8,6 +8,8 @@ import os
 from PIL import Image
 import torchvision.transforms as transforms
 
+photos_folder = "Photos"
+
 # Pobierz model i config
 repo_id = "Kiwinicki/sat2map-generator"
 generator_path = hf_hub_download(repo_id=repo_id, filename="generator.pth")
@@ -51,11 +53,72 @@ def process_image(image):
     
     return output_image
 
-iface = gr.Interface(
-    fn=process_image,
-    inputs=gr.Image(type="pil"),
-    outputs="image",
-    title="Satellite to Map Generator"
-)
 
-iface.launch()
+def load_images_from_folder(folder):
+    images = []
+    for filename in os.listdir(folder):
+        if filename.lower().endswith(('.png', '.jpg', '.jpeg')):
+            img_path = os.path.join(folder, filename)
+            img = Image.open(img_path)
+            images.append((img, filename))
+    return images
+
+
+
+def load_image_from_gallery(images, index):
+    if images and 0 <= index < len(images):
+        image = images[index]
+        if isinstance(image, tuple):
+            image = image[0]
+        return image
+    return None
+
+
+def gallery_click_event(images, evt: gr.SelectData):
+    index = evt.index
+    selected_img = load_image_from_gallery(images, index)
+    return selected_img
+
+
+def clear_image():
+    return None
+
+
+def app():
+    images = load_images_from_folder(photos_folder)
+
+    with gr.Blocks(css=".container { background-color: white; }") as demo:
+        with gr.Row():
+            with gr.Column():
+                selected_image = gr.Image(label="Input Image", type="pil")
+                clear_button = gr.Button("Clear")
+
+            with gr.Column():
+                image_gallery = gr.Gallery(label="Image Gallery", elem_id="gallery", type="pil", value=[img for img, _ in images])
+
+            with gr.Column():
+                result_image = gr.Image(label="Result Image", type="pil")
+
+        image_gallery.select(
+            fn=gallery_click_event,
+            inputs=image_gallery,
+            outputs=selected_image
+        )
+
+        selected_image.change(
+            fn=process_image,
+            inputs=selected_image,
+            outputs=result_image
+        )
+
+        clear_button.click(
+            fn=clear_image,
+            inputs=None,
+            outputs=selected_image
+        )
+
+    demo.launch()
+
+
+if __name__ == "__main__":
+    app()