from fastapi import APIRouter, HTTPException
from app.models.lm_generator import AiAgentLMGenerator
from app.models.docx_generator import DocxGenerator
from pydantic import BaseModel

router = APIRouter()

class LMRequest(BaseModel):
    name: str
    job: str
    job_description: str
    email: str
    phone: str
    competences: list[str]
    github_url: str = None
    linkedin_url: str = None

class DocxRequest(BaseModel):
    lm_content: str

@router.post("/generatelmcontent")
def generate_lm_content(request: LMRequest):
    try:
        generator = AiAgentLMGenerator()
        user_data = request.dict()
        content = generator.generate(user_data)
        return {"lm_content": content}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))

@router.post("/generateDocx")
def generate_docx(request: DocxRequest):
    try:
        docx_gen = DocxGenerator()
        docx_file = docx_gen.generate_docx(request.lm_content)
        return {"docx_file": docx_file}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))