from fastapi import FastAPI
from pydantic import BaseModel
import subprocess

app = FastAPI()

class CodeRequest(BaseModel):
    user_story: str

@app.post("/generate-code")
def generate_code(request: CodeRequest):
    result = subprocess.run(
        ["ollama", "run", "deepseek-coder-v2", request.user_story],
        capture_output=True,
        text=True
    )
    return {"generated_code": result.stdout.strip()}

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=7860)