Update app.py

app.py

@@ -1,18 +1,17 @@
 import streamlit as st
 import os
-from PIL import Image, ImageDraw
+from PIL import Image
 import random
+from io import BytesIO
 
 # Function to arrange images dynamically
-def arrange_images(image_files, output_path="dungeon_layout.png"):
+def arrange_images(image_files, canvas_size=(3000, 3000)):
     if not image_files:
         return None
 
     # Initialize layout
     positions = []  # Keeps track of image positions (x1, y1, x2, y2)
-    canvas_size = (3000, 3000)  # Large canvas for positioning
     canvas = Image.new("RGBA", canvas_size, "white")
-    draw = ImageDraw.Draw(canvas)
 
     def get_center(pos):
         """Calculate center of a bounding box (x1, y1, x2, y2)."""
@@ -79,12 +78,11 @@ def arrange_images(image_files, output_path="dungeon_layout.png"):
                     canvas.paste(img, (x1, y1))
                     placed = True
 
-                    # Draw a connecting line
-                    new_center = get_center((x1, y1, x2, y2))
-                    draw.line([target_center, new_center], fill="black", width=5)
-
-    canvas.save(output_path)
-    return output_path
+    # Save canvas to memory
+    buffer = BytesIO()
+    canvas.save(buffer, format="PNG")
+    buffer.seek(0)
+    return buffer
 
 # Streamlit App
 st.title("Dynamic Dungeon Map Generator")
@@ -93,15 +91,38 @@ st.write("Automatically generates dungeon maps by arranging PNG images dynamical
 # Directory for images
 map_dir = "."  # Top-level directory
 
+# Canvas size and scroll offsets
+canvas_size = 3000
+scroll_offset = {"x": 0, "y": 0}
+
+# WASD Controls for scrolling
+st.sidebar.title("Scroll Map")
+scroll_buttons = st.sidebar.columns(3)
+if scroll_buttons[0].button("⬅️"):
+    scroll_offset["x"] -= 100
+if scroll_buttons[1].button("⬆️"):
+    scroll_offset["y"] -= 100
+if scroll_buttons[2].button("➡️"):
+    scroll_offset["x"] += 100
+if st.sidebar.button("⬇️"):
+    scroll_offset["y"] += 100
+
 # Scan the directory for .png files
 image_files = [f for f in os.listdir(map_dir) if f.endswith(".png")]
 
 if image_files:
     # Add "Create Map" button
     if st.button("🗺️ Create Map"):
-        layout_image = arrange_images(image_files)
+        scroll_offset = {"x": 0, "y": 0}  # Reset scroll offset
+        layout_image = arrange_images(image_files, canvas_size=(canvas_size, canvas_size))
         if layout_image:
-            st.image(layout_image, caption="Generated Dungeon Map Layout")
+            st.image(
+                layout_image,
+                caption="Generated Dungeon Map Layout",
+                use_column_width=True,
+                output_format="PNG",
+                clamp=True
+            )
         else:
             st.write("Failed to generate a map. Please check the images in the directory.")
 else: