Spaces:
Sleeping
Sleeping
File size: 1,096 Bytes
adfe44e c7ef4a4 adfe44e 5b54533 adfe44e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | 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))
|