Create ScriptOrchestrator.py

pages/ScriptOrchestrator.py

@@ -0,0 +1,105 @@
+import streamlit as st
+import sys, os, subprocess, hashlib
+
+# ─── Ensure utils/ is importable ──────────────────────────────
+UTILS_PATH = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+if UTILS_PATH not in sys.path:
+    sys.path.append(UTILS_PATH)
+
+from utils.docgen import generate_doc
+from utils.summarizer import summarize_text
+
+st.title("🛠️ Local Script Orchestration & Execution")
+st.write("Scan, queue, and execute Omniscient scripts with results ingested back into the system.")
+
+# Init state
+if "local_scripts" not in st.session_state:
+    st.session_state.local_scripts = []
+if "script_runs" not in st.session_state:
+    st.session_state.script_runs = []
+
+SCAN_DIRS = [
+    "/opt/omniscient/bin",
+    "/opt/omniscient/scripts",
+    "/opt/omniscient/ai"
+]
+
+def scan_scripts():
+    scripts = []
+    for d in SCAN_DIRS:
+        if os.path.exists(d):
+            for root, _, files in os.walk(d):
+                for f in files:
+                    if f.endswith((".sh", ".py")):
+                        path = os.path.join(root, f)
+                        try:
+                            with open(path, "r", errors="ignore") as fh:
+                                content = fh.read()
+                            sha1 = hashlib.sha1(content.encode()).hexdigest()
+                            doc = generate_doc(f, path, content)
+                            summary = summarize_text(content)
+
+                            scripts.append({
+                                "name": f,
+                                "path": path,
+                                "sha1": sha1,
+                                "doc": doc,
+                                "summary": summary
+                            })
+                        except Exception as e:
+                            st.error(f"⚠️ Error reading {f}: {e}")
+    return scripts
+
+def run_script(script_path: str):
+    """Execute a script safely and capture output"""
+    try:
+        if script_path.endswith(".sh"):
+            result = subprocess.run(["bash", script_path], capture_output=True, text=True, timeout=60)
+        elif script_path.endswith(".py"):
+            result = subprocess.run(["python3", script_path], capture_output=True, text=True, timeout=60)
+        else:
+            return {"error": "Unsupported script type"}
+
+        return {
+            "script": script_path,
+            "stdout": result.stdout,
+            "stderr": result.stderr,
+            "returncode": result.returncode
+        }
+    except Exception as e:
+        return {"script": script_path, "error": str(e)}
+
+# ─── Scan Scripts ─────────────────────────────────────────────
+if st.button("🔍 Scan Local Scripts"):
+    scripts = scan_scripts()
+    st.session_state.local_scripts = scripts
+    st.success(f"✅ Found {len(scripts)} scripts.")
+
+# ─── Script Selection ─────────────────────────────────────────
+if st.session_state.local_scripts:
+    script_names = [s["name"] for s in st.session_state.local_scripts]
+    selected = st.multiselect("Select scripts to execute", script_names)
+
+    if st.button("▶️ Run Selected Scripts"):
+        for s in st.session_state.local_scripts:
+            if s["name"] in selected:
+                run_result = run_script(s["path"])
+                st.session_state.script_runs.append(run_result)
+                if "error" in run_result:
+                    st.error(f"❌ {s['name']} failed: {run_result['error']}")
+                elif run_result["returncode"] != 0:
+                    st.error(f"❌ {s['name']} exited with {run_result['returncode']}")
+                    st.code(run_result["stderr"])
+                else:
+                    st.success(f"✅ {s['name']} ran successfully")
+                    st.code(run_result["stdout"][:500])  # preview output
+
+# ─── Display Past Runs ────────────────────────────────────────
+if st.session_state.script_runs:
+    st.subheader("📜 Script Run History")
+    for r in st.session_state.script_runs[-5:]:  # show last 5
+        st.markdown(f"**Script:** {r.get('script')}  \n**Return Code:** {r.get('returncode')}")
+        if r.get("stdout"):
+            st.code(r["stdout"][:300])
+        if r.get("stderr"):
+            st.code(r["stderr"][:300])