Update app.py

app.py

@@ -95,43 +95,37 @@ map_dir = "."  # Top-level directory
 canvas_size = 3000
 scroll_offset = {"x": 0, "y": 0}
 
-# Generate map on load
+# Initialize session state
 if "layout_image" not in st.session_state:
-    # Scan the directory for .png files
+    st.session_state["layout_image"] = None
+if "canvas" not in st.session_state:
+    st.session_state["canvas"] = None
+if "saved" not in st.session_state:
+    st.session_state["saved"] = False
+
+# Generate map if not saved and layout_image is empty
+if st.session_state["layout_image"] is None or not st.session_state["saved"]:
     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))
         st.session_state["layout_image"] = layout_image
         st.session_state["canvas"] = canvas
-    else:
-        st.write("No PNG files found in the top-level directory.")
-
-# 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
 
 # Display map
-if "layout_image" in st.session_state:
+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,
     )
 
-# Add save button
+# Save map button
 if st.button("💾 Save Map"):
     save_path = "saved_dungeon_map.png"
     st.session_state["canvas"].save(save_path)
+    st.session_state["saved"] = True
     st.success(f"Map saved as {save_path}")
 
 # Regenerate map button
@@ -141,9 +135,20 @@ if st.button("🗺️ Regenerate Map"):
         layout_image, canvas = 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["saved"] = False  # Mark as unsaved
         st.experimental_rerun()
-    else:
-        st.write("No PNG files found in the top-level directory.")
+
+# 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
 
 # Sidebar for file upload
 st.sidebar.title("Options")