Update app.py

app.py

@@ -728,63 +728,111 @@ with st.sidebar:
 
 # 7) Pages implementation
 def landing_page():
-    # Background hero with placeholder image (replace BACKGROUND_URL with your image path or URL)
-    BACKGROUND_URL = "/app/background_placeholder.jpg"  # <- replace this (or provide URL)
+    # Background hero image
+    BACKGROUND_URL = "/app/background_placeholder.jpg"  # Replace with your image or URL
+
+    # -------------------- HERO SECTION --------------------
     st.markdown(f"""
     <div style="
         background-image: url('{BACKGROUND_URL}');
         background-size: cover;
         background-position: center;
-        padding: 48px 28px;
-        border-radius: 12px;
-        margin-bottom: 18px;
+        padding: 64px 32px;
+        border-radius: 16px;
+        margin-bottom: 24px;
         position: relative;
+        box-shadow: 0 6px 24px rgba(0,0,0,0.4);
         ">
-      <div style="background: rgba(11,11,11,0.55); padding:22px; border-radius:10px; max-width:900px;">
-        <h1 style='color:#FF8C00; margin:0'>GeoMate V2</h1>
-        <p style='color:#e8eef6; margin:6px 0 0; font-size:16px'>
-          AI geotechnical copilot — soil recognition, classification, locator (EE), RAG-powered Q&A, and dynamic reports.
+      <div style="background: rgba(15,15,15,0.65);
+                  padding:32px;
+                  border-radius:14px;
+                  max-width:980px;">
+        <h1 style='color:#FF8C00; font-size:42px; margin:0 0 12px'>
+            🌍 GeoMate V2
+        </h1>
+        <p style='color:#f2f4f7; font-size:17px; margin:0 0 14px; line-height:1.6'>
+          Your AI Geotechnical Copilot — soil recognition, classification, locator (Google Earth Engine), 
+          RAG-powered Q&A, and dynamic professional reports.
         </p>
-        <div style='margin-top:8px; color:#cfcfcf; font-size:13px'>
-          Quick: Classifier • GSD • Locator • RAG • Reports
+        <div style='margin-top:10px; color:#d6d6d6; font-size:14px'>
+          🚀 Quick: Classifier • GSD • Locator • RAG • OCR • Reports
         </div>
       </div>
     </div>
     """, unsafe_allow_html=True)
 
-    st.markdown("<div style='display:flex;align-items:center;gap:12px'>"
-                "<div style='width:76px;height:76px;border-radius:14px;background:linear-gradient(135deg,#ff7a00,#ff3a3a);display:flex;align-items:center;justify-content:center;box-shadow:0 8px 24px rgba(0,0,0,0.6)'>"
-                "<span style='font-size:34px'>🛰️</span></div>"
-                "<div><h1 style='margin:0;color:#FF8C00'>GeoMate V2</h1>"
-                "<div class='small-muted'>AI geotechnical copilot — soil recognition, classification, locator, RAG, and reports</div></div></div>", unsafe_allow_html=True)
+    # -------------------- BRAND HEADER --------------------
+    st.markdown("""
+    <div style="display:flex;align-items:center;gap:16px; margin-bottom:14px">
+        <div style="width:80px;height:80px;
+                    border-radius:16px;
+                    background:linear-gradient(135deg,#ff7a00,#ff3a3a);
+                    display:flex;align-items:center;justify-content:center;
+                    box-shadow:0 6px 18px rgba(0,0,0,0.55)">
+            <span style='font-size:36px'>🛰️</span>
+        </div>
+        <div>
+            <h1 style='margin:0;color:#FF8C00;font-size:30px'>GeoMate V2</h1>
+            <div style='color:#888; font-size:14px'>
+                AI geotechnical copilot — smarter, faster & field-ready
+            </div>
+        </div>
+    </div>
+    """, unsafe_allow_html=True)
     st.markdown("---")
+
+    # -------------------- CONTENT GRID --------------------
     col1, col2 = st.columns([2,1])
+
     with col1:
         st.markdown("<div class='gm-card'>", unsafe_allow_html=True)
-        st.write("GeoMate is built to help geotechnical engineers: classify soils (USCS/AASHTO), plot GSD, fetch Earth Engine data, chat with a RAG-backed LLM, and generate professional geotechnical reports.")
+        st.write("GeoMate is built to help geotechnical engineers streamline workflows:")
+        st.markdown("""
+        - 🧪 **Soil Recognition & Classification** (USCS, AASHTO)  
+        - 📈 **Grain Size Distribution (GSD) Curve plotting**  
+        - 🌍 **Earth Engine data fetch for sites**  
+        - 🤖 **RAG + LLM Q&A** for textbook-driven answers  
+        - 📑 **Professional report generation** (PDF-ready)  
+        - 📷 **OCR-based input from site drawings**  
+        """)
         st.markdown("</div>", unsafe_allow_html=True)
