from fastapi import APIRouter, HTTPException

generated_code = []  # In-memory storage

router = APIRouter()

@router.post("/code-generation")
async def generate_code_demo(
    prompt: str = "example prompt",
    language: str = "python",
    model: str = "gpt-3.5"
):
    """Demo code generation endpoint"""
    if not prompt or not language:
        raise HTTPException(status_code=400, detail="Missing required parameters")

    code_id = len(generated_code) + 1
    generated_code.append({
        "id": code_id,
        "prompt": prompt,
        "language": language,
        "model": model
    })

    return {
        "message": "Code generated successfully",
        "code_id": code_id,
        "language": language,
        "features": {
            "tokens": 150,
            "completion_time": 2.5
        }
    }