Update app.py

app.py

@@ -1302,53 +1302,38 @@ def locator_page():
 
     st.markdown("👉 Draw a polygon/rectangle on the map (use draw tool). After drawing, click **Compute Summaries**.")
 
-    # VITAL: Render the map and capture its state from the frontend on every script run.
     map_data = m.to_streamlit(height=700, responsive=True)
 
-    # Immediately save any newly drawn geometry to the session_state to persist it.
-    # geemap returns the last drawn feature in the 'last_object' key.
     if map_data and map_data.get("last_object"):
         st.session_state["roi_geojson"] = json.dumps(map_data["last_object"])
 
-    # Define a robust helper function to retrieve the ROI.
     def get_roi_from_map():
         if "roi_geojson" in st.session_state and st.session_state["roi_geojson"]:
             try:
-            # The data from the map is a full GeoJSON "Feature".
-            # We need to extract only the 'geometry' part for Earth Engine.
                 feature = json.loads(st.session_state["roi_geojson"])
-            return ee.Geometry(feature['geometry'])
+                return ee.Geometry(feature['geometry'])
             except (json.JSONDecodeError, KeyError, Exception) as e:
                 st.warning(f"Could not parse the stored ROI. Please try drawing it again. Error: {e}")
-            # Clear the invalid ROI to prevent repeated errors
                 del st.session_state["roi_geojson"]
-            return None
-    return None # Return None if no ROI has been saved yet.
+                return None
+        return None
 
-    # Initialize the button's state if it doesn't exist
     if "compute_button" not in st.session_state:
-         st.session_state["compute_button"] = False
-    
-    # When the button is clicked, set its state to True for the subsequent rerun
+        st.session_state["compute_button"] = False
+
     if st.button("Compute Summaries"):
         st.session_state["compute_button"] = True
 
-    # This block runs ONLY if the compute button was clicked in the previous run.
     if st.session_state.get("compute_button", False):
-    # Immediately reset the button's state to False.
-    # This prevents the computation from running again on the next interaction.
         st.session_state["compute_button"] = False
     
-    # Retrieve the ROI using our reliable helper function
-         roi = get_roi_from_map()
-
-    if roi is None:
-        st.error("No drawn ROI found. Please draw a polygon or rectangle and press 'Compute Summaries' again.")
-    else:
-        # --- YOUR COMPUTATION LOGIC GOES HERE ---
-        # This is where you place the code that was below the original ROI check.
-        st.success("Polygon found — computing (this may take a few seconds)...")
+        roi = get_roi_from_map()
 
+        if roi is None:
+            st.error("No drawn ROI found. Please draw a polygon or rectangle and press 'Compute Summaries' again.")
+        else:
+            st.success("Polygon found — computing (this may take a few seconds)...")
+        
         
         # choose soil depth via UI (0-200cm). If soil_img is OpenLandMap provide depth selection; if SoilGrids, show its bands
         chosen_soil_band = None