-        st.markdown("### Quick Actions")
-        c1, c2, c3 = st.columns(3)
-        if c1.button("🧪 Classifier"):
-            st.session_state["page"] = "Classifier"; st.rerun()
-        if c2.button("📈 GSD Curve"):
-            st.session_state["page"] = "GSD"; st.rerun()
-        if c3.button("🌍 Locator"):
-            st.session_state["page"] = "Locator"; st.rerun()
-        c4, c5, c6 = st.columns(3)
-        if c4.button("🤖 GeoMate Ask"):
-            st.session_state["page"] = "RAG"; st.rerun()
-        if c5.button("📷 OCR"):
-            st.session_state["page"] = "OCR"; st.rerun()
-        if c6.button("📑 Reports"):
-            st.session_state["page"] = "Reports"; st.rerun()
+
+        # Quick Actions as styled cards
+        st.markdown("### ⚡ Quick Actions")
+        qa_cols = st.columns(3)
+        quick_actions = [
+            ("🧪", "Classifier", "Classifier"),
+            ("📈", "GSD Curve", "GSD"),
+            ("🌍", "Locator", "Locator"),
+            ("🤖", "GeoMate Ask", "RAG"),
+            ("📷", "OCR", "OCR"),
+            ("📑", "Reports", "Reports"),
+        ]
+        for i, (icon, label, page) in enumerate(quick_actions):
+            with qa_cols[i % 3]:
+                if st.button(f"{icon} {label}", use_container_width=True):
+                    st.session_state["page"] = page
+                    st.rerun()
+
+    # -------------------- SITE SUMMARY --------------------
     with col2:
-        st.markdown("<div class='gm-card' style='text-align:center'>", unsafe_allow_html=True)
-        st.markdown("<h3 style='color:#FF8C00'>Live Site Summary</h3>", unsafe_allow_html=True)
         site = st.session_state["sites"][st.session_state["active_site"]]
-        st.write(f"Site: **{site.get('Site Name')}**")
-        st.write(f"USCS: {site.get('USCS')}, AASHTO: {site.get('AASHTO')}")
-        st.write(f"GSD saved: {'Yes' if site.get('GSD') else 'No'}")
+        st.markdown("""
+        <div style="
+            background: #1e1e1e;
+            border-radius:14px;
+            padding:20px;
+            text-align:center;
+            box-shadow:0 4px 14px rgba(0,0,0,0.5);
+        ">
+          <h3 style='color:#FF8C00; margin:0 0 10px'>📍 Live Site Summary</h3>
+        """, unsafe_allow_html=True)
+
+        st.write(f"🏗️ **Site:** {site.get('Site Name')}")
+        st.write(f"🧱 **USCS:** {site.get('USCS', '—')}")
+        st.write(f"🛣️ **AASHTO:** {site.get('AASHTO', '—')}")
+        st.write(f"📊 **GSD saved:** {'✅ Yes' if site.get('GSD') else '❌ No'}")
+
         st.markdown("</div>", unsafe_allow_html=True)
 
 # Soil Classifier page (conversational, step-by-step)
@@ -1280,30 +1328,39 @@ def locator_page():
     # Render map and capture ROI (geemap only)
     # ----------------------------
     st.markdown("👉 Draw a polygon/rectangle/circle using the draw tool, then press **Compute Summaries**")
-    
+
     # Render the map inside Streamlit
     m.to_streamlit(height=600, responsive=True)
     
-    # Capture ROI directly from geemap draw_features
-    roi = None
-    if m.draw_features:
-        try:
-            last_feature = m.draw_features[-1]  # last drawn geometry
-            if isinstance(last_feature, dict):
-                geom = last_feature.get("geometry") or last_feature
-                roi = ee.Geometry(geom)
-                st.session_state["roi_geojson"] = json.dumps(last_feature)
-                st.success("ROI captured successfully ✅")
-        except Exception as e:
-            st.warning(f"Could not parse drawn ROI: {e}")
-            roi = None
-    elif "roi_geojson" in st.session_state:
-        # Restore ROI from session_state if already drawn earlier
-        try:
-            geom = json.loads(st.session_state["roi_geojson"]).get("geometry")
-            roi = ee.Geometry(geom)
-        except Exception:
-            roi = None
+    # ----------------------------
+    # ROI capture helper
+    # ----------------------------
+    def get_roi_from_map():
+        """Return ee.Geometry or None if no ROI drawn"""
+        if m.draw_features:
+            try:
+                last_feature = m.draw_features[-1]  # last drawn geometry
+                if isinstance(last_feature, dict):
+                    geom = last_feature.get("geometry") or last_feature
+                    st.session_state["roi_geojson"] = json.dumps(last_feature)
+                    return ee.Geometry(geom)
+            except Exception as e:
+                st.warning(f"Could not parse drawn ROI: {e}")
+                return None
+        elif "roi_geojson" in st.session_state:
+            # Restore ROI if already saved in session
+            try:
+                geom = json.loads(st.session_state["roi_geojson"]).get("geometry")
+                return ee.Geometry(geom)
+            except Exception:
+                return None
+        return None
+    
+    # Call helper immediately to update ROI
+    roi = get_roi_from_map()
+    if roi:
+        st.success("ROI captured successfully ✅")
+
     
 
     # --- Compute Summaries ---