Spaces:
Build error
Build error
File size: 1,248 Bytes
097744c 915beec 097744c 2d34b67 097744c f03a9ee 915beec f03a9ee 097744c 53d57fc 097744c 53d57fc aef9bf8 |
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 |
from fastapi import APIRouter, UploadFile, File
from typing import Annotated
from app.modules.question_tests_retrieval.models.jd2text import jobdes2text
# from app.modules.question_tests_retrieval.models.text2vector import text2vector
from app.modules.question_tests_retrieval.models.question_tests_logic import get_question_tests
qtretrieval_router = APIRouter(prefix="/qtretrieval", tags=["qtretrieval"])
@qtretrieval_router.get("/")
async def index():
return {"message": "Welcome to question retrieval page"}
@qtretrieval_router.post("/send_jd")
# only upload .txt file
async def send_jd(txt_file: Annotated[UploadFile, File(..., description="The JD file (only .txt file)", media_type=["text/plain"])]):
try:
# read the txt file with format
jobdes = txt_file.file.read().decode("utf-8")
sumaryjd_text = jobdes2text(jobdes)
if get_question_tests(sumaryjd_text):
return {"message": "Send JD successfully and get question test successfully",
"sumaryjd_text": sumaryjd_text}
else:
return {"message": "Please upload only .txt file", "error": str(e)}
except Exception as e:
return {"message": "Please upload only .txt file", "error": str(e)}
|