Delete api.py

api.py

@@ -1,206 +0,0 @@
-\
-import asyncio
-import os
-import uuid
-from datetime import datetime
-from typing import Dict, List, Optional, Literal
-
-from fastapi import FastAPI, HTTPException, Request
-from fastapi.responses import HTMLResponse, JSONResponse
-from pydantic import BaseModel, Field
-
-# ===== In-Memory Stores =====
-agents: Dict[str, dict] = {}
-queues: Dict[str, List[dict]] = {}
-jobs: Dict[str, dict] = {}
-results: Dict[str, dict] = {}
-waiters: Dict[str, asyncio.Condition] = {}
-
-ADMIN_SECRET = os.environ.get("ADMIN_SECRET")  # optional
-
-def now() -> str:
-    return datetime.utcnow().isoformat() + "Z"
-
-def cond_for(agent_id: str) -> asyncio.Condition:
-    if agent_id not in waiters:
-        waiters[agent_id] = asyncio.Condition()
-    return waiters[agent_id]
-
-# ===== Models =====
-class RegisterBody(BaseModel):
-    display_name: str = Field(..., min_length=1, max_length=80)
-
-class RegisterResp(BaseModel):
-    agent_id: str
-    token: str
-
-class PlanStep(BaseModel):
-    act: Literal["get","wait","click","type","sleep","screenshot","script","scroll"]
-    by: Optional[Literal["css","xpath","id","name","tag","link_text","partial_link_text"]] = None
-    sel: Optional[str] = None
-    text: Optional[str] = None
-    timeout: Optional[int] = 20
-    sec: Optional[float] = None
-    js: Optional[str] = None
-    label: Optional[str] = None
-
-class EnqueuePlanBody(BaseModel):
-    agent_id: str
-    plan: List[PlanStep]
-    admin_secret: Optional[str] = None
-
-class NextJobQuery(BaseModel):
-    agent_id: str
-    token: str
-    wait: int = 25
-
-class SubmitResultBody(BaseModel):
-    agent_id: str
-    token: str
-    job_id: str
-    status: Literal["completed","failed"]
-    message: Optional[str] = None
-    artifacts: Optional[dict] = None  # can contain screenshots, titles, etc.
-
-# ===== App =====
-app = FastAPI(title="Hybrid Agent Orchestrator (Refactor)", version="1.0")
-
-def assert_agent(agent_id: str, token: str) -> dict:
-    a = agents.get(agent_id)
-    if not a:
-        raise HTTPException(status_code=404, detail="unknown agent")
-    if a["token"] != token:
-        raise HTTPException(status_code=401, detail="bad token")
-    return a
-
-def push_job(agent_id: str, payload: dict) -> dict:
-    if agent_id not in agents:
-        raise HTTPException(status_code=404, detail="unknown agent")
-    job_id = str(uuid.uuid4())
-    job = {"id": job_id, "agent_id": agent_id, "payload": payload, "status":"queued", "created_at": now(), "updated_at": now()}
-    jobs[job_id] = job
-    queues.setdefault(agent_id, []).append(job)
-
-    async def notify():
-        async with cond_for(agent_id):
-            cond_for(agent_id).notify_all()
-    asyncio.create_task(notify())
-
-    return job
-
-# ===== Endpoints =====
-@app.post("/register", response_model=RegisterResp)
-async def register(body: RegisterBody):
-    agent_id = str(uuid.uuid4())
-    token = str(uuid.uuid4())
-    agents[agent_id] = {"agent_id": agent_id, "token": token, "display_name": body.display_name, "registered_at": now(), "last_seen": now()}
-    queues[agent_id] = []
-    return RegisterResp(agent_id=agent_id, token=token)
-
-@app.get("/agents")
-async def list_agents():
-    return {"agents": list(agents.values())}
-
-@app.post("/enqueue-plan")
-async def enqueue_plan(body: EnqueuePlanBody):
-    if ADMIN_SECRET and body.admin_secret != ADMIN_SECRET:
-        raise HTTPException(status_code=401, detail="admin_secret invalid")
-    job = push_job(body.agent_id, {"type":"plan","steps":[s.model_dump() for s in body.plan]})
-    return {"job": job}
-
-@app.get("/next-job")
-async def next_job(agent_id: str, token: str, wait: int = 25):
-    assert_agent(agent_id, token)
-    agents[agent_id]["last_seen"] = now()
-
-    q = queues.get(agent_id, [])
-    if q:
-        job = q.pop(0); job["status"]="taken"; job["updated_at"]=now(); return {"job": job}
-
-    c = cond_for(agent_id)
-    try:
-        async with asyncio.timeout(wait):
-            async with c:
-                await c.wait()
-            q = queues.get(agent_id, [])
-            if q:
-                job = q.pop(0); job["status"]="taken"; job["updated_at"]=now(); return {"job": job}
-    except TimeoutError:
-        pass
-    return JSONResponse(status_code=204, content={"detail":"no job"})
-
-@app.get("/jobs/{job_id}")
-async def get_job(job_id: str):
-    return {"job": jobs.get(job_id), "result": results.get(job_id)}
-
-@app.post("/submit-result")
-async def submit_result(body: SubmitResultBody):
-    assert_agent(body.agent_id, body.token)
-    job = jobs.get(body.job_id)
-    if not job: raise HTTPException(status_code=404, detail="unknown job")
-    job["status"] = body.status; job["updated_at"] = now()
-    results[body.job_id] = {"status": body.status, "message": body.message, "artifacts": body.artifacts, "submitted_at": now()}
-    return {"ok": True}
-
-# ===== Minimal HTML (build plan server-side; secrets live here) =====
-HTML = """
-<!doctype html><html><head><meta charset='utf-8'/><meta name='viewport' content='width=device-width,initial-scale=1'/>
-<title>Hybrid Agent Orchestrator</title>
-<style>body{font-family:system-ui,Segoe UI,Roboto,Helvetica,Arial,sans-serif;margin:20px} .card{border:1px solid #ddd;border-radius:12px;padding:16px;margin-bottom:16px}
-input,select{width:100%;padding:8px;border:1px solid #ccc;border-radius:8px} button{padding:10px 14px;border-radius:10px;border:0;background:#111;color:#fff;cursor:pointer}
-pre{background:#f7f7f7;padding:8px;border-radius:8px;overflow:auto}
-</style></head><body>
-<div class="card"><h2>Agents</h2><div id="agents"></div><button onclick="loadAgents()">Refresh</button></div>
-<div class="card"><h2>Build & Send Plan (example: login to Google)</h2>
-<label>Agent ID</label><input id="agent_id" placeholder="paste agent_id"/>
-<label>Email</label><input id="email"/>
-<label>Password</label><input id="password" type="password"/>
-<label>Target URL</label><input id="url" value="https://accounts.google.com/signin"/>
-<button onclick="sendPlan()">Send Plan</button>
-<div id="out"></div>
-</div>
-<div class="card"><h2>Check Job</h2><label>Job ID</label><input id="job_id"/><button onclick="checkJob()">Check</button><div id="status"></div></div>
-<script>
-async function loadAgents(){ const r=await fetch('/agents'); const j=await r.json(); document.getElementById('agents').innerHTML='<pre>'+JSON.stringify(j,null,2)+'</pre>'; }
-loadAgents();
-function buildPlan(email,password,url){
-  // Sensitive step building happens here in the browser for demo; in production, build on server side with a POST that returns steps
-  return [
-    {"act":"get","sel":url},
-    {"act":"wait","by":"css","sel":"input[type=email]", "timeout":25},
-    {"act":"type","by":"css","sel":"input[type=email]", "text":email},
-    {"act":"click","by":"css","sel":"#identifierNext"},
-    {"act":"wait","by":"css","sel":"input[type=password]", "timeout":25},
-    {"act":"type","by":"css","sel":"input[type=password]","text":password},
-    {"act":"click","by":"css","sel":"#passwordNext"},
-    {"act":"wait","by":"css","sel":"body", "timeout":30},
-    {"act":"screenshot","label":"after_login"}
-  ];
-}
-async function sendPlan(){
-  const agent_id = document.getElementById('agent_id').value;
-  const email = document.getElementById('email').value;
-  const password = document.getElementById('password').value;
-  const url = document.getElementById('url').value;
-  const plan = buildPlan(email,password,url);
-  const r = await fetch('/enqueue-plan',{method:'POST', headers:{'Content-Type':'application/json'}, body:JSON.stringify({agent_id, plan})});
-  const j = await r.json();
-  document.getElementById('out').innerHTML='<pre>'+JSON.stringify(j,null,2)+'</pre>';
-  document.getElementById('job_id').value = j.job.id;
-}
-async function checkJob(){
-  const job_id = document.getElementById('job_id').value;
-  const r = await fetch('/jobs/'+job_id);
-  const j = await r.json();
-  let html = '<pre>'+JSON.stringify(j,null,2)+'</pre>';
-  if (j.result && j.result.artifacts && j.result.artifacts.screenshots && j.result.artifacts.screenshots.after_login){
-    html += '<img src="data:image/png;base64,'+j.result.artifacts.screenshots.after_login+'" style="max-width:100%;border:1px solid #ddd;border-radius:8px"/>';
-  }
-  document.getElementById('status').innerHTML = html;
-}
-</script>
-</body></html>
-"""
-@app.get("/", response_class=HTMLResponse)
-async def home():
-    return HTMLResponse(HTML)