Update app.py

app.py

@@ -4,6 +4,7 @@ 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")
@@ -83,6 +84,11 @@ def arrange_images(image_files, canvas_size=(3000, 3000)):
     buffer.seek(0)
     return buffer, canvas, room_details
 
+# Function to create a base64 link
+def create_base64_download_link(file_content, filename, file_type):
+    b64 = base64.b64encode(file_content.encode() if file_type == "txt" else file_content).decode()
+    return f"[📥 Download {filename}](data:application/{file_type};base64,{b64})"
+
 # Sidebar Title
 st.sidebar.markdown("#### 🏰 Dynamic Dungeon Map Generator")
 
@@ -97,6 +103,8 @@ 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"] = []
+if "saved_files" not in st.session_state:
+    st.session_state["saved_files"] = []
 
 # Generate map if layout_image is empty
 if st.session_state["layout_image"] is None:
@@ -110,15 +118,26 @@ if st.session_state["layout_image"] is None:
 # Sidebar 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}")
+    filename_png = f"dungeon_map_{now.strftime('%Y%m%d_%H%M%S')}.png"
+    filename_txt = f"room_layout_{now.strftime('%Y%m%d_%H%M%S')}.txt"
+
+    # Save the PNG map
+    st.session_state["canvas"].save(filename_png)
+    st.sidebar.success(f"Map saved as {filename_png}")
+
+    # Save room details as a text file
+    room_details_content = "\n".join(st.session_state["room_details"])
+    with open(filename_txt, "w") as f:
+        f.write(room_details_content)
+    st.sidebar.success(f"Room layout saved as {filename_txt}")
 
-    # 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}")
+    # Generate base64 download links
+    with open(filename_png, "rb") as f:
+        png_base64 = create_base64_download_link(f.read(), filename_png, "octet-stream")
+    txt_base64 = create_base64_download_link(room_details_content, filename_txt, "txt")
+
+    # Add files to save history
+    st.session_state["saved_files"].append((png_base64, txt_base64))
 
 if st.sidebar.button("🗺️ Regenerate Map"):
     image_files = [f for f in os.listdir(map_dir) if f.endswith(".png")]
@@ -129,10 +148,15 @@ if st.sidebar.button("🗺️ Regenerate Map"):
         st.session_state["room_details"] = room_details
         st.rerun()
 
+# Display save history in the sidebar
+st.sidebar.markdown("### 📜 Save History")
+for idx, (png_link, txt_link) in enumerate(st.session_state["saved_files"], start=1):
+    st.sidebar.markdown(f"{idx}. {png_link} | {txt_link}")
+
 # Zoom Controls
 st.sidebar.title("Zoom")
 show_zoomed = st.sidebar.checkbox("Show Zoomed Version")
-zoom_level = st.sidebar.slider("Zoom Level", min_value=0.5, max_value=2.0, value=1.0, step=0.1)
+zoom_level = st.sidebar.slider("Zoom Level", min_value=0.1, max_value=2.0, value=1.0, step=0.1)
 
 if show_zoomed and st.session_state["canvas"] is not None:
     zoomed_canvas = st.session_state["canvas"].resize(