Update blackgat/dashboard/dashboard.py

blackgat/dashboard/dashboard.py

@@ -1,47 +1,59 @@
+# dashboard/app.py
 import streamlit as st
 import requests
 
-API = "http://localhost:8000"
-
-st.title("🧠 BlackGat AI Control Panel")
-
-st.header("🔥 HeatSeeker")
-url = st.text_input("Target URL", "https://example.com/admin")
-params = st.text_area("Query Params", '{"q":"test"}')
-status_code = st.number_input("HTTP Status", min_value=100, max_value=599, value=500)
-
-if st.button("Run HeatSeeker"):
-    resp = requests.post(f"{API}/score", json={
-        "url": url,
-        "params": eval(params),
-        "status_code": status_code
-    })
-    st.success(f"Score: {resp.json()['score']}")
-
-st.header("📝 Scribe - AI Report")
-vuln_type = st.text_input("Vuln Type", "XSS")
-vuln_payload = st.text_input("Payload", "<script>alert(1)</script>")
-vuln_impact = st.text_input("Impact", "Session hijacking")
-
-if st.button("Generate Report"):
-    data = {"type": vuln_type, "url": url, "payload": vuln_payload, "impact": vuln_impact}
-    r = requests.post(f"{API}/report", json=data)
-    st.code(r.json()["report"], language="markdown")
-
-st.header("🔗 KillChainAI")
-finding = st.text_area("Findings", "[auth_bypass, XSS, IDOR]")
-if st.button("Suggest Chain"):
-    r = requests.post(f"{API}/chain", json=finding)
-    st.code(r.json()["chain"])
-
-st.header("🧠 ReconGPT")
-prompt = st.text_area("Prompt", "Find XSS using SVG payloads")
-if st.button("Send Prompt"):
-    r = requests.post(f"{API}/recon", json={"prompt": prompt})
-    st.code(r.json()["task"])
-
-st.header("🕵️ Hackphyr")
-target_info = st.text_area("Scenario", "Login allows user injection.")
-if st.button("Suggest Exploit"):
-    r = requests.post(f"{API}/exploit", json={"data": target_info})
-    st.code(r.json()["exploit"])
+API_URL = "http://localhost:8000"
+
+st.set_page_config(page_title="BlackGat AI", layout="wide")
+st.title("🧠 BlackGat AI — Bug Bounty Automation")
+
+tool = st.sidebar.selectbox("Choose Agent", ["Scribe", "KillChain", "HeatSeeker", "ReconGPT", "Exploit Suggestion"])
+
+if tool == "Scribe":
+    st.subheader("📄 Generate Bug Bounty Report")
+    vuln_type = st.text_input("Vulnerability Type")
+    vuln_url = st.text_input("Vulnerable URL")
+    payload = st.text_input("Payload")
+    impact = st.text_area("Impact")
+    if st.button("Generate"):
+        res = requests.post(f"{API_URL}/scribe", json={
+            "type": vuln_type,
+            "url": vuln_url,
+            "payload": payload,
+            "impact": impact
+        })
+        st.markdown(res.json()["report"])
+
+elif tool == "KillChain":
+    st.subheader("🔗 Suggest Chained Attack")
+    findings = st.text_area("Paste findings (JSON or summary)")
+    if st.button("Suggest"):
+        res = requests.post(f"{API_URL}/killchain", json={"data": findings})
+        st.markdown(res.json()["chain"])
+
+elif tool == "HeatSeeker":
+    st.subheader("🌡️ Risk Scoring")
+    url = st.text_input("Target URL")
+    status_code = st.number_input("Status Code", min_value=100, max_value=599, value=200)
+    params = st.text_input("Params JSON", '{"user":"admin"}')
+    if st.button("Score"):
+        res = requests.post(f"{API_URL}/score", json={
+            "url": url,
+            "params": eval(params),
+            "status_code": status_code
+        })
+        st.success(f"Score: {res.json()['score']}")
+
+elif tool == "ReconGPT":
+    st.subheader("🛰️ Recon Task Generator")
+    prompt = st.text_area("What do you want to find?", "Find all login endpoints.")
+    if st.button("Generate Task"):
+        res = requests.post(f"{API_URL}/recon", json={"prompt": prompt})
+        st.markdown(res.json()["task"])
+
+elif tool == "Exploit Suggestion":
+    st.subheader("💥 Attacker Simulation")
+    scenario = st.text_area("Bug scenario or data")
+    if st.button("Get Exploit Advice"):
+        res = requests.post(f"{API_URL}/exploit", json={"data": scenario})
+        st.markdown(res.json()["exploit"])