Update app.py

app.py

@@ -481,225 +481,27 @@ def compute_gsd_metrics(diams: List[float], passing: List[float]) -> Dict[str, f
 
 
 
-# 4) Session state initialization
-if "sites" not in st.session_state:
-    # initialize with a default site
-    st.session_state["sites"] = ({
-        # ---------------------------
-        # Basic Project Info
-        # ---------------------------
-        "Site Name": None,
-        "Project Name": None,
-        "Site ID": None,
-        "Coordinates": "",
-        "lat": None,
-        "lon": None,
-        "Project Description": "",
-    
-        # ---------------------------
-        # Soil Recognition / Classifier
-        # ---------------------------
-        "Soil Class": None,
-        "Soil Recognizer Confidence": None,
-        "USCS": None,
-        "AASHTO": None,
-        "GI": None,
-        "GSD": None,                        # Grain size distribution metrics
-        "classifier_inputs": {},            # User/lab/OCR extracted inputs
-        "classifier_decision": None,        # Full text decision from classifier
-    
-        # ---------------------------
-        # Site Characterization
-        # ---------------------------
-        "Topography": None,                 # manual topo entry
-        "Drainage": None,                   # manual drainage notes
-        "Current Land Use": None,           # linked to environmental data
-        "Regional Geology": None,           # manual geology notes
-    
-        # ---------------------------
-        # Investigations & Lab
-        # ---------------------------
-        "Field Investigation": [],          # e.g. borehole logs
-        "Laboratory Results": [],           # lab test results
-    
-        # ---------------------------
-        # Geotechnical Parameters (incl. REPORT_FIELDS)
-        # ---------------------------
-        "Load Bearing Capacity": None,      # kPa or psf
-        "Skin Shear Strength": None,        # kPa
-        "Relative Compaction": None,        # %
-        "Rate of Consolidation": None,      # mm/yr or days
-        "Nature of Construction": None,     # text
-        "Borehole Count": None,             # number
-        "Max Depth (m)": None,              # m
-        "SPT N (avg)": None,                # blows/ft
-        "CBR (%)": None,                    # %
-        "Allowable Bearing (kPa)": None,    # kPa
-    
-        # ---------------------------
-        # Earth Engine Data
-        # ---------------------------
-        "Soil Profile": {                   # SoilGrids / OpenLandMap
-            "Clay": None,                   # % clay
-            "Sand": None,                   # % sand
-            "Silt": None,                   # % silt
-            "OrganicCarbon": None,          # % organic carbon
-            "pH": None                      # soil pH
-        },
-        "Topo Data": None,                  # Avg elevation
-        "Seismic Data": None,               # PGA/g
-        "Flood Data": None,                 # % flood occurrence
-        "Environmental Data": {
-            "Landcover Stats": None,        # histogram
-            "Forest Loss": None,            # optional (Hansen)
-            "Urban Fraction": None          # optional calc
-        },
-        "Weather Data": {                   # climate summaries
-            "Rainfall": None,
-            "Temperature": None,
-            "Humidity": None
-        },
-        "Atmospheric Data": {               # pollution, aerosols
-            "AerosolOpticalDepth": None,
-            "NO2": None,
-            "CO": None,
-            "PM2.5": None                   # from Sentinel-5P aerosol index
-        },
-    
-        # ---------------------------
-        # Map & Visualization
-        # ---------------------------
-        "ROI": None,                        # GeoJSON of ROI
-        "roi_coords": None,                 # Flattened coords (lat,lon)
-        "map_snapshot": None,               # PNG snapshot
-    
-        # ---------------------------
-        # AI / Reporting
-        # ---------------------------
-        "chat_history": [],
-        "LLM_Report_Text": None,
-        "report_convo_state": 0,
-        "report_missing_fields": [],
-        "report_answers": {}
-    })
-
-
-if "active_site" not in st.session_state:
-    st.session_state["active_site"] = 0
-
-if "llm_model" not in st.session_state:
-    st.session_state["llm_model"] = "groq/compound"
-
-# -------------------
-# API Keys
-# -------------------
-GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
-GEM_API_KEY = os.environ.get("GEM_API")
-if "GEM_API" in st.secrets:
-    GEM_API_KEY = st.secrets["GEM_API"]
+import streamlit as st
+import os
+import pandas as pd
 
 # -------------------
-# Universal LLM Call
+# 1) Session state initialization
 # -------------------
-def llm_generate(prompt: str, model: str = None, max_tokens: int = 512) -> str:
-    """Universal LLM call for Groq, Gemini, DeepSeek."""
-    model_name = model or st.session_state["llm_model"]
-
-    try:
-        # ------------------- GEMINI -------------------
-        if model_name.lower().startswith("gemini"):
-            import requests
-            if not GEM_API_KEY:
-                return "[LLM error: No GEM_API key found in secrets]"
-
-            url = f"https://generativelanguage.googleapis.com/v1beta/models/{model_name}:generateContent?key={GEM_API_KEY}"
-            headers = {"Content-Type": "application/json"}
-            payload = {
-                "contents": [{"role": "user", "parts": [{"text": prompt}]}],
-                "generationConfig": {"temperature": 0.2, "maxOutputTokens": max_tokens}
-            }
-            resp = requests.post(url, headers=headers, json=payload, timeout=60)
-            resp.raise_for_status()
-            data = resp.json()
-            return data["candidates"][0]["content"]["parts"][0]["text"].strip()
-
-        # ------------------- GROQ / DEEPSEEK -------------------
-        elif "groq" in model_name or "llama" in model_name or "deepseek" in model_name:
-            from groq import Groq
-            if not GROQ_API_KEY:
-                return "[LLM error: No GROQ_API key found in secrets/env]"
-
-            client = Groq(api_key=GROQ_API_KEY)
-            completion = client.chat.completions.create(
-                model=model_name,
-                messages=[{"role": "user", "content": prompt}],
-                temperature=0.2,
-                max_tokens=max_tokens
-            )
-            return completion.choices[0].message.content
-
-        else:
-            return f"[LLM error: Unknown model {model_name}]"
-
-    except Exception as e:
-        return f"[LLM error: {e}]"
-
-# 5) UI helper: nice CSS for chat bubbles & page styling
-st.markdown("""
-<style>
-/* Background and card styling */
-body { background: #0b0b0b; color: #e9eef6; }
-.stApp > .main > .block-container { padding-top: 18px; }
-/* Landing and cards */
-.gm-card { background: linear-gradient(180deg, rgba(255,122,0,0.04), rgba(255,122,0,0.02)); border-radius:12px; padding:14px; border:1px solid rgba(255,122,0,0.06);}
-.gm-cta { background: linear-gradient(90deg,#ff7a00,#ff3a3a); color:white; padding:10px 14px; border-radius:10px; font-weight:700; }
-/* Chat bubbles */
-.chat-bot { background: #0f1720; border-left:4px solid #FF7A00; padding:10px 12px; border-radius:12px; margin:6px 0; color:#e9eef6; }
-.chat-user { background: #1a1f27; padding:10px 12px; border-radius:12px; margin:6px 0; color:#cfe6ff; text-align:right;}
-.small-muted { color:#9aa7bf; font-size:12px; }
-</style>
-""", unsafe_allow_html=True)
-
-# 6) Sidebar: navigation, site selector, model selector
-from streamlit_option_menu import option_menu
-
-with st.sidebar:
-    st.markdown("<h2 style='color:#FF8C00;margin:8px 0'>GeoMate V2</h2>", unsafe_allow_html=True)
-    # LLM model selector
-    st.session_state["llm_model"] = st.selectbox("Select LLM model", options=[
-        "meta-llama/llama-4-maverick-17b-128e-instruct",
-        "llama-3.1-8b-instant",
-        "meta-llama/llama-guard-4-12b",
-        "llama-3.3-70b-versatile",
-        "groq/compound",
-        "deepseek-r1-distill-llama-70b",   # ✅ Added DeepSeek
-        "gemini-1.5-pro",                  # ✅ Added Gemini
-        "gemini-1.5-flash"
-    ], index=0)
-    st.markdown("---")
-
-    # Site management controls
-    st.markdown("### Project Sites")
-    site_names = [s.get("Site Name", f"Site {i}") for i,s in enumerate(st.session_state["sites"])]
-    # Add new site input
-    new_site_name = st.text_input("New site name", value="", key="new_site_name_input")
-    if st.button("➕ Add / Create Site"):
-        if new_site_name.strip() == "":
-            st.warning("Enter a name for the new site.")
-        elif len(st.session_state["sites"]) >= MAX_SITES:
-            st.error(f"Maximum of {MAX_SITES} sites allowed.")
-        else:
-            idx = len(st.session_state["sites"])
-            #idx = len(st.session_state["sites"]) + 1
-            st.session_state["sites"].append({
-            "Site Name": new_site_name.strip(),
-            "Project Name": "Project - " + new_site_name.strip(),
-            "Site ID": idx,
+if "sites" not in st.session_state:
+    st.session_state["sites"] = [
+        {
+            # ---------------------------
+            # Basic Project Info
+            # ---------------------------
+            "Site Name": "Default Site",
+            "Project Name": "Project - Default Site",
+            "Site ID": 0,
             "Coordinates": "",
             "lat": None,
             "lon": None,
             "Project Description": "",
-        
+
             # ---------------------------
             # Soil Recognition / Classifier
             # ---------------------------
@@ -708,75 +510,75 @@ with st.sidebar:
             "USCS": None,
             "AASHTO": None,
             "GI": None,
-            "GSD": None,                        # Grain size distribution metrics
-            "classifier_inputs": {},            # User/lab/OCR extracted inputs
-            "classifier_decision": None,        # Full text decision from classifier
-        
+            "GSD": None,
+            "classifier_inputs": {},
+            "classifier_decision": None,
+
             # ---------------------------
             # Site Characterization
             # ---------------------------
-            "Topography": None,                 # manual topo entry
-            "Drainage": None,                   # manual drainage notes
-            "Current Land Use": None,           # linked to environmental data
-            "Regional Geology": None,           # manual geology notes
-        
+            "Topography": None,
+            "Drainage": None,
+            "Current Land Use": None,
+            "Regional Geology": None,
+
             # ---------------------------
             # Investigations & Lab
             # ---------------------------
-            "Field Investigation": [],          # e.g. borehole logs
-            "Laboratory Results": [],           # lab test results
-        
+            "Field Investigation": [],
+            "Laboratory Results": [],
+
             # ---------------------------
-            # Geotechnical Parameters (incl. REPORT_FIELDS)
+            # Geotechnical Parameters
             # ---------------------------
-            "Load Bearing Capacity": None,      # kPa or psf
-            "Skin Shear Strength": None,        # kPa
-            "Relative Compaction": None,        # %
-            "Rate of Consolidation": None,      # mm/yr or days
-            "Nature of Construction": None,     # text
-            "Borehole Count": None,             # number
-            "Max Depth (m)": None,              # m
-            "SPT N (avg)": None,                # blows/ft
-            "CBR (%)": None,                    # %
-            "Allowable Bearing (kPa)": None,    # kPa
-        
+            "Load Bearing Capacity": None,
+            "Skin Shear Strength": None,
+            "Relative Compaction": None,
+            "Rate of Consolidation": None,
+            "Nature of Construction": None,
+            "Borehole Count": None,
+            "Max Depth (m)": None,
+            "SPT N (avg)": None,
+            "CBR (%)": None,
+            "Allowable Bearing (kPa)": None,
+
             # ---------------------------
             # Earth Engine Data
             # ---------------------------
-            "Soil Profile": {                   # SoilGrids / OpenLandMap
-                "Clay": None,                   # % clay
-                "Sand": None,                   # % sand
-                "Silt": None,                   # % silt
-                "OrganicCarbon": None,          # % organic carbon
-                "pH": None                      # soil pH
+            "Soil Profile": {
+                "Clay": None,
+                "Sand": None,
+                "Silt": None,
+                "OrganicCarbon": None,
+                "pH": None
             },
-            "Topo Data": None,                  # Avg elevation
-            "Seismic Data": None,               # PGA/g
-            "Flood Data": None,                 # % flood occurrence
+            "Topo Data": None,
+            "Seismic Data": None,
+            "Flood Data": None,
             "Environmental Data": {
-                "Landcover Stats": None,        # histogram
-                "Forest Loss": None,            # optional (Hansen)
-                "Urban Fraction": None          # optional calc
+                "Landcover Stats": None,
+                "Forest Loss": None,
+                "Urban Fraction": None
             },
-            "Weather Data": {                   # climate summaries
+            "Weather Data": {
                 "Rainfall": None,
                 "Temperature": None,
                 "Humidity": None
             },
-            "Atmospheric Data": {               # pollution, aerosols
+            "Atmospheric Data": {
                 "AerosolOpticalDepth": None,
                 "NO2": None,
                 "CO": None,
-                "PM2.5": None                   # from Sentinel-5P aerosol index
+                "PM2.5": None
             },
-        
+
             # ---------------------------
             # Map & Visualization
             # ---------------------------
-            "ROI": None,                        # GeoJSON of ROI
-            "roi_coords": None,                 # Flattened coords (lat,lon)
-            "map_snapshot": None,               # PNG snapshot
-        
+            "ROI": None,
+            "roi_coords": None,
+            "map_snapshot": None,
+
             # ---------------------------
             # AI / Reporting
             # ---------------------------
@@ -785,9 +587,85 @@ with st.sidebar:
             "report_convo_state": 0,
             "report_missing_fields": [],
             "report_answers": {}
-        })
+        }
+    ]
+
+if "active_site" not in st.session_state:
+    st.session_state["active_site"] = 0
+
+if "llm_model" not in st.session_state:
+    st.session_state["llm_model"] = "groq/compound"
+
+MAX_SITES = 10  # optional: maximum number of sites
 
+# -------------------
+# 2) Sidebar: Site management
+# -------------------
+with st.sidebar:
+    st.markdown("<h2 style='color:#FF8C00;margin:8px 0'>GeoMate V2</h2>", unsafe_allow_html=True)
 
+    # Site management controls
+    st.markdown("### Project Sites")
+
+    site_names = [s.get("Site Name", f"Site {i}") for i, s in enumerate(st.session_state["sites"])]
+
+    # Add new site
+    new_site_name = st.text_input("New site name", value="", key="new_site_name_input")
+    if st.button("➕ Add / Create Site"):
+        if new_site_name.strip() == "":
+            st.warning("Enter a name for the new site.")
+        elif len(st.session_state["sites"]) >= MAX_SITES:
+            st.error(f"Maximum of {MAX_SITES} sites allowed.")
+        else:
+            idx = len(st.session_state["sites"])
+            st.session_state["sites"].append({
+                "Site Name": new_site_name.strip(),
+                "Project Name": "Project - " + new_site_name.strip(),
+                "Site ID": idx,
+                "Coordinates": "",
+                "lat": None,
+                "lon": None,
+                "Project Description": "",
+                "Soil Class": None,
+                "Soil Recognizer Confidence": None,
+                "USCS": None,
+                "AASHTO": None,
+                "GI": None,
+                "GSD": None,
+                "classifier_inputs": {},
+                "classifier_decision": None,
+                "Topography": None,
+                "Drainage": None,
+                "Current Land Use": None,
+                "Regional Geology": None,
+                "Field Investigation": [],
+                "Laboratory Results": [],
+                "Load Bearing Capacity": None,
+                "Skin Shear Strength": None,
+                "Relative Compaction": None,
+                "Rate of Consolidation": None,
+                "Nature of Construction": None,
+                "Borehole Count": None,
+                "Max Depth (m)": None,
+                "SPT N (avg)": None,
+                "CBR (%)": None,
+                "Allowable Bearing (kPa)": None,
+                "Soil Profile": {"Clay": None, "Sand": None, "Silt": None, "OrganicCarbon": None, "pH": None},
+                "Topo Data": None,
+                "Seismic Data": None,
+                "Flood Data": None,
+                "Environmental Data": {"Landcover Stats": None, "Forest Loss": None, "Urban Fraction": None},
+                "Weather Data": {"Rainfall": None, "Temperature": None, "Humidity": None},
+                "Atmospheric Data": {"AerosolOpticalDepth": None, "NO2": None, "CO": None, "PM2.5": None},
+                "ROI": None,
+                "roi_coords": None,
+                "map_snapshot": None,
+                "chat_history": [],
+                "LLM_Report_Text": None,
+                "report_convo_state": 0,
+                "report_missing_fields": [],
+                "report_answers": {}
+            })
             st.success(f"Site '{new_site_name.strip()}' created.")
             st.session_state["active_site"] = idx
             st.rerun()
@@ -795,15 +673,26 @@ with st.sidebar:
     # Active site selector
     if site_names:
         active_index = st.selectbox("Active Site", options=list(range(len(site_names))),
-                                    format_func=lambda x: site_names[x], index=st.session_state["active_site"])
+                                    format_func=lambda x: site_names[x],
+                                    index=st.session_state["active_site"])
         st.session_state["active_site"] = active_index
+
     st.markdown("---")
-    st.write("Active Site JSON (live)")
-    st.json(st.session_state["sites"][st.session_state["active_site"]])
+
+    # Display site info as table instead of JSON
+    active_site = st.session_state["sites"][st.session_state["active_site"]]
+    df = pd.DataFrame(active_site.items(), columns=["Field", "Value"])
+    st.dataframe(df, use_container_width=True, height=400)
 
     st.markdown("---")
     st.markdown("© GeoMate • Advanced geotechnical copilot", unsafe_allow_html=True)
 
+# -------------------
+# 3) Main Page Content
+# -------------------
+st.title("🌍 GeoMate V2 — Geotechnical Copilot")
+st.write("Select a site in the sidebar to view and manage project data.")
+
 # 7) Pages implementation
 import streamlit as st
 import torch