Update app.py

app.py

@@ -120,4 +120,177 @@ plots_metadata, next_plot_x_offset = load_plot_metadata()
 # --- Load ALL Objects for Rendering ---
 # This could be slow with many plots! Consider optimization later.
 all_initial_objects = []
-for plot in plots_metadata:
+for plot in plots_metadata:
+    all_initial_objects.extend(load_plot_objects(plot['filename'], plot['x_offset']))
+
+# --- Sidebar ---
+with st.sidebar:
+    st.title("🏗️ World Controls")
+
+    st.header("Navigation (Plots)")
+    st.caption("Click to teleport player to the start of a plot.")
+
+    # Use columns for a horizontal button layout if desired
+    cols = st.columns(3)
+    col_idx = 0
+    for plot in plots_metadata:
+        # Use an emoji + name for the button
+        button_label = f"➡️ {plot.get('name', plot['id'])}"
+        if cols[col_idx].button(button_label, key=f"nav_{plot['id']}"):
+            # Send command to JS to move the player
+            target_x = plot['x_offset']
+            streamlit_js_eval(js_code=f"teleportPlayer({target_x});")
+            # No rerun needed here, JS handles the move
+
+        col_idx = (col_idx + 1) % 3 # Cycle through columns
+
+    st.markdown("---")
+
+    # --- Object Placement ---
+    st.header("Place Objects")
+    object_types = ["None", "Simple House", "Tree", "Rock", "Fence Post"]
+    current_object_index = 0
+    try:
+        current_object_index = object_types.index(st.session_state.selected_object)
+    except ValueError:
+        st.session_state.selected_object = "None" # Reset if invalid
+
+    selected_object_type_widget = st.selectbox(
+        "Select Object:",
+        options=object_types,
+        index=current_object_index,
+        key="selected_object_widget"
+    )
+    if selected_object_type_widget != st.session_state.selected_object:
+         st.session_state.selected_object = selected_object_type_widget
+         # No rerun needed just for selection, JS will read state on next placement
+
+
+    st.markdown("---")
+
+    # --- Saving ---
+    st.header("Save New Plot")
+    st.session_state.new_plot_name = st.text_input(
+        "Name for New Plot:",
+        value=st.session_state.new_plot_name,
+        placeholder="My Awesome Creation"
+    )
+
+    if st.button("💾 Save Current Work as New Plot"):
+        # 1. Trigger JS function to get data
+        # This function `getSaveData` should be defined in index.html
+        # It should collect data for objects considered "new"
+        # and return them as a JSON string.
+        # The positions should be WORLD positions.
+        js_get_data_code = "getSaveData();"
+        # Use streamlit_js_eval to run JS and get data back into session_state
+        st.session_state.save_request_data = streamlit_js_eval(
+             js_code=js_get_data_code,
+             key="get_save_data" # Unique key for this eval call
+             # Removed want_output=True, value goes to key if specified
+        )
+        # Need to trigger a rerun to process the data AFTER it arrives
+        # Using sync() might help ensure the value is available immediately after? Let's try without first.
+        st.rerun() # Rerun to process the returned data below
+
+
+# --- Process Save Data (if received from JS via key) ---
+# This runs AFTER the rerun triggered by the save button
+save_data_from_js = st.session_state.get("save_request_data", None)
+
+if save_data_from_js:
+    st.info("Received save data from client...")
+    try:
+        objects_to_save = json.loads(save_data_from_js) # Parse JSON string from JS
+
+        if isinstance(objects_to_save, list) and len(objects_to_save) > 0:
+            # Determine filename for the new plot
+            new_plot_index = len(plots_metadata) + 1
+            plot_name_sanitized = "".join(c for c in st.session_state.new_plot_name if c.isalnum() or c in (' ', '_')).rstrip() or f"Plot_{new_plot_index}"
+            new_filename = f"plot_{new_plot_index:03d}_{plot_name_sanitized.replace(' ','_')}.csv"
+
+            # Save the data, converting world coords to relative coords inside the func
+            save_ok = save_plot_data(new_filename, objects_to_save, next_plot_x_offset)
+
+            if save_ok:
+                # IMPORTANT: Clear the plot metadata cache so it reloads with the new file
+                load_plot_metadata.clear()
+                # Clear the JS data request state
+                st.session_state.save_request_data = None
+                # Reset the new plot name field
+                st.session_state.new_plot_name = ""
+                # Reset newly placed objects in JS? Needs another call.
+                try:
+                     streamlit_js_eval(js_code="resetNewlyPlacedObjects();", key="reset_js_state")
+                except Exception as js_e:
+                     st.warning(f"Could not reset JS state after save: {js_e}")
+
+                st.success(f"New plot '{plot_name_sanitized}' saved!")
+                # Rerun AGAIN to reload plots list, redraw minimap, and update JS injection
+                st.rerun()
+            else:
+                st.error("Failed to save plot data to file.")
+                # Clear the request data even if save failed to prevent retry loop
+                st.session_state.save_request_data = None
+
+        elif isinstance(objects_to_save, list) and len(objects_to_save) == 0:
+            st.warning("Nothing new to save.")
+            st.session_state.save_request_data = None # Clear request
+        else:
+             st.error(f"Received invalid save data format from client: {type(objects_to_save)}")
+             st.session_state.save_request_data = None # Clear request
+
+
+    except json.JSONDecodeError:
+        st.error("Failed to decode save data from client. Data might be corrupted or empty.")
+        print("Received raw data:", save_data_from_js) # Log for debugging
+        st.session_state.save_request_data = None # Clear request
+    except Exception as e:
+        st.error(f"Error processing save: {e}")
+        st.exception(e)
+        st.session_state.save_request_data = None # Clear request
+
+
+# --- Main Area ---
+st.header("Shared 3D World")
+st.caption("Build side-by-side with others. Changes load on refresh/save.")
+
+# --- Load and Prepare HTML ---
+html_file_path = 'index.html'
+html_content_with_state = None # Initialize
+
+try:
+    with open(html_file_path, 'r', encoding='utf-8') as f:
+        html_template = f.read()
+
+    # --- Inject Python state into JavaScript ---
+    js_injection_script = f"""
+<script>
+    // Set global variables BEFORE the main script runs
+    window.ALL_INITIAL_OBJECTS = {json.dumps(all_initial_objects)}; // All objects from all plots
+    window.SELECTED_OBJECT_TYPE = {json.dumps(st.session_state.selected_object)};
+    window.PLOT_WIDTH = {json.dumps(PLOT_WIDTH)};
+    window.NEXT_PLOT_X_OFFSET = {json.dumps(next_plot_x_offset)}; // Needed for save calculation
+    console.log("Streamlit State Injected:", {{
+        selectedObject: window.SELECTED_OBJECT_TYPE,
+        initialObjectsCount: window.ALL_INITIAL_OBJECTS ? window.ALL_INITIAL_OBJECTS.length : 0,
+        plotWidth: window.PLOT_WIDTH,
+        nextPlotX: window.NEXT_PLOT_X_OFFSET
+    }});
+</script>
+"""
+    html_content_with_state = html_template.replace('</head>', js_injection_script + '\n</head>', 1)
+
+    # --- Embed HTML Component ---
+    components.html(
+        html_content_with_state,
+        height=750,
+        scrolling=False
+    )
+
+except FileNotFoundError:
+    st.error(f"CRITICAL ERROR: Could not find the file '{html_file_path}'.")
+    st.warning(f"Make sure `{html_file_path}` is in the same directory as `app.py` and `{SAVE_DIR}` exists.")
+except Exception as e:
+    st.error(f"An critical error occurred during HTML preparation or component rendering: {e}")
+    st.exception(e)