Update app.py

app.py

@@ -4,7 +4,6 @@ from PIL import Image
 import random
 from io import BytesIO
 import datetime
-import base64
 
 # Adjust Streamlit layout to wide mode
 st.set_page_config(layout="wide")
@@ -12,10 +11,11 @@ st.set_page_config(layout="wide")
 # Function to arrange images dynamically
 def arrange_images(image_files, canvas_size=(3000, 3000)):
     if not image_files:
-        return None
+        return None, []
 
     positions = []  # Keeps track of image positions (x1, y1, x2, y2)
     canvas = Image.new("RGBA", canvas_size, "white")
+    room_details = []  # To store room layout details
 
     def get_center(pos):
         """Calculate center of a bounding box (x1, y1, x2, y2)."""
@@ -42,9 +42,10 @@ def arrange_images(image_files, canvas_size=(3000, 3000)):
         x2, y2 = x1 + width, y1 + height
         positions.append((x1, y1, x2, y2))
         canvas.paste(img, (x1, y1))
+        room_details.append(f"Room 1: {image_files[0]} at center")
 
     # Place remaining images
-    for img_file in image_files[1:]:
+    for idx, img_file in enumerate(image_files[1:], start=2):
         placed = False
         img_path = os.path.join(map_dir, img_file)
 
@@ -74,21 +75,16 @@ def arrange_images(image_files, canvas_size=(3000, 3000)):
                 if not does_overlap((x1, y1, x2, y2), positions):
                     positions.append((x1, y1, x2, y2))
                     canvas.paste(img, (x1, y1))
+                    room_details.append(f"Room {idx}: {img_file} at ({x1}, {y1})")
                     placed = True
 
     buffer = BytesIO()
     canvas.save(buffer, format="PNG")
     buffer.seek(0)
-    return buffer, canvas
+    return buffer, canvas, room_details
 
-# Function to create a base64 download link
-def create_download_link(image_bytes, filename):
-    b64 = base64.b64encode(image_bytes.getvalue()).decode()
-    href = f'<a href="data:image/png;base64,{b64}" download="{filename}">📥 Download {filename}</a>'
-    return href
-
-# Streamlit App
-st.title("Dynamic Dungeon Map Generator")
+# Sidebar Title
+st.sidebar.markdown("#### 🏰 Dynamic Dungeon Map Generator")
 
 # Directory for images
 map_dir = "."
@@ -99,29 +95,38 @@ if "layout_image" not in st.session_state:
     st.session_state["layout_image"] = None
 if "canvas" not in st.session_state:
     st.session_state["canvas"] = None
+if "room_details" not in st.session_state:
+    st.session_state["room_details"] = []
 
 # Generate map if layout_image is empty
 if st.session_state["layout_image"] is None:
     image_files = [f for f in os.listdir(map_dir) if f.endswith(".png")]
     if image_files:
-        layout_image, canvas = arrange_images(image_files, canvas_size=(canvas_size, canvas_size))
+        layout_image, canvas, room_details = arrange_images(image_files, canvas_size=(canvas_size, canvas_size))
         st.session_state["layout_image"] = layout_image
         st.session_state["canvas"] = canvas
+        st.session_state["room_details"] = room_details
 
 # Sidebar Controls
-st.sidebar.title("Map Controls")
 if st.sidebar.button("💾 Save Map"):
     now = datetime.datetime.now()
     filename = f"dungeon_map_{now.strftime('%Y%m%d_%H%M%S')}.png"
     st.session_state["canvas"].save(filename)
     st.sidebar.success(f"Map saved as {filename}")
 
+    # Save room details as text
+    layout_filename = f"room_layout_{now.strftime('%Y%m%d_%H%M%S')}.txt"
+    with open(layout_filename, "w") as f:
+        f.write("\n".join(st.session_state["room_details"]))
+    st.sidebar.success(f"Room layout saved as {layout_filename}")
+
 if st.sidebar.button("🗺️ Regenerate Map"):
     image_files = [f for f in os.listdir(map_dir) if f.endswith(".png")]
     if image_files:
-        layout_image, canvas = arrange_images(image_files, canvas_size=(canvas_size, canvas_size))
+        layout_image, canvas, room_details = arrange_images(image_files, canvas_size=(canvas_size, canvas_size))
         st.session_state["layout_image"] = layout_image
         st.session_state["canvas"] = canvas
+        st.session_state["room_details"] = room_details
         st.rerun()
 
 # Zoom Controls