api.py

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import StreamingResponse
from pydantic import BaseModel
from agents import agent
from database import get_db, Tasks
import time

app = FastAPI()
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_methods=["*"],
    allow_headers=["*"]
)

class RunRequest(BaseModel):
    task: str
    folder_path: str

def stream_agent(request):
    db = next(get_db())
    task = Tasks(task=request.task, status="running")
    db.add(task)
    db.commit()

    for chunk in agent.stream({
        "task": request.task,
        "plan": "",
        "architecture": "",
        "folder_path": request.folder_path,
        "file_name": "",
        "code": "",
        "tests": "",
        "test_results": "",
        "status": "",
        "retry_count": 0
    }):
        node_name = list(chunk.keys())[0]
        yield f"{node_name} done\n"

    task.status = "done"
    db.commit()
    db.close()

@app.post("/run")
def run(request: RunRequest):
    return StreamingResponse(stream_agent(request), media_type="text/plain")