Update app.py

app.py

@@ -1736,14 +1736,16 @@ def reports_page():
     st.header("📑 Reports — Classification & Full Geotechnical")
     site = st.session_state["sites"][st.session_state["active_site"]]
 
-    # ---------------- Classification Report ----------------
-    st.subheader("Classification-only report")
+    # =====================================================
+    # Classification Report
+    # =====================================================
+    st.subheader("📘 Classification-only Report")
     if site.get("classifier_decision"):
         st.markdown("You have a saved classification for this site.")
         if st.button("Generate Classification PDF"):
             fname = f"classification_{site['Site Name'].replace(' ','_')}.pdf"
 
-            # Collect references (rag_history + manual ones later)
+            # Collect references from rag_history
             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"]]:
@@ -1756,12 +1758,14 @@ def reports_page():
             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")
+            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.")
 
-    # ---------------- Quick Report Form ----------------
-    st.markdown("### Quick report form (edit values and request LLM analysis)")
+    # =====================================================
+    # 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[0].markdown("**Parameter**")
@@ -1781,10 +1785,12 @@ def reports_page():
             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.")
+            st.success("✅ Saved quick report values to active site.")
 
-    # ---------------- LLM Analysis ----------------
-    st.markdown("#### LLM-powered analysis")
+    # =====================================================
+    # LLM Analysis (Humanized Report Text)
+    # =====================================================
+    st.markdown("#### 🤖 LLM-powered analysis")
     if st.button("Ask GeoMate (generate analysis & recommendations)"):
         context = {
             "site_name": site.get("Site Name"),
@@ -1808,7 +1814,7 @@ def reports_page():
         st.markdown("**GeoMate analysis**")
         st.markdown(resp)
 
-        # extract values like [[Bearing Capacity: 150 kN/m2]]
+        # Extract structured values from [[FIELD: value unit]]
         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()
@@ -1820,55 +1826,35 @@ def reports_page():
                 site["Relative Compaction"] = f"{val} {unit}"
 
         site["LLM_Report_Text"] = resp
-        st.success("LLM analysis saved to site under 'LLM_Report_Text'.")
+        st.success("✅ LLM analysis saved to site under 'LLM_Report_Text'.")
 
-    # ---------------- Full Report Chatbot ----------------
+    # =====================================================
+    # Full Geotechnical Report
+    # =====================================================
     st.markdown("---")
-    st.subheader("Full Geotechnical Report (chatbot will gather missing fields)")
-    if st.button("Start Report Chatbot"):
-        st.session_state["sites"][st.session_state["active_site"]]["report_convo_state"] = 0
-        st.rerun()
-
-    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]
-        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])
-            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["report_convo_state"] = state + 1
-                    st.rerun()
-            with c2:
-                if st.button("Skip", key=f"report_skip_{state}"):
-                    site[field] = "Not provided"
-                    site["report_convo_state"] = state + 1
-                    st.rerun()
-        else:
-            st.success("All report questions asked. You can generate the full report now.")
-            ext_ref_text = st.text_area("Optional: External references (one per line)", value="")
-            ext_refs = [r.strip() for r in ext_ref_text.splitlines() if r.strip()]
+    st.subheader("📕 Full Geotechnical Report")
 
-            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"]:
-                        for m in re.findall(r"\[ref:([^\]]+)\]", h["text"]):
-                            faiss_refs.append(m)
-            all_refs = list(set(ext_refs + faiss_refs))
+    ext_ref_text = st.text_area("Optional: External references (one per line)", value="")
+    ext_refs = [r.strip() for r in ext_ref_text.splitlines() if r.strip()]
+
+    # Add FAISS / rag references
+    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"]:
+                for m in re.findall(r"\[ref:([^\]]+)\]", h["text"]):
+                    faiss_refs.append(m)
+    all_refs = list(set(ext_refs + faiss_refs))
 
-            outname = f"Full_Geotech_Report_{site.get('Site Name','site')}.pdf"
-            mapimg = site.get("map_snapshot")
+    if st.button("Generate Full Geotechnical Report PDF"):
+        outname = f"Full_Geotech_Report_{site.get('Site Name','site')}.pdf"
+        mapimg = site.get("map_snapshot")
 
-            build_full_geotech_pdf(site, outname, include_map_image=mapimg, ext_refs=all_refs)
+        # ✅ Classification results also included inside full report
+        build_full_geotech_pdf(site, outname, include_map_image=mapimg, ext_refs=all_refs)
 
-            with open(outname, "rb") as f:
-                st.download_button("Download Full Geotechnical Report", f, file_name=outname, mime="application/pdf")
+        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: