Update app.py

app.py

@@ -23,6 +23,13 @@ from reportlab.lib.units import mm
 from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle, Image as RLImage, PageBreak
 from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
 
+# Ensure icon fonts load (fix desktop icon display for option_menu)
+st.markdown("""
+<!-- Load icon fonts used by streamlit_option_menu -->
+<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bootstrap-icons@1.11.3/font/bootstrap-icons.css">
+<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/css/all.min.css">
+""", unsafe_allow_html=True)
+
 # Optional imports handled gracefully
 try:
     import geemap
@@ -485,6 +492,71 @@ def build_full_geotech_pdf(site: Dict[str, Any], filename: str, include_map_imag
     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)
     elems = []
+        # --- Title page (collect company/user info from site if present) ---
+    elems.append(Paragraph("GEOTECHNICAL INVESTIGATION REPORT", title_style))
+    elems.append(Spacer(1,12))
+    # Company / client block (if present)
+    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>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 (simple listing with section titles) ---
+    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 Figures & Tables",
+        "8.0 Appendices & References"
+    ]
+    for i, t in enumerate(toc_items, start=1):
+        elems.append(Paragraph(f"{i}. {t}", body))
+    elems.append(PageBreak())
+
+    # --- (existing summary and subsequent sections as before) ---
+    elems.append(Paragraph("SUMMARY", h1))
+    ...
+    # after '6.0 RECOMMENDATIONS' insertion, include LLM analysis if present
+    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))
+
+    # --- Figures/Tables index (simple lists from site data, if present) ---
+    elems.append(PageBreak())
+    elems.append(Paragraph("List of Tables & Figures", h1))
+    # Example: list lab samples and any GSD figure
+    if site.get("Laboratory Results"):
+        elems.append(Paragraph("Tables:", body))
+        for r in site.get("Laboratory Results", []):
+            elems.append(Paragraph(f"- Laboratory sample: {r.get('sampleId','-')}", body))
+    if site.get("GSD"):
+        elems.append(Paragraph("- GSD Curve (see section 3)", body))
+
+    # --- Appendices & References ---
+    elems.append(PageBreak())
+    elems.append(Paragraph("Appendices", h1))
+    elems.append(Paragraph("A. Test certificates and raw data (provided as CSV or appendices).", body))
+    elems.append(PageBreak())
+    elems.append(Paragraph("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))
+
     # Cover
     elems.append(Paragraph("GEOTECHNICAL INVESTIGATION REPORT", title_style))
     elems.append(Spacer(1,6))
@@ -839,6 +911,31 @@ with st.sidebar:
 
 # 7) Pages implementation
 def landing_page():
+    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)
+    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;
+        position: relative;
+        ">
+      <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.
+        </p>
+        <div style='margin-top:8px; color:#cfcfcf; font-size:13px'>
+          Quick: Classifier • GSD • Locator • RAG • 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>"
@@ -1386,7 +1483,17 @@ def locator_page():
         pass
 
     # Enable drawing
-    m.add_draw_control(polyline=False, circle=False, circlemarker=False, rectangle=True, polygon=True)
+    from geemap import Draw
+    # Add draw control (rectangle + polygon only)
+    draw_control = Draw(
+    export=False,
+    draw_polygon=True,
+    draw_rectangle=True,
+    draw_circle=False,
+    draw_polyline=False,
+    draw_marker=False
+    )
+    m.add_control(draw_control)
     m.to_streamlit(height=700, responsive=True)
     st.markdown("👉 Draw a polygon/rectangle on the map (draw tool). After drawing click **Compute Summaries**.")
 
@@ -1690,19 +1797,19 @@ def rag_page():
                     site["Relative Compaction"] = f"{val} {unit}"
             st.success("Response saved and any recognized numeric fields auto-stored in the site data.")
 
-# Reports page — conversational missing-parameter bot & PDF generation
-REPORT_FIELDS = [
-    ("Load Bearing Capacity","kPa or psf"),
-    ("Skin Shear Strength","kPa"),
-    ("Relative Compaction","%"),
-    ("Rate of Consolidation","mm/yr or days"),
-    ("Nature of Construction","text"),
-    ("Borehole Count","number"),
-    ("Max Depth (m)","m"),
-    ("SPT N (avg)","blows/ft"),
-    ("CBR (%)","%"),
-    ("Allowable Bearing (kPa)","kPa")
-]
+        # Reports page — conversational missing-parameter bot & PDF generation
+        REPORT_FIELDS = [
+            ("Load Bearing Capacity","kPa or psf"),
+            ("Skin Shear Strength","kPa"),
+            ("Relative Compaction","%"),
+            ("Rate of Consolidation","mm/yr or days"),
+            ("Nature of Construction","text"),
+            ("Borehole Count","number"),
+            ("Max Depth (m)","m"),
+            ("SPT N (avg)","blows/ft"),
+            ("CBR (%)","%"),
+            ("Allowable Bearing (kPa)","kPa")
+        ]
 
 def reports_page():
     st.header("📑 Reports — Classification & Full Geotechnical")
@@ -1729,6 +1836,77 @@ def reports_page():
     else:
         st.info("No classification saved for this site yet. Use the Classifier page.")
 
+    # --------- Dynamic form for quick report inputs + LLM analysis ----------
+    st.markdown("### Quick report form (edit values and request LLM analysis)")
+    site = st.session_state["sites"][st.session_state["active_site"]]
+
+    # Build form-style table
+    with st.form(key="report_quick_form"):
+    cols = st.columns([2,1,1])  # name, value, unit/notes
+    # header row visually
+    cols[0].markdown("**Parameter**")
+    cols[1].markdown("**Value**")
+    cols[2].markdown("**Unit / Notes**")
+
+    # build inputs dynamically from REPORT_FIELDS
+    inputs = {}
+    for (fld, unit) in REPORT_FIELDS:
+        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), key=f"quick_{fld}")
+        c3.markdown(unit)
+
+    submitted = st.form_submit_button("Save values to site")
+    if submitted:
+        for fld, _ in REPORT_FIELDS:
+            val = inputs.get(fld, "").strip()
+            site[fld] = val if val != "" else "Not provided"
+        st.success("Saved quick report values to active site.")
+
+    # LLM analysis button
+    st.markdown("#### LLM-powered analysis")
+    if st.button("Ask GeoMate (generate analysis & recommendations)"):
+    # prepare context for the LLM from the site
+    context = {
+        "site_name": site.get("Site Name"),
+        "project": site.get("Project Name"),
+        "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",[])]
+        },
+        "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."
+    )
+    resp = groq_generate(prompt, model=st.session_state["llm_model"], max_tokens=600)
+    # display
+    st.markdown("**GeoMate analysis**")
+    st.markdown(resp)
+    # try to extract numeric fields (same bracket format as elsewhere)
+    import re
+    matches = re.findall(r"\[\[([A-Za-z0-9 _/-]+):\s*([0-9.+-eE]+)\s*([A-Za-z%\/]*)\]\]", resp)
+    for m in matches:
+        field = m[0].strip()
+        val = m[1].strip()
+        unit = m[2].strip()
+        # map likely field names:
+        if "bearing" in field.lower():
+            site["Load Bearing Capacity"] = f"{val} {unit}"
+        elif "skin" in field.lower():
+            site["Skin Shear Strength"] = f"{val} {unit}"
+        elif "compaction" in field.lower():
+            site["Relative Compaction"] = f"{val} {unit}"
+    # store the analysis text so it can be included in the PDF later
+    site["LLM_Report_Text"] = resp
+    st.success("LLM analysis saved to site under 'LLM_Report_Text'.")
+
     st.markdown("---")
     st.subheader("Full Geotechnical Report (chatbot will gather missing fields)")
     if st.button("Start Report Chatbot"):