Update app.py

app.py

@@ -727,113 +727,174 @@ with st.sidebar:
     st.markdown("© GeoMate • Advanced geotechnical copilot", unsafe_allow_html=True)
 
 # 7) Pages implementation
+import streamlit as st
+import torch
+from torchvision import transforms
+from PIL import Image
+import os
+
+# -------------------- SESSION INIT --------------------
+if "page" not in st.session_state:
+    st.session_state["page"] = "Landing"
+if "sites" not in st.session_state:
+    st.session_state["sites"] = [{"Site Name": "Default Site", "USCS": None, "AASHTO": None, "GSD": None}]
+if "active_site" not in st.session_state:
+    st.session_state["active_site"] = 0
+
+# -------------------- MODEL PLACEHOLDER --------------------
+# Load soil recognizer model here (replace with your trained PyTorch model)
+# model = torch.load("soil_model.pth", map_location="cpu")
+# model.eval()
+
+# Placeholder soil classes
+soil_classes = ["Gravel", "Sand", "Silt", "Clay", "Loam"]
+
+# -------------------- LANDING PAGE --------------------
 def landing_page():
-    # Background hero image
-    BACKGROUND_URL = "/app/background_placeholder.jpg"  # Replace with your image or URL
+    BACKGROUND_IMAGES = [
+        "/app/bg1.jpg",  # replace with your images
+        "/app/bg2.jpg",
+        "/app/bg3.jpg"
+    ]
 
