api.py

from typing import Annotated
from fastapi import FastAPI, APIRouter, UploadFile, Form
from fastapi.staticfiles import StaticFiles
from llama_cpu import load_model, resume_to_job
from helpers import create_session_data
import os
from pyppeteer import launch
from pyppeteer_stealth import stealth

pyppeteer_browser = {}

app = FastAPI()
app.mount("/frontend", StaticFiles(directory="frontend"), name="frontend")
router = APIRouter()

browser = None


@app.on_event("startup")
async def startup_event():
    browser = await launch()
    page = await browser.newPage()
    await stealth(page)
    pyppeteer_browser['page'] = page


if os.path.exists("sessions") is not True:
    os.mkdir('sessions')

app.llm_chain = load_model()


@router.post("/convert")
async def convert_resume(my_resume: UploadFile, job_ad_url: Annotated[str, Form()], session: Annotated[str, Form()]):
    resume_path = await create_session_data(session, my_resume)
    new_resume = await resume_to_job(resume_path, job_ad_url, session, app.llm_chain, pyppeteer_browser['page'])
    return {"newResume": new_resume}


app.include_router(router, prefix="/api")