Update app.py

app.py

@@ -126,4 +126,159 @@ def inject_osint_context(history, task_result: Dict[str, Any]):
     """
     pretty = json.dumps(task_result, indent=2, default=str)
     blob = f"""
-### OSINT Result Injected:
+### OSINT Result Injected:
+
+{pretty}
+
+"""
+
+    history.append({"role": "system", "content": blob})
+    return history
+
+
+# ---------------------------------------------------------------------
+# Dashboard callbacks (Mode B)
+# ---------------------------------------------------------------------
+
+def ui_lookup_ip(ip, enrich, mitre):
+    raw = call_task("lookup_ip", {"ip": ip, "enrich": enrich, "map_mitre": mitre})
+    normal = format_result_for_ui(raw)
+    return normal["summary"], normal["markdown"], normal["json"], normal["mitre"], normal["stix"], raw
+
+
+def ui_lookup_domain(domain, enrich, mitre):
+    raw = call_task("lookup_domain", {"domain": domain, "enrich": enrich, "map_mitre": mitre})
+    normal = format_result_for_ui(raw)
+    return normal["summary"], normal["markdown"], normal["json"], normal["mitre"], normal["stix"], raw
+
+
+def ui_lookup_hash(h, ht, enrich, mitre):
+    raw = call_task("lookup_hash", {"hash": h, "hash_type": ht, "enrich": enrich, "map_mitre": mitre})
+    normal = format_result_for_ui(raw)
+    return normal["summary"], normal["markdown"], normal["json"], normal["mitre"], normal["stix"], raw
+
+
+def ui_correlate_iocs(iocs):
+    parsed = [l.strip() for l in iocs.splitlines() if l.strip()]
+    raw = call_task("correlate_iocs", {"iocs": parsed})
+    normal = format_result_for_ui(raw)
+    return normal["summary"], normal["markdown"], normal["json"], normal["mitre"], raw
+
+
+def ui_quickscan(target):
+    raw = call_task("quickscan", {"target": target})
+    normal = format_result_for_ui(raw)
+    return normal["summary"], normal["markdown"], normal["json"], raw
+
+
+# ---------------------------------------------------------------------
+# MCP Bridge (Mode D)
+# ---------------------------------------------------------------------
+
+def ui_mcp_bridge(tool, args_json):
+    try:
+        payload = json.loads(args_json)
+    except Exception as exc:
+        err = {"error": f"Invalid JSON: {exc}"}
+        return json.dumps(err, indent=2), "", err
+
+    raw = call_task(tool, payload)
+    normal = format_result_for_ui(raw)
+    return normal["json"], normal["markdown"], raw
+
+
+# ---------------------------------------------------------------------
+# UI — Now with Analyst Copilot
+# ---------------------------------------------------------------------
+
+def build_interface() -> gr.Blocks:
+    with gr.Blocks(title="Parrot OSINT MCP Console") as demo:
+        gr.Markdown("# Parrot OSINT MCP Console")
+
+        # Store OSINT task results for injection into the Copilot
+        osint_result_state = gr.State([])
+
+        # ------------------------------------------
+        # MODE B — Dashboard
+        # ------------------------------------------
+        with gr.Tab("OSINT Dashboard"):
+            with gr.Tab("IP Lookup"):
+                ip = gr.Textbox(label="IP Address")
+                enrich = gr.Checkbox(value=True, label="Enrichment")
+                mitre = gr.Checkbox(value=True, label="MITRE mapping")
+                btn = gr.Button("Run")
+                summary = gr.Textbox(label="Summary")
+                md = gr.Markdown()
+                js = gr.Code(language="json")
+                mt = gr.Code(language="json")
+                st = gr.Code(language="json")
+
+                btn.click(
+                    ui_lookup_ip,
+                    inputs=[ip, enrich, mitre],
+                    outputs=[summary, md, js, mt, st, osint_result_state],
+                )
+
+            # You already know: similar tabs for domain, hash, correlation, quickscan
+            # (keeping focus on Copilot integration)
+
+        # ------------------------------------------
+        # MODE D — MCP Bridge
+        # ------------------------------------------
+        with gr.Tab("MCP Bridge"):
+            tool = gr.Dropdown(sorted(TASK_REGISTRY.keys()))
+            args = gr.Code(label="Args JSON")
+            out_js = gr.Code(language="json")
+            out_md = gr.Markdown()
+
+            bridge_btn = gr.Button("Call Tool")
+            bridge_btn.click(
+                ui_mcp_bridge,
+                inputs=[tool, args],
+                outputs=[out_js, out_md, osint_result_state],
+            )
+
+        # ------------------------------------------
+        # MODE C — Analyst Copilot
+        # ------------------------------------------
+        with gr.Tab("Analyst Copilot"):
+            gr.Markdown("### Streaming TI Assistant with OSINT Context Injection")
+
+            system_msg = gr.Textbox(
+                label="System Prompt",
+                value=("You are a threat intelligence analyst. "
+                       "You think slowly, explain clearly, identify TTPs, "
+                       "and recommend next investigative steps."),
+            )
+
+            model = gr.Textbox(
+                label="HF Model (e.g., openai/gpt-oss-20b)",
+                value="openai/gpt-oss-20b",
+            )
+
+            chatbot = gr.ChatInterface(
+                respond,
+                additional_inputs=[
+                    system_msg,
+                    model,
+                    gr.OAuthToken(label="HF Token"),
+                    gr.Slider(0.1, 2.0, value=0.7, step=0.1, label="Temperature"),
+                    gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p"),
+                    gr.Slider(1, 2048, value=512, step=1, label="Max Tokens"),
+                ],
+                type="messages",
+            )
+
+            inject_btn = gr.Button("Inject Latest OSINT Result")
+            inject_btn.click(
+                inject_osint_context,
+                inputs=[chatbot._chatbot_state, osint_result_state],
+                outputs=[chatbot._chatbot_state],
+            )
+
+    return demo
+
+
+if __name__ == "__main__":
+    demo = build_interface()
+    demo.launch()