-    # -------------------- HERO SECTION --------------------
-    st.markdown(f"""
-    <div style="
-        background-image: url('{BACKGROUND_URL}');
-        background-size: cover;
-        background-position: center;
-        padding: 64px 32px;
+    # Inject CSS + animations
+    st.markdown("""
+    <style>
+    body {background:#000}
+    /* Background slider */
+    @keyframes bgFade {
+        0% {opacity:1;}
+        30% {opacity:1;}
+        33% {opacity:0;}
+        97% {opacity:0;}
+        100% {opacity:1;}
+    }
+    .hero {
+        position: relative;
         border-radius: 16px;
+        overflow: hidden;
+        height: 420px;
         margin-bottom: 24px;
-        position: relative;
-        box-shadow: 0 6px 24px rgba(0,0,0,0.4);
-        ">
-      <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:10px; color:#d6d6d6; font-size:14px'>
-          🚀 Quick: Classifier • GSD • Locator • RAG • OCR • Reports
-        </div>
-      </div>
-    </div>
+        box-shadow: 0 8px 28px rgba(0,0,0,0.65);
+    }
+    .hero img {
+        position: absolute;
+        top:0;left:0;
+        width:100%;height:100%;
+        object-fit: cover;
+        animation: bgFade 15s infinite;
+    }
+    .hero img:nth-child(2) { animation-delay: 5s; }
+    .hero img:nth-child(3) { animation-delay: 10s; }
+
+    /* Overlay content */
+    .overlay {
+        position: absolute;
+        top:0;left:0;
+        width:100%;height:100%;
+        background: rgba(15,15,15,0.65);
+        display:flex;
+        flex-direction:column;
+        justify-content:center;
+        padding: 40px;
+        animation: fadeIn 1.5s ease-in-out;
+    }
+    @keyframes fadeIn {
+        from {opacity:0; transform: translateY(20px);}
+        to {opacity:1; transform: translateY(0);}
+    }
+    h1.hero-title {
+        color:#FF8C00;
+        font-size:44px;
+        margin:0;
+        font-weight:800;
+    }
+    p.hero-sub {
+        color:#f0f0f0;
+        font-size:18px;
+        margin-top:12px;
+        line-height:1.6;
+    }
+    .quick-btn {
+        display:inline-block;
+        margin:10px 8px 0 0;
+        padding:12px 22px;
+        border-radius:10px;
+        background: linear-gradient(135deg,#ff7a00,#ff3a3a);
+        color:white;
+        font-weight:600;
+        text-decoration:none;
+        transition: all .3s ease;
+    }
+    .quick-btn:hover {
+        transform: translateY(-3px);
+        box-shadow:0 6px 18px rgba(0,0,0,0.4);
+    }
+    </style>
     """, 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
+    # Hero with slider
+    st.markdown(f"""
+    <div class="hero">
+        <img src="{BACKGROUND_IMAGES[0]}"/>
+        <img src="{BACKGROUND_IMAGES[1]}"/>
+        <img src="{BACKGROUND_IMAGES[2]}"/>
+
+        <div class="overlay">
+            <h1 class="hero-title">GeoMate V2</h1>
+            <p class="hero-sub">
+                AI Geotechnical Copilot — soil recognition, classification, 
+                locator (EE), RAG-powered Q&A, OCR, and dynamic reports.
+            </p>
+            <div style="margin-top:18px">
+                <a href="#" onclick="window.parent.postMessage({{type: 'streamlit:setSessionState', key:'page', value:'Classifier'}}, '*')" class="quick-btn">🧪 Classifier</a>
+                <a href="#" onclick="window.parent.postMessage({{type: 'streamlit:setSessionState', key:'page', value:'GSD'}}, '*')" class="quick-btn">📈 GSD</a>
+                <a href="#" onclick="window.parent.postMessage({{type: 'streamlit:setSessionState', key:'page', value:'Locator'}}, '*')" class="quick-btn">🌍 Locator</a>
+                <a href="#" onclick="window.parent.postMessage({{type: 'streamlit:setSessionState', key:'page', value:'RAG'}}, '*')" class="quick-btn">🤖 Ask</a>
+                <a href="#" onclick="window.parent.postMessage({{type: 'streamlit:setSessionState', key:'page', value:'Reports'}}, '*')" class="quick-btn">📑 Reports</a>
             </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 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)
-
-        # 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()
+    # Live site summary
+    site = st.session_state["sites"][st.session_state["active_site"]]
+    st.markdown(f"""
+    <div style="
+        background:#1e1e1e;
+        border-radius:14px;
+        padding:20px;
+        margin-top:16px;
+        text-align:center;
+        box-shadow:0 4px 14px rgba(0,0,0,0.5);
+        animation: fadeIn 1.2s ease-in-out;
+    ">
+        <h3 style='color:#FF8C00; margin:0 0 10px'>📍 Live Site Summary</h3>
+        <p style='color:#fff'>🏗️ <b>Site:</b> {site.get('Site Name')}</p>
+        <p style='color:#fff'>🧱 <b>USCS:</b> {site.get('USCS') or "—"}</p>
+        <p style='color:#fff'>🛣️ <b>AASHTO:</b> {site.get('AASHTO') or "—"}</p>
+        <p style='color:#fff'>📊 <b>GSD saved:</b> {"✅ Yes" if site.get("GSD") else "❌ No"}</p>
+    </div>
+    """, unsafe_allow_html=True)
+
+# -------------------- MAIN APP --------------------
+st.set_page_config(page_title="GeoMate V2", layout="wide")
+if page == "Home":
+    landing_page()
+elif st.session_state["page"] == "Soil recognizer":
+    soil_recognizer_page()
+elif st.session_state["page"] == "Classifier":
+    soil_classifier_page()
+elif st.session_state["page"] == "GSD":
+    gsd_page()
+elif st.session_state["page"] == "Locator":
+    locator_page()
+elif st.session_state["page"] == "RAG":
+    rag_page()
+elif st.session_state["page"] == "Reports":
+   reports_page()
+
 
-    # -------------------- SITE SUMMARY --------------------
-    with col2:
-        site = st.session_state["sites"][st.session_state["active_site"]]
-        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)
 def soil_classifier_page():
@@ -1325,49 +1386,51 @@ def locator_page():
         pass
     
     # ----------------------------
-    # --- Render the map ---
+    from streamlit_folium import st_folium
+
+    # ... (your earlier dataset loading / EE init code) ...
+    
+    # Render geemap map
     m.to_streamlit(height=600, responsive=True)
     
-    # --- ROI capture ---
-    roi = None
+    # Overlay a folium draw capture using st_folium on same map (transparent)
+    # Note: we re-render using st_folium to capture geometry interactions.
+    # We get a “last_active_drawing” from the st_folium output.
     
-    # Case 1: New drawing in this run
-    if m.user_roi:
-        try:
-            roi = m.user_roi
-            st.session_state["roi_geojson"] = roi.toGeoJSONString()
-            st.success("✅ ROI captured from user drawing")
-        except Exception as e:
-            st.warning(f"Could not parse m.user_roi: {e}")
+    # Convert your geemap map to a folium map object for st_folium
+    fol_map = m.folium_map  # geemap.foliumap.Map wraps a base folium.Map
+    result = st_folium(fol_map, width=700, height=600, returned_objects=["last_active_drawing"])
     
-    elif m.draw_features:
-        try:
-            last_feature = m.draw_features[-1]
-            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 from draw_features")
-        except Exception as e:
-            st.warning(f"Could not parse m.draw_features: {e}")
+    # Attempt to get ROI from drawn feature
+    roi = None
+    if result and "last_active_drawing" in result and result["last_active_drawing"]:
+        feat = result["last_active_drawing"]
+        geom = feat.get("geometry")
+        if geom:
+            try:
+                roi = ee.Geometry(geom)
+                st.session_state["roi_geojson"] = feat
+                st.success("✅ ROI captured!")
+            except Exception as e:
+                st.error(f"Failed to convert drawn geometry to ee.Geometry: {e}")
     
-    # Case 2: Restore ROI from session_state if nothing new drawn
-    elif "roi_geojson" in st.session_state:
+    # Restore from session if none drawn now
+    if roi is None and "roi_geojson" in st.session_state:
+        saved = st.session_state["roi_geojson"]
         try:
-            stored = json.loads(st.session_state["roi_geojson"])
-            geom = stored.get("geometry") if isinstance(stored, dict) else stored
-            roi = ee.Geometry(geom)
-            st.info("♻️ ROI restored from session state")
+            geom = saved.get("geometry")
+            if geom:
+                roi = ee.Geometry(geom)
+                st.info("♻️ ROI restored from earlier session")
         except Exception as e:
             st.warning(f"Could not restore ROI: {e}")
     
-    # --- Button to trigger computation ---
+    # Then your “Compute Summaries” button logic:
     if st.button("Compute Summaries"):
         if roi is None:
             st.error("⚠️ No ROI found. Please draw a polygon/rectangle/circle and try again.")
         else:
-            st.success("🚀 ROI found — computing summaries...")
-          
-
+            st.success("ROI ready — performing computations...")
 
         chosen_soil_band = None
         if soil_img:
@@ -1550,6 +1613,7 @@ def rag_page():
 # -------------------
 # Report fields (still needed in reports_page)
 # -------------------
+
 REPORT_FIELDS = [
     ("Load Bearing Capacity", "kPa or psf"),
     ("Skin Shear Strength", "kPa"),
@@ -1579,172 +1643,209 @@ from reportlab.lib import colors
 from reportlab.lib.pagesizes import A4
 from reportlab.lib.units import mm
 
-# -------------------------------
-# PDF Builder
-# -------------------------------
+# =============================
+# LLM Helper (Groq API)
+# =============================
+import requests, json, os
+import streamlit as st
+from datetime import datetime
+import tempfile
+from typing import Dict, Any, Optional, List
+from reportlab.platypus import (
+    SimpleDocTemplate, Paragraph, Spacer, PageBreak, Table, TableStyle, Image as RLImage
+)
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+from reportlab.lib import colors
+from reportlab.lib.pagesizes import A4
+from reportlab.lib.units import mm
+
+
+def groq_llm_analyze(prompt: str, section_title: str,
+                     model_name: str = None,
+                     max_tokens: int = 500) -> str:
+    """
+    Query Groq API for a humanized explanation/analysis of one report section.
+    Uses global GROQ_API_KEY and llm_model.
+    """
+    # model
+    if not model_name:
+        model_name = st.session_state.get("llm_model", "groq_compound")
+
+    # key
+    key = None
+    if "GROQ_API_KEY" in st.secrets:
+        key = st.secrets["GROQ_API_KEY"]
+    else:
+        key = st.session_state.get("GROQ_API_KEY")
+
+    if not key:
+        return f"[LLM unavailable] {section_title}: {prompt[:200]}..."
+
+    url = "https://api.groq.com/v1/chat/completions"
+    headers = {"Authorization": f"Bearer {key}", "Content-Type": "application/json"}
+    payload = {
+        "model": model_name,
+        "messages": [
+            {"role": "system", "content": "You are GeoMate, a geotechnical engineering assistant. Respond professionally with concise analysis and insights."},
+            {"role": "user", "content": f"Section: {section_title}\n\nInput: {prompt}\n\nWrite a professional engineering analysis for this section."}
+        ],
+        "temperature": 0.2,
+        "max_tokens": max_tokens,
+    }
+    try:
+        resp = requests.post(url, headers=headers, json=payload, timeout=60)
+        resp.raise_for_status()
+        data = resp.json()
+        return data["choices"][0]["message"]["content"].strip()
+    except Exception as e:
+        return f"[LLM error in {section_title}] {e}"
+
+
+# =============================
+# Build Full Geotechnical Report
+# =============================
 def build_full_geotech_pdf(
     site: Dict[str, Any],
     filename: str,
     include_map_image: Optional[bytes] = None,
     ext_refs: Optional[List[str]] = None
 ):
-    """
-    Build a professional PDF report using site data + references.
-    """
     styles = getSampleStyleSheet()
-    title_style = ParagraphStyle(
-        "title", parent=styles["Title"], fontSize=20, alignment=1,
-        textColor=colors.HexColor("#FF7A00")
-    )
-    h1 = ParagraphStyle(
-        "h1", parent=styles["Heading1"], fontSize=14,
-        textColor=colors.HexColor("#1F4E79"), spaceAfter=6
-    )
+    title_style = ParagraphStyle("title", parent=styles["Title"], fontSize=22,
+                                 alignment=1, textColor=colors.HexColor("#FF6600"), spaceAfter=12)
+    h1 = ParagraphStyle("h1", parent=styles["Heading1"], fontSize=14,
+                        textColor=colors.HexColor("#1F4E79"), spaceBefore=10, spaceAfter=6)
     body = ParagraphStyle("body", parent=styles["BodyText"], fontSize=10.5, leading=13)
-    bullet = ParagraphStyle("bullet", parent=body, leftIndent=12, bulletIndent=6)
 
-    doc = SimpleDocTemplate(
-        filename, pagesize=A4,
-        leftMargin=18*mm, rightMargin=18*mm,
-        topMargin=18*mm, bottomMargin=18*mm
-    )
+    doc = SimpleDocTemplate(filename, pagesize=A4,
+                            leftMargin=18*mm, rightMargin=18*mm,
+                            topMargin=18*mm, bottomMargin=18*mm)
     elems = []
 
-    # --- Title page ---
+    # Title Page
     elems.append(Paragraph("GEOTECHNICAL INVESTIGATION REPORT", title_style))
     elems.append(Spacer(1, 12))
-    company = site.get("Company Name", "Client / Company: Not provided")
-    contact = site.get("Company Contact", "")
-    elems.append(Paragraph(f"<b>{company}</b>", body))
-    if contact:
-        elems.append(Paragraph(contact, body))
-    elems.append(Spacer(1, 12))
+    elems.append(Paragraph(f"<b>Client:</b> {site.get('Company Name','-')}", body))
+    elems.append(Paragraph(f"<b>Contact:</b> {site.get('Company Contact','-')}", body))
     elems.append(Paragraph(f"<b>Project:</b> {site.get('Project Name','-')}", body))
     elems.append(Paragraph(f"<b>Site:</b> {site.get('Site Name','-')}", body))
     elems.append(Paragraph(f"<b>Date:</b> {datetime.today().strftime('%Y-%m-%d')}", body))
     elems.append(PageBreak())
 
-    # --- Table of contents ---
+    # TOC
     elems.append(Paragraph("TABLE OF CONTENTS", h1))
     toc_items = [
-        "1.0 Introduction",
-        "2.0 Site description and geology",
-        "3.0 Field investigation & laboratory testing",
-        "4.0 Evaluation of geotechnical properties",
-        "5.0 Provisional site classification",
-        "6.0 Recommendations",
-        "7.0 LLM Analysis",
-        "8.0 Figures & Tables",
-        "9.0 Appendices & References"
+        "1.0 Summary", "2.0 Introduction", "3.0 Site Description & Geology",
+        "4.0 Field & Laboratory Testing", "5.0 Evaluation of Geotechnical Properties",
+        "6.0 Provisional Classification", "7.0 Recommendations",
+        "8.0 LLM Analysis", "9.0 Figures & Tables", "10.0 Appendices & References"
     ]
-    for i, t in enumerate(toc_items, start=1):
+    for i, t in enumerate(toc_items, 1):
         elems.append(Paragraph(f"{i}. {t}", body))
     elems.append(PageBreak())
 
-    # --- Summary ---
-    elems.append(Paragraph("SUMMARY", h1))
-    summary_bullets = [
-        f"Site: {site.get('Site Name','-')}.",
-        f"General geology: {site.get('Soil Profile','Not provided')}.",
-        f"Key lab tests: {', '.join([r.get('sampleId','') for r in site.get('Laboratory Results',[])]) if site.get('Laboratory Results') else 'No lab results provided.'}",
-        f"Classification: USCS = {site.get('USCS','Not provided')}; AASHTO = {site.get('AASHTO','Not provided')}.",
-        "Primary recommendation: See Recommendations section."
-    ]
-    for s in summary_bullets:
-        elems.append(Paragraph(f"• {s}", bullet))
+    # Sections with LLM calls
+    elems.append(Paragraph("1.0 SUMMARY", h1))
+    elems.append(Paragraph(groq_llm_analyze(json.dumps(site, indent=2), "Summary"), body))
     elems.append(PageBreak())
 
-    # --- Introduction ---
-    elems.append(Paragraph("1.0 INTRODUCTION", h1))
-    intro_text = site.get("Project Description", "Project description not provided.")
-    elems.append(Paragraph(intro_text, body))
-
-    # --- Site description & geology ---
-    elems.append(Paragraph("2.0 SITE DESCRIPTION AND GEOLOGY", h1))
-    site_geo = [
-        f"Topography: {site.get('Topography','Not provided')}",
-        f"Drainage: {site.get('Drainage','Not provided')}",
-        f"Current land use: {site.get('Current Land Use','Not provided')}",
-        f"Regional geology: {site.get('Regional Geology','Not provided')}"
-    ]
-    for t in site_geo:
-        elems.append(Paragraph(t, body))
+    elems.append(Paragraph("2.0 INTRODUCTION", h1))
+    elems.append(Paragraph(groq_llm_analyze(site.get("Project Description",""), "Introduction"), body))
+
+    elems.append(Paragraph("3.0 SITE DESCRIPTION & GEOLOGY", h1))
+    geology_text = f"Topo: {site.get('Topography')}, Drainage: {site.get('Drainage')}, Land Use: {site.get('Current Land Use')}, Geology: {site.get('Regional Geology')}"
+    elems.append(Paragraph(groq_llm_analyze(geology_text, "Geology & Site Description"), body))
     elems.append(PageBreak())
 
-    # --- Field & lab testing ---
-    elems.append(Paragraph("3.0 FIELD INVESTIGATION & LABORATORY TESTING", h1))
-    if site.get("Field Investigation"):
-        for item in site["Field Investigation"]:
-            elems.append(Paragraph(f"<b>{item.get('id','Test')}</b> — depth {item.get('depth','-')}", body))
-            for layer in item.get("layers", []):
-                elems.append(Paragraph(f"- {layer.get('depth','')} : {layer.get('description','')}", body))
-    else:
-        elems.append(Paragraph("No field investigation data supplied.", body))
-
-    lab_rows = site.get("Laboratory Results", [])
-    if lab_rows:
-        elems.append(Spacer(1, 6))
-        elems.append(Paragraph("Laboratory Results", h1))
-        data = [["Sample ID","Material","LL","PI","Linear Shrinkage","%Clay","%Silt","%Sand","%Gravel","Expansiveness"]]
-        for r in lab_rows:
-            data.append([
-                r.get("sampleId","-"), r.get("material","-"),
-                str(r.get("liquidLimit","-")), str(r.get("plasticityIndex","-")),
-                str(r.get("linearShrinkage","-")), str(r.get("percentClay","-")),
-                str(r.get("percentSilt","-")), str(r.get("percentSand","-")),
-                str(r.get("percentGravel","-")), r.get("potentialExpansiveness","-")
-            ])
-        t = Table(data, repeatRows=1, colWidths=[40*mm,40*mm,18*mm,18*mm,22*mm,20*mm,20*mm,20*mm,20*mm,30*mm])
-        t.setStyle(TableStyle([
-            ('BACKGROUND',(0,0),(-1,0),colors.HexColor("#1F4E79")),
-            ('TEXTCOLOR',(0,0),(-1,0),colors.white),
-            ('GRID',(0,0),(-1,-1),0.4,colors.grey),
-            ('BOX',(0,0),(-1,-1),1,colors.HexColor("#FF7A00"))
-        ]))
-        elems.append(t)
+    elems.append(Paragraph("4.0 FIELD & LABORATORY TESTING", h1))
+    elems.append(Paragraph(groq_llm_analyze(json.dumps(site.get('Laboratory Results',[]), indent=2), "Field & Lab Testing"), body))
     elems.append(PageBreak())
 
-    # --- Evaluation & classification ---
-    elems.append(Paragraph("4.0 EVALUATION OF GEOTECHNICAL PROPERTIES", h1))
-    elems.append(Paragraph(site.get("Evaluation","Evaluation not provided."), body))
-    elems.append(Paragraph("5.0 PROVISIONAL SITE CLASSIFICATION", h1))
-    elems.append(Paragraph(site.get("Provisional Classification","Not provided."), body))
-    elems.append(Paragraph("6.0 RECOMMENDATIONS", h1))
-    elems.append(Paragraph(site.get("Recommendations","Not provided."), body))
-
-    # --- LLM Analysis ---
-    elems.append(Paragraph("7.0 LLM ANALYSIS (GeoMate)", h1))
-    llm_text = site.get("LLM_Report_Text", None)
-    if llm_text:
-        elems.append(Paragraph(llm_text.replace("\n","\n\n"), body))
-    else:
-        elems.append(Paragraph("No LLM analysis saved for this site.", body))
+    elems.append(Paragraph("5.0 EVALUATION OF GEOTECHNICAL PROPERTIES", h1))
+    elems.append(Paragraph(groq_llm_analyze(json.dumps(site, indent=2), "Evaluation of Properties"), body))
 
-    # --- Map snapshot ---
-    if include_map_image:
-        try:
-            tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
-            tmp.write(include_map_image)
-            tmp.flush()
-            elems.append(PageBreak())
-            elems.append(Paragraph("Map Snapshot", h1))
-            elems.append(RLImage(tmp.name, width=160*mm, height=90*mm))
-        except Exception:
-            pass
+    elems.append(Paragraph("6.0 PROVISIONAL CLASSIFICATION", h1))
+    class_text = f"USCS={site.get('USCS')}, AASHTO={site.get('AASHTO')}"
+    elems.append(Paragraph(groq_llm_analyze(class_text, "Soil Classification"), body))
 
-    # --- References ---
+    elems.append(Paragraph("7.0 RECOMMENDATIONS", h1))
+    elems.append(Paragraph(groq_llm_analyze(json.dumps(site, indent=2), "Recommendations"), body))
+
+    elems.append(Paragraph("8.0 LLM ANALYSIS (GeoMate)", h1))
+    elems.append(Paragraph(groq_llm_analyze(json.dumps(site, indent=2), "LLM Insights"), body))
+
+    # Map snapshot
+    if include_map_image:
+        tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
+        tmp.write(include_map_image)
+        tmp.flush()
+        elems.append(PageBreak())
+        elems.append(Paragraph("9.0 MAP SNAPSHOT", h1))
+        elems.append(RLImage(tmp.name, width=160*mm, height=90*mm))
+
+    # References
     elems.append(PageBreak())
-    elems.append(Paragraph("9.0 APPENDICES & REFERENCES", h1))
+    elems.append(Paragraph("10.0 REFERENCES", h1))
     if ext_refs:
         for r in ext_refs:
             elems.append(Paragraph(f"- {r}", body))
     else:
-        elems.append(Paragraph("- No external references provided.", body))
+        elems.append(Paragraph("No external references provided.", body))
+
+    doc.build(elems)
+    return filename
+
+
+# =============================
+# Build Classification Report
+# =============================
+def build_classification_pdf(
+    site: Dict[str, Any],
+    classification: Dict[str, Any],
+    filename: str
+):
+    styles = getSampleStyleSheet()
+    title_style = ParagraphStyle("title", parent=styles["Title"], fontSize=18,
+                                 textColor=colors.HexColor("#FF6600"), alignment=1)
+    h1 = ParagraphStyle("h1", parent=styles["Heading1"], fontSize=12, textColor=colors.HexColor("#1F4E79"))
+    body = ParagraphStyle("body", parent=styles["BodyText"], fontSize=10)
+
+    doc = SimpleDocTemplate(filename, pagesize=A4,
+                            leftMargin=18*mm, rightMargin=18*mm,
+                            topMargin=18*mm, bottomMargin=18*mm)
+    elems = []
+
+    # Title Page
+    elems.append(Paragraph("SOIL CLASSIFICATION REPORT", title_style))
+    elems.append(Spacer(1, 12))
+    elems.append(Paragraph(f"Site: {site.get('Site Name','Unnamed')}", body))
+    elems.append(Paragraph(f"Date: {datetime.today().strftime('%Y-%m-%d')}", body))
+    elems.append(PageBreak())
+
+    # Sections
+    elems.append(Paragraph("1.0 DETERMINISTIC RESULTS", h1))
+    elems.append(Paragraph(groq_llm_analyze(json.dumps(classification, indent=2), "Deterministic Results"), body))
+    elems.append(PageBreak())
+
+    elems.append(Paragraph("2.0 ENGINEERING CHARACTERISTICS", h1))
+    elems.append(Paragraph(groq_llm_analyze(json.dumps(classification.get('engineering_characteristics',{}), indent=2), "Engineering Characteristics"), body))
+    elems.append(PageBreak())
+
+    elems.append(Paragraph("3.0 DECISION PATHS", h1))
+    dp_text = f"USCS path: {classification.get('USCS_decision_path')}, AASHTO path: {classification.get('AASHTO_decision_path')}"
+    elems.append(Paragraph(groq_llm_analyze(dp_text, "Decision Paths"), body))
+    elems.append(PageBreak())
+
+    elems.append(Paragraph("4.0 LLM ANALYSIS", h1))
+    elems.append(Paragraph(groq_llm_analyze(json.dumps(classification, indent=2), "LLM Analysis"), body))
 
     doc.build(elems)
     return filename
 
 
+# -------------------------------
+# Reports Page
+# -------------------------------
 # -------------------------------
 # Reports Page
 # -------------------------------
@@ -1758,31 +1859,20 @@ def reports_page():
         st.markdown("You have a saved classification for this site.")
         if st.button("Generate Classification PDF"):
             fname = f"classification_{site['Site Name'].replace(' ','_')}.pdf"
-            buffer = io.BytesIO()
-            doc = SimpleDocTemplate(buffer, pagesize=A4)
-            elems = []
-            elems.append(Paragraph("Soil Classification Report", getSampleStyleSheet()['Title']))
-            elems.append(Spacer(1,6))
-            elems.append(Paragraph(f"Site: {site.get('Site Name')}", getSampleStyleSheet()['Normal']))
-            elems.append(Spacer(1,6))
-            elems.append(Paragraph("Classification result:", getSampleStyleSheet()['Heading2']))
-            elems.append(Paragraph(site.get("classifier_decision","-"), getSampleStyleSheet()['BodyText']))
-
-            # Add FAISS citations if present in rag_history
+
+            # Collect references (rag_history + manual ones later)
+            refs = []
             if "rag_history" in st.session_state and site.get("Site ID") in st.session_state["rag_history"]:
-                refs = []
                 for h in st.session_state["rag_history"][site["Site ID"]]:
-                    if h["who"]=="bot" and "[ref:" in h["text"]:
+                    if h["who"] == "bot" and "[ref:" in h["text"]:
                         for m in re.findall(r"\[ref:([^\]]+)\]", h["text"]):
                             refs.append(m)
-                if refs:
-                    elems.append(Spacer(1,12))
-                    elems.append(Paragraph("References:", getSampleStyleSheet()['Heading2']))
-                    for r in set(refs):
-                        elems.append(Paragraph(f"- {r}", getSampleStyleSheet()['Normal']))
 
-            doc.build(elems)
+            # Build classification PDF
+            buffer = io.BytesIO()
+            build_classification_pdf(site, site.get("classifier_decision"), buffer)
             buffer.seek(0)
+
             st.download_button("Download Classification PDF", buffer, file_name=fname, mime="application/pdf")
     else:
         st.info("No classification saved for this site yet. Use the Classifier page.")
@@ -1790,14 +1880,14 @@ def reports_page():
     # ---------------- Quick Report Form ----------------
     st.markdown("### Quick report form (edit values and request LLM analysis)")
     with st.form(key="report_quick_form"):
-        cols = st.columns([2,1,1])
+        cols = st.columns([2, 1, 1])
         cols[0].markdown("**Parameter**")
         cols[1].markdown("**Value**")
         cols[2].markdown("**Unit / Notes**")
 
         inputs = {}
         for (fld, unit) in REPORT_FIELDS:
-            c1, c2, c3 = st.columns([2,1,1])
+            c1, c2, c3 = st.columns([2, 1, 1])
             c1.markdown(f"**{fld}**")
             default_val = site.get(fld, "")
             inputs[fld] = c2.text_input(fld, value=str(default_val), label_visibility="collapsed", key=f"quick_{fld}")
@@ -1819,22 +1909,23 @@ def reports_page():
             "site_summary": {
                 "USCS": site.get("USCS"), "AASHTO": site.get("AASHTO"), "GI": site.get("GI"),
                 "Soil Profile": site.get("Soil Profile"),
-                "Key lab results": [r.get("sampleId") for r in site.get("Laboratory Results",[])]
+                "Key lab results": [r.get("sampleId") for r in site.get("Laboratory Results", [])]
             },
-            "inputs": {fld: site.get(fld,"Not provided") for fld,_ in REPORT_FIELDS}
+            "inputs": {fld: site.get(fld, "Not provided") for fld, _ in REPORT_FIELDS}
         }
         prompt = (
             "You are GeoMate AI, an engineering assistant. Given the following site context and "
             "engineering parameters (some may be 'Not provided'), produce:\n1) short executive summary, "
             "2) geotechnical interpretation (classification, key risks), 3) recommended remedial/improvement "
             "options and 4) short design notes. Provide any numeric outputs in the format [[FIELD: value unit]].\n\n"
-            f"Context: {json.dumps(context)}\n\nAnswer concisely and professionally."
+            f"Context: {json.dumps(context)}"
         )
-        resp = groq_generate(prompt, model=st.session_state["llm_model"], max_tokens=600)
-        
+        resp = groq_llm_analyze(prompt, section_title="GeoMate Analysis")
+
         st.markdown("**GeoMate analysis**")
         st.markdown(resp)
 
+        # extract values like [[Bearing Capacity: 150 kN/m2]]
         matches = re.findall(r"\[\[([A-Za-z0-9 _/-]+):\s*([0-9.+-eE]+)\s*([A-Za-z%\/]*)\]\]", resp)
         for m in matches:
             field, val, unit = m[0].strip(), m[1].strip(), m[2].strip()
@@ -1858,16 +1949,16 @@ def reports_page():
     state = site.get("report_convo_state", -1)
     if state >= 0:
         st.markdown("Chatbot will ask for missing fields. You can answer or type 'skip' to leave blank.")
-        show_table = [(f, site.get(f, "Not provided")) for f,_ in REPORT_FIELDS]
+        show_table = [(f, site.get(f, "Not provided")) for f, _ in REPORT_FIELDS]
         st.table(show_table)
 
         if state < len(REPORT_FIELDS):
             field, unit = REPORT_FIELDS[state]
             ans = st.text_input(f"GeoMate — Please provide '{field}' ({unit})", key=f"report_in_{state}")
-            c1, c2 = st.columns([1,1])
+            c1, c2 = st.columns([1, 1])
             with c1:
                 if st.button("Submit", key=f"report_submit_{state}"):
-                    site[field] = ans.strip() if ans.strip() not in ("skip","don't know","dont know","na","n/a","") else "Not provided"
+                    site[field] = ans.strip() if ans.strip() not in ("skip", "don't know", "dont know", "na", "n/a", "") else "Not provided"
                     site["report_convo_state"] = state + 1
                     st.rerun()
             with c2:
@@ -1883,7 +1974,7 @@ def reports_page():
             faiss_refs = []
             if "rag_history" in st.session_state and site.get("Site ID") in st.session_state["rag_history"]:
                 for h in st.session_state["rag_history"][site["Site ID"]]:
-                    if h["who"]=="bot" and "[ref:" in h["text"]:
+                    if h["who"] == "bot" and "[ref:" in h["text"]:
                         for m in re.findall(r"\[ref:([^\]]+)\]", h["text"]):
                             faiss_refs.append(m)
             all_refs = list(set(ext_refs + faiss_refs))
@@ -1895,7 +1986,7 @@ def reports_page():
 
             with open(outname, "rb") as f:
                 st.download_button("Download Full Geotechnical Report", f, file_name=outname, mime="application/pdf")
-                
+
 # 8) Page router
 if "page" not in st.session_state:
     st.session_state["page"] = "Home"