Update app.py

app.py

@@ -24,6 +24,17 @@ def execute_task(req: TaskRequest):
         raise HTTPException(status_code=400, detail="Invalid task")
     return {"output": output}
 
+def execute_tasks(team: str, task: str, prompt: str):
+    # You can route tasks to different modules here
+    if team == "Red" and task == "Exploit":
+        # Stub: Replace with AI logic or subprocess
+        output = f"Running red team exploit with prompt: {prompt}"
+    elif team == "Blue" and task == "Analyze Logs":
+        # Example: Call AI model or subprocess
+        output = f"Parsing logs with AI assist: {prompt}"
+    else:
+        raise HTTPException(status_code=400, detail="Invalid task")
+    return {"output": output}
 
 
 import gradio as gr
@@ -32,10 +43,8 @@ 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")
+ 
+    return execute_tasks(team, task, prompt)
 
 iface = gr.Interface(
     fn=execute_task,