Update app.py

app.py

@@ -69,18 +69,39 @@ def persist_sites():
         st.error(f"⚠️ Error saving sites: {e}")
 
 def get_active_site():
-    """Return the active site dict"""
-    idx = st.session_state.get("active_site_idx")
-    if idx is None or idx >= len(st.session_state["sites"]):
-        return None
-    return st.session_state["sites"][idx]
+    # Fetch current index safely
+    idx = st.session_state.get("active_site_idx", 0)
+    
+    # Ensure sites list exists
+    sites = st.session_state.get("sites", [])
+    
+    # If no sites exist, create a default site
+    if not sites:
+        st.session_state["sites"] = [{"Site Name": "Site 1"}]
+        st.session_state["active_site_idx"] = 0
+        return st.session_state["sites"][0]
+    
+    # Ensure idx is within bounds
+    if idx < 0 or idx >= len(sites):
+        st.session_state["active_site_idx"] = 0
+        idx = 0
+    
+    return sites[idx]
+
+
+def save_active_site(site_data):
+    sites = st.session_state.get("sites", [])
+    
+    if not sites:
+        st.session_state["sites"] = [site_data]
+        st.session_state["active_site_idx"] = 0
+    else:
+        idx = st.session_state.get("active_site_idx", 0)
+        if idx < 0 or idx >= len(sites):
+            idx = 0
+            st.session_state["active_site_idx"] = 0
+        st.session_state["sites"][idx] = site_data
 
-def save_active_site(site: dict):
-    """Save updates to current site and persist"""
-    idx = st.session_state.get("active_site_idx")
-    if idx is not None and idx < len(st.session_state["sites"]):
-        st.session_state["sites"][idx] = site
-        persist_sites()
 
 def create_new_site(name: str):
     """Create new site with maximum soil details and set active"""