Update app.py

app.py

@@ -921,4 +921,90 @@ def reports_ui():
         if st.button("Generate Full Geotechnical Report (PDF)"):
             with st.spinner("Building PDF..."):
                 pdf_bytes = build_full_report_pdf(site)
-            st.download_button("Download Report PDF", data=pdf_bytes, file_name=f"GeoMate_FullReport_{site.get('Site Name','site')}.pdf", mime="application/pdf")
+            st.download_button("Download Report PDF", data=pdf_bytes, file_name=f"GeoMate_FullReport_{site.get('Site Name','site')}.pdf", mime="application/pdf")
+# -----------------------------
+# --- GeoMate Ask (RAG chat) - placeholder
+# -----------------------------
+def rag_ui():
+    st.header("🤖 GeoMate Ask — RAG (placeholder)")
+    site = get_active_site()
+    if not site:
+        st.info("Create/select a site first.")
+        return
+    st.markdown("This RAG interface is wired as a placeholder. Connect your FAISS DB and LLM backend to enable retrieval and answers.")
+    q = st.text_input("Ask a geotechnical question (example: 'What foundation would you recommend for this site?')")
+    if st.button("Ask"):
+        user_q = q.strip()
+        if not user_q:
+            st.warning("Enter a question.")
+            return
+        # Save to site chat history
+        site.setdefault("chat_history", []).append(("user", user_q))
+        # Placeholder answer (you should call your LLM here)
+        answer = "Placeholder answer: connect your LLM / FAISS RAG backend to get a real answer. The system will use site data and retrieved documents."
+        site["chat_history"].append(("bot", answer))
+        save_active_site(site)
+        st.markdown(f"**GeoMate:** {answer}")
+
+# -----------------------------
+# --- Soil Recognizer (image classifier placeholder)
+# -----------------------------
+def soil_recognizer_ui():
+    st.header("🖼️ Soil Recognizer")
+    site = get_active_site()
+    if not site:
+        st.info("Create/select a site first.")
+        return
+    uploaded = st.file_uploader("Upload a soil photo (JPG/PNG)", type=["jpg","jpeg","png"])
+    if uploaded:
+        try:
+            im = Image.open(uploaded).convert("RGB")
+            st.image(im, caption="Uploaded image", use_column_width=True)
+            # placeholder classifier: average brightness to pick a class (toy)
+            arr = np.array(im.convert("L"))
+            mean = arr.mean()
+            if mean > 140:
+                pred = "Sand / Light soil"
+            elif mean > 80:
+                pred = "Silt"
+            else:
+                pred = "Clay / Dark soil"
+            st.success(f"Predicted soil type: {pred} (placeholder model)")
+            site["Soil Type (image)"] = pred
+            save_active_site(site)
+        except Exception as e:
+            st.error(f"Image processing failed: {e}")
+
+# -----------------------------
+# --- Main UI Router
+# -----------------------------
+def main_ui():
+    sidebar_ui()
+    page = ss.get("page", "Landing")
+    # use option_menu on sidebar to control as well
+    with st.sidebar:
+        choice = option_menu(None, ["Landing","Soil Recognizer","Soil Classifier","GSD Curve","Locator","GeoMate Ask","Reports"],
+                             icons=["house","image","bar-chart","bar-chart-line","geo-alt","robot","file-earmark-text"], menu_icon="compass", default_index=0)
+        # sync page choice
+        ss["page"] = choice
+    page = ss["page"]
+
+    if page == "Landing":
+        landing_ui()
+    elif page == "Soil Recognizer":
+        soil_recognizer_ui()
+    elif page == "Soil Classifier":
+        soil_classifier_ui()
+    elif page == "GSD Curve":
+        gsd_curve_ui()
+    elif page == "Locator":
+        locator_ui()
+    elif page == "GeoMate Ask":
+        rag_ui()
+    elif page == "Reports":
+        reports_ui()
+    else:
+        st.write("Unknown page.")
+
+if __name__ == "__main__":
+    main_ui()