Update app.py

app.py

@@ -1,11 +1,25 @@
 from dotenv import load_dotenv
-load_dotenv()
-
 import streamlit as st
-import sys, os
+import sys, os, requests, pathlib
+
+# ─── Load Configuration from .env ───────────────────────────────
+def load_config():
+    load_dotenv()
+    config = {
+        "APP_MODE": os.getenv("APP_MODE", "streamlit"),
+        "OLLAMA_MODEL": os.getenv("OLLAMA_MODEL", "llama3"),
+        "HF_API_KEY": os.getenv("HF_API_KEY", ""),
+        "BACKEND_URL": os.getenv("BACKEND_URL", "http://localhost:9000"),
+        "TAIPY_URL": os.getenv("TAIPY_URL", "http://localhost:8080"),
+    }
+    return config
+
+config = load_config()
 
-# ─── Make utils/ globally importable ─────────────────────────────
-sys.path.append(os.path.abspath(os.path.dirname(__file__)))
+# ─── Ensure utils/ importable ───────────────────────────────────
+ROOT_PATH = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+if ROOT_PATH not in sys.path:
+    sys.path.insert(0, ROOT_PATH)
 
 # ─── Page Config ────────────────────────────────────────────────
 st.set_page_config(page_title="Omniscient Dashboard", layout="wide")
@@ -14,46 +28,54 @@ st.set_page_config(page_title="Omniscient Dashboard", layout="wide")
 st.title("🤖 Omniscient LLM Dashboard")
 st.write("Welcome! This Space is running **Streamlit multipage mode** with Taipy integration.")
 
+# ─── Backend Health Check ───────────────────────────────────────
+def backend_status(url):
+    try:
+        res = requests.get(f"{url}/health", timeout=3)
+        return "🟢 Online" if res.ok else "🟠 Unresponsive"
+    except Exception:
+        return "🔴 Offline"
+
 # ─── Metrics Section ────────────────────────────────────────────
-col1, col2, col3 = st.columns(3)
+col1, col2, col3, col4 = st.columns(4)
 
-# Active chat sessions
 active_sessions = len(st.session_state.get("messages", []))
 col1.metric("💬 Active Sessions", active_sessions)
 
-# Uploaded files counter
 uploaded_files = len(st.session_state.get("uploaded_files", []))
 col2.metric("📂 Uploaded Files", uploaded_files)
 
-# Error counter
 errors = len(st.session_state.get("errors", []))
 col3.metric("⚠️ Errors", errors)
 
+col4.metric("🔌 API Backend", backend_status(config["BACKEND_URL"]))
+
 st.success("System ready ✅")
 
-# ─── Sidebar Navigation Guide ───────────────────────────────────
-st.markdown("""
-### 📑 Navigation
-The sidebar is auto-generated by Streamlit multipage apps.  
-We group modules conceptually:
-
-- **Chat**
-  - 💬 Chatbot  
-
-- **Analysis**
-  - 👁️ Omnieye → keyword/file analysis + web scraping  
-  - 📜 Omnilog → log parsing, summaries, syslog tail  
-  - 📂 ScriptDigest → script/documentation generator  
-  - 📚 RAG → document upload + retrieval-augmented Q&A  
-  - 🗂️ BulkDigest → batch multi-file AI summarizer  
-
-- **Tools**
-  - 🌐 Web Scraper → URL fetch & summarize  
-  - 🖥️ SysMonitor → local system metrics from `/proc/`  
-  - 🧪 Example → placeholder demo  
-""")
-
-# ─── Embed Taipy Realtime Dashboard ───────────────────────────
+# ─── Dynamic Sidebar Navigation ─────────────────────────────────
+st.sidebar.header("📑 Navigation")
+
+pages_path = os.path.join(os.path.dirname(__file__), "pages")
+if os.path.isdir(pages_path):
+    st.sidebar.write("**Modules:**")
+    pages = sorted([p.stem for p in pathlib.Path(pages_path).glob("*.py")])
+    for page in pages:
+        st.sidebar.markdown(f"- {page}")
+
+st.sidebar.markdown("---")
+st.sidebar.info("🧠 Omniscient Framework v0.1 — AI-driven OSINT & DFIR Hub")
+
+# ─── Session Debug (optional) ───────────────────────────────────
+with st.expander("🧩 Session Diagnostics", expanded=False):
+    st.json({k: v for k, v in st.session_state.items() if len(str(v)) < 1000})
+
+# ─── Embed Taipy Realtime Dashboard ─────────────────────────────
 st.subheader("📊 Realtime System Dashboard (Taipy)")
-st.write("Live CPU and Memory metrics via Taipy GUI:")
-st.components.v1.iframe("http://localhost:8080", height=600)
+
+if "SPACE_ID" in st.secrets:  # running on Hugging Face
+    st.info("Taipy dashboard is disabled on Hugging Face Spaces (requires local port).")
+else:
+    try:
+        st.components.v1.iframe(config["TAIPY_URL"], height=600)
+    except Exception:
+        st.warning("Unable to load Taipy dashboard.")