Update app.py

app.py

@@ -1,88 +1,50 @@
-from dotenv import load_dotenv
-import streamlit as st
-import sys, os, requests, pathlib, contextlib
-
-# ─── 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()
-
-# ─── 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")
-
-# ─── Title ──────────────────────────────────────────────────────
-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):
+from fastapi import FastAPI, UploadFile, Form
+from fastapi.responses import JSONResponse
+from utils.summarizer import summarize_text
+from utils.file_utils import normalize_log_line, keyword_search
+from utils.docgen import generate_doc
+
+app = FastAPI(title="Omniscient Backend API")
+
+# ─── Health Check ─────────────────────────────────────────────
+@app.get("/health")
+def health():
+    return JSONResponse({"ok": True, "message": "Omniscient backend alive"})
+
+# ─── File Summarization ───────────────────────────────────────
+@app.post("/summarize")
+async def summarize(file: UploadFile):
     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, col4 = st.columns(4)
-
-active_sessions = len(st.session_state.get("messages", []))
-col1.metric("💬 Active Sessions", active_sessions)
-
-uploaded_files = len(st.session_state.get("uploaded_files", []))
-col2.metric("📂 Uploaded Files", uploaded_files)
-
-errors = len(st.session_state.get("errors", []))
-col3.metric("⚠️ Errors", errors)
-
-col4.metric("🔌 API Backend", backend_status(config["BACKEND_URL"]))
-
-st.success("System ready ✅")
-
-# ─── 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)")
+        content = (await file.read()).decode("utf-8", errors="ignore")
+        summary = summarize_text(content)
+        return {"file": file.filename, "summary": summary}
+    except Exception as e:
+        return {"error": str(e)}
 
-space_id = None
-with contextlib.suppress(Exception):
-    secrets_obj = getattr(st, "secrets", None)
-    if secrets_obj:
-        with contextlib.suppress(Exception):
-            space_id = secrets_obj.get("SPACE_ID", None)
+# ─── Log Normalization & Search ───────────────────────────────
+@app.post("/analyze-log")
+async def analyze_log(file: UploadFile, query: str = Form(None)):
+    try:
+        content = (await file.read()).decode("utf-8", errors="ignore")
+        normalized = [normalize_log_line(line) for line in content.splitlines()]
+        result = {
+            "total_lines": len(normalized),
+            "unique_entries": len(set(normalized)),
+            "preview": normalized[:50],
+        }
+        if query:
+            matches = keyword_search("\n".join(normalized), query)
+            result["matches"] = matches[:50]
+        return {"file": file.filename, "analysis": result}
+    except Exception as e:
+        return {"error": str(e)}
 
-if space_id:
-    st.info("Taipy dashboard is disabled on Hugging Face Spaces (requires local port).")
-else:
+# ─── Script Documentation Generator ──────────────────────────
+@app.post("/generate-doc")
+async def gen_doc(file: UploadFile):
     try:
-        st.components.v1.iframe(config["TAIPY_URL"], height=600)
+        content = (await file.read()).decode("utf-8", errors="ignore")
+        doc = generate_doc(file.filename, "uploaded", content)
+        return {"file": file.filename, "doc": doc}
     except Exception as e:
-        st.warning(f"Unable to load Taipy dashboard: {e}")
+        return {"error": str(e)}