Update app.py

app.py

@@ -17,12 +17,14 @@ from gamestate import GameState
 
 # --- Constants ---
 SAVE_DIR = "saved_worlds"
+GLOBAL_SAVES_DIR = "global_saves"  # Folder for global game saves
 PLOT_WIDTH = 50.0  # Width of each plot in 3D space
 PLOT_DEPTH = 50.0  # Depth of each plot
 CSV_COLUMNS = ['obj_id', 'type', 'pos_x', 'pos_y', 'pos_z', 'rot_x', 'rot_y', 'rot_z', 'rot_order']
 
-# --- Ensure Save Directory Exists ---
+# --- Ensure Required Directories Exist ---
 os.makedirs(SAVE_DIR, exist_ok=True)
+os.makedirs(GLOBAL_SAVES_DIR, exist_ok=True)
 
 @st.cache_data(ttl=3600)
 def load_plot_metadata():
@@ -40,7 +42,7 @@ def load_plot_metadata():
     parsed_plots = []
     for filename in plot_files:
         try:
-            parts = filename[:-4].split('_')  # Remove .csv
+            parts = filename[:-4].split('_')  # Remove .csv extension
             grid_x = int(parts[1][1:])  # After 'X'
             grid_z = int(parts[2][1:])  # After 'Z'
             plot_name = " ".join(parts[3:]) if len(parts) > 3 else f"Plot ({grid_x},{grid_z})"
@@ -144,6 +146,11 @@ if 'new_plot_name' not in st.session_state:
     st.session_state.new_plot_name = ""
 if 'js_save_data_result' not in st.session_state:
     st.session_state.js_save_data_result = None
+if 'player_position' not in st.session_state:
+    # This can be updated from JS if needed
+    st.session_state.player_position = {"x": 0, "y": 0, "z": 0}
+if 'loaded_global_state' not in st.session_state:
+    st.session_state.loaded_global_state = None
 
 plots_metadata = load_plot_metadata()
 all_initial_objects = []
@@ -181,7 +188,7 @@ with st.sidebar:
          st.session_state.selected_object = selected_object_type_widget
 
     st.markdown("---")
-    st.header("Save Work")
+    st.header("Save Work (Per Plot)")
     st.caption("Saves newly placed objects to the current plot. A new plot file is created for new areas.")
     if st.button("💾 Save Current Work", key="save_button"):
         from streamlit_js_eval import streamlit_js_eval
@@ -189,7 +196,61 @@ with st.sidebar:
         streamlit_js_eval(js_code=js_get_data_code, key="js_save_processor")
         st.rerun()
 
-# --- Process Save Data from JS ---
+    st.markdown("---")
+    st.header("Global Save & Load")
+    # Global Save: Persists the entire game state plus current player position.
+    if st.button("💾 Global Save"):
+        global_save_data = {
+             "timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
+             "game_state": game_state.get_state(),
+             "player_position": st.session_state.get("player_position", {"x": 0, "y": 0, "z": 0})
+        }
+        default_save_name = f"save_{time.strftime('%Y%m%d_%H%M%S')}.json"
+        save_file_path = os.path.join(GLOBAL_SAVES_DIR, default_save_name)
+        with open(save_file_path, "w", encoding="utf-8") as f:
+             json.dump(global_save_data, f, indent=2)
+        st.success(f"Global state saved to {default_save_name}")
+        # Also store it in session state for download purposes.
+        st.session_state.loaded_global_state = global_save_data
+
+    # List global save files available in the folder.
+    st.subheader("📂 Global Saves")
+    save_files = sorted([f for f in os.listdir(GLOBAL_SAVES_DIR) if f.endswith(".json")])
+    for file in save_files:
+        if st.button(f"Load {file}", key=file):
+            file_path = os.path.join(GLOBAL_SAVES_DIR, file)
+            with open(file_path, "r", encoding="utf-8") as f:
+                loaded_save = json.load(f)
+            # Update the shared game state with loaded data.
+            with game_state.lock:
+                game_state.world_state = loaded_save.get("game_state", [])
+            st.session_state.player_position = loaded_save.get("player_position", {"x": 0, "y": 0, "z": 0})
+            st.session_state.loaded_global_state = loaded_save
+            st.success(f"Global state loaded from {file}")
+
+    st.markdown("---")
+    st.header("Download Global Save as Markdown")
+    # Use the loaded global state if available; otherwise, use the last global save data.
+    current_save = st.session_state.get("loaded_global_state", {"timestamp": "N/A", "game_state": [], "player_position": {"x": 0, "y": 0, "z": 0}})
+    # Auto-generate a default markdown file name
+    default_md_name = current_save.get("timestamp", "save").replace(":", "").replace(" ", "_") + ".md"
+    download_name = st.text_input("Override File Name", value=default_md_name)
+    if st.button("Generate Markdown & Download"):
+        md_outline = f"""# Global Save: {download_name}
+- ⏰ **Timestamp:** {current_save.get("timestamp", "N/A")}
+- 🎮 **Number of Game Objects:** {len(current_save.get("game_state", []))}
+- 🧭 **Player Position:** {current_save.get("player_position", {"x":0, "y":0, "z":0})}
+
+## Game Objects:
+"""
+        # Optionally, list brief info for each object.
+        for i, obj in enumerate(current_save.get("game_state", []), start=1):
+            obj_type = obj.get("type", "Unknown")
+            pos = (obj.get("pos_x", 0), obj.get("pos_y", 0), obj.get("pos_z", 0))
+            md_outline += f"- {i}. ✨ **{obj_type}** at {pos}\n"
+        st.download_button("Download Markdown Save", data=md_outline, file_name=download_name, mime="text/markdown")
+
+# --- Process Save Data from JS (Per Plot Save) ---
 save_data_from_js = st.session_state.get("js_save_processor", None)
 if save_data_from_js is not None:
     st.info("Received save data from client...")
@@ -238,9 +299,9 @@ if save_data_from_js is not None:
 
 # --- Main Area ---
 st.header("Infinite Shared 3D World")
-st.caption("Move to empty areas to expand the world. Use the sidebar 'Save' to store your work.")
+st.caption("Move to empty areas to expand the world. Use the sidebar 'Save' controls to store your work.")
 
-# Prepare state to inject into HTML/JS
+# Inject state into JS—including the shared GAME_STATE from our GameState singleton.
 injected_state = {
     "ALL_INITIAL_OBJECTS": all_initial_objects,
     "PLOTS_METADATA": plots_metadata,