Create gradio_ui.py

gradio_ui.py

@@ -0,0 +1,26 @@
+
+# gradio_ui.py
+import gradio as gr
+import requests
+
+API_BASE = "http://localhost:8000"  # Adjust to your FastAPI base URL
+
+def execute_task(team: str, task: str, prompt: str):
+    response = requests.post(f"{API_BASE}/execute", json={
+        "team": team, "task": task, "prompt": prompt
+    })
+    return response.json().get("output", "No output")
+
+iface = gr.Interface(
+    fn=execute_task,
+    inputs=[
+        gr.Dropdown(["Red", "Blue"], label="Team"),
+        gr.Dropdown(["Recon", "Exploit", "Analyze Logs", "Generate Rule"], label="Task"),
+        gr.Textbox(label="Prompt / Parameters")
+    ],
+    outputs="text",
+    title="AI-Driven CyberOps Interface"
+)
+
+if __name__ == "__main__":
+    iface.launch()