Update app.py

app.py

@@ -13,13 +13,22 @@ def generate_map(image_files, canvas_size=(3000, 3000)):
     if not image_files:
         return canvas
 
-    for i, img_file in enumerate(image_files):
+    grid_size = int(len(image_files) ** 0.5) + 1  # Calculate grid size
+    img_size = canvas_size[0] // grid_size       # Size for each image cell
+
+    for idx, img_file in enumerate(image_files):
         img_path = os.path.join(map_dir, img_file)
         with Image.open(img_path) as img:
-            img = img.resize((500, 500))  # Resize for demonstration
-            x = (i % 5) * 600  # Example placement logic
-            y = (i // 5) * 600
-            canvas.paste(img, (x, y))
+            # Maintain aspect ratio
+            img.thumbnail((img_size, img_size), Image.ANTIALIAS)
+
+            # Calculate grid position
+            row, col = divmod(idx, grid_size)
+            x = col * img_size
+            y = row * img_size
+
+            # Paste the image on the canvas
+            canvas.paste(img, (x, y), mask=img if img.mode == "RGBA" else None)
 
     return canvas