Update app.py

app.py

@@ -1008,31 +1008,38 @@ def ocr_page():
 
 # Locator Page (Earth Engine integration)
 
+import streamlit as st
+import geemap.foliumap as geemap
 import ee
 import os
-import json
 
-# Load EE service account credentials from Hugging Face secret
+# ----------------------------
+# Authenticate Earth Engine
+# ----------------------------
 EE_JSON = os.getenv("EARTHENGINE_TOKEN")
 
 if EE_JSON:
-    creds = ee.ServiceAccountCredentials(
-        email='YOUR_SERVICE_ACCOUNT_EMAIL',
-        key_data=json.loads(EE_JSON)
-    )
-    ee.Initialize(creds)
+    try:
+        creds = ee.ServiceAccountCredentials(
+            email=os.getenv("SERVICE_ACCOUNT"),   # <-- replace with your service account email
+            key_data=EE_JSON                      # pass raw JSON string, not dict
+        )
+        ee.Initialize(creds)
+    except Exception as e:
+        st.error(f"❌ Failed to initialize Earth Engine: {e}")
 else:
-    ee.Initialize()  # fallback if running locally with `earthengine authenticate`
+    try:
+        ee.Initialize()  # fallback if running locally
+    except Exception as e:
+        st.error(f"❌ Local EE init failed: {e}")
+
+
 def locator_page():    
     st.header("🌍 Locator — Interactive Selection")
     st.markdown("➡️ Draw a point, rectangle, or polygon on the map to select your area of interest.")
 
     site = st.session_state["sites"][st.session_state["active_site"]]
 
-    if not EE_JSON:
-        st.error("❌ Earth Engine/Geemap not available in this environment. Please add `geemap` & `earthengine-api` to requirements.txt.")
-        return
-
     try:
         # ----------------------------
         # Create interactive map
@@ -1046,12 +1053,13 @@ def locator_page():
         # Save map HTML for display
         map_file = "/tmp/locator_draw.html"
         m.to_html(map_file)
-        map_html = open(map_file, "r").read()
-        st.components.v1.html(map_html, height=600)
+        st.components.v1.html(open(map_file, "r").read(), height=600)
 
+        # ----------------------------
         # Button to process drawn geometry
+        # ----------------------------
         if st.button("Fetch Earth Data (EE)"):
-            drawn = m.user_roi  # get user drawn geometry
+            drawn = m.user_roi
             if drawn is None:
                 st.warning("⚠️ No region selected. Please draw a point or polygon on the map.")
                 return
@@ -1075,7 +1083,7 @@ def locator_page():
             site["Flood Data"] = f"Flood occurrence probability: {flood_val:.1f}%" if flood_val else "No flood data"
 
             # ----------------------------
-            # Soil (OpenLandMap b200 = 100–200 cm)
+            # Soil (OpenLandMap, b200 = 100–200 cm depth)
             # ----------------------------
             soil = ee.Image("OpenLandMap/SOL/SOL_CLAY-WFRACTION_USDA-3A1A1A_M/v02").select("b200")
             soil_sample = soil.sample(region=geom, scale=250).first()
@@ -1129,14 +1137,13 @@ def locator_page():
             m2.addLayer(gsw, {"min": 0, "max": 100, "palette": ["white", "blue"]}, "Flood Occurrence")
             m2.addLayer(soil, {"min": 0.0, "max": 0.6, "palette": ["#ffffcc","#c2e699","#78c679","#31a354"]}, "Soil Clay (100–200 cm)")
             m2.addLayer(seismic, {"min": 0, "max": 1, "palette": ["green", "yellow", "red"]}, "Seismic Hazard")
-            m2.addLayer(landcover, {"min": 10, "max": 100, "palette": ["006400","ffbb22","ffff4c","f096ff","fa0000","b4b4b4","f0f0f0","0064c8","0096a0","00cf75","fae6a0"]}, "Landcover")
+            m2.addLayer(landcover, {"min": 10, "max": 100}, "Landcover")
             m2.addLayer(forest, {"palette": ["white", "green"]}, "Forest Loss")
             m2.addLayer(urban, {"min": 0, "max": 1, "palette": ["white", "black"]}, "Urban Footprint")
 
             m2.addLayer(drawn, {}, "Selected AOI")
             m2.addLayerControl()
 
-            # Save final map
             map_file2 = "/tmp/locator_results.html"
             m2.to_html(map_file2)
             st.components.v1.html(open(map_file2, "r").read(), height=600)