Update app.py

app.py

@@ -694,24 +694,34 @@ body { background: #0b0b0b; color: #e9eef6; }
 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",
+    st.markdown("### ⚙️ LLM Model Selector")
+    st.session_state["llm_model"] = st.selectbox(
+        "Choose AI Model",
+        [
+            "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",   # ✅ DeepSeek
+            "gemini-1.5-pro",                  # ✅ Gemini Pro
+            "gemini-1.5-flash",     # ✅ Gemini Flash
+            "gemini-2.5-pro"
+        ],
+        index=0 if "llm_model" not in st.session_state else
+        ["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",   # ✅ DeepSeek
         "gemini-1.5-pro",                  # ✅ Gemini Pro
-        "gemini-1.5-flash"                 # ✅ Gemini Flash
-    ], index=0)
-
-    st.markdown("---")
-
+        "gemini-1.5-flash",  # ✅ Gemini Flash
+        "gemini-2.5-pro"
+        ].index(
+            st.session_state.get("llm_model", "groq/compound")
+        )
+    )
     # -------------------
     # Site management
     # -------------------
@@ -797,17 +807,8 @@ with st.sidebar:
     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
-from torchvision import transforms
-from PIL import Image
 import os
 
 # -------------------- SESSION INIT --------------------
@@ -2015,20 +2016,7 @@ if "page" not in st.session_state:
 # ===============================
 # Sidebar or Top-Bar Model Selector
 # ===============================
-with st.sidebar:
-    st.markdown("### ⚙️ LLM Model Selector")
-    st.session_state["llm_model"] = st.selectbox(
-        "Choose AI Model",
-        [
-            "groq/compound",                  # Groq general model
-            "deepseek-r1-distill-llama-70b",  # DeepSeek
-            "gemini-1.5-pro",                 # Gemini
-        ],
-        index=0 if "llm_model" not in st.session_state else
-        ["groq/compound", "deepseek-r1-distill-llama-70b", "gemini-1.5-pro"].index(
-            st.session_state.get("llm_model", "groq/compound")
-        )
-    )
+    
 
 # ===============================
 # Page Menu (Horizontal)