app.py

import uvicorn
from fastapi import FastAPI, HTTPException
from model import generate_code

app = FastAPI()

@app.get("/")
def home():
    return {"message": "Code Generation API is running!"}

@app.post("/generate")
def generate(prompt: str, max_tokens: int = 256):
    if not prompt:
        raise HTTPException(status_code=400, detail="Prompt cannot be empty.")

    try:
        code = generate_code(prompt, max_tokens)
        return {"generated_code": code}
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Internal Server Error: {str(e)}")

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