Update app.py

app.py

@@ -3,6 +3,8 @@ import os
 from PIL import Image
 import random
 from io import BytesIO
+import datetime
+import base64
 
 # Function to arrange images dynamically
 def arrange_images(image_files, canvas_size=(3000, 3000)):
@@ -84,6 +86,12 @@ def arrange_images(image_files, canvas_size=(3000, 3000)):
     buffer.seek(0)
     return buffer, canvas
 
+# 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")
 st.write("Automatically generates dungeon maps by arranging PNG images dynamically.")
@@ -102,6 +110,8 @@ if "canvas" not in st.session_state:
     st.session_state["canvas"] = None
 if "saved" not in st.session_state:
     st.session_state["saved"] = False
+if "saved_filename" not in st.session_state:
+    st.session_state["saved_filename"] = None
 
 # Generate map if not saved and layout_image is empty
 if st.session_state["layout_image"] is None or not st.session_state["saved"]:
@@ -116,17 +126,25 @@ if st.session_state["layout_image"] is not None:
     st.image(
         st.session_state["layout_image"],
         caption="Generated Dungeon Map Layout",
-        use_container_width=True,
+        use_column_width=True,
         output_format="PNG",
         clamp=True,
     )
 
 # Save map button
 if st.button("💾 Save Map"):
-    save_path = "saved_dungeon_map.png"
-    st.session_state["canvas"].save(save_path)
+    now = datetime.datetime.now()
+    filename = f"dungeon_map_{now.strftime('%Y%m%d_%H%M%S')}.png"
+    st.session_state["canvas"].save(filename)
     st.session_state["saved"] = True
-    st.success(f"Map saved as {save_path}")
+    st.session_state["saved_filename"] = filename
+
+    st.success(f"Map saved as {filename}")
+    # Generate Base64 download link
+    buffer = BytesIO()
+    st.session_state["canvas"].save(buffer, format="PNG")
+    buffer.seek(0)
+    st.markdown(create_download_link(buffer, filename), unsafe_allow_html=True)
 
 # Regenerate map button
 if st.button("🗺️ Regenerate Map"):
@@ -136,6 +154,7 @@ if st.button("🗺️ Regenerate Map"):
         st.session_state["layout_image"] = layout_image
         st.session_state["canvas"] = canvas
         st.session_state["saved"] = False  # Mark as unsaved
+        st.session_state["saved_filename"] = None
         st.experimental_rerun()
 
 # WASD Controls for scrolling