Spaces:
Build error
Build error
HuyDN
commited on
Commit
•
f03a9ee
1
Parent(s):
097744c
Phase1/HuyDN: Update phase 1
Browse files- app/modules/crud_question_test/models/crud_question_tests.py +4 -2
- app/modules/question_tests_retrieval/__init__.py +7 -10
- app/modules/question_tests_retrieval/models/question_tests_logic.py +0 -2
- data/question_tests/{Question_AI_Senior_1.json → 6322f947_dc0d_41d2_a321_e0f0e57bf070_Question_AI_Senior_1.json} +0 -0
app/modules/crud_question_test/models/crud_question_tests.py
CHANGED
@@ -1,12 +1,14 @@
|
|
|
|
1 |
from app.configs.database import firebase_bucket, firebase_db
|
2 |
|
3 |
# CRUD operation
|
4 |
def upload_file_question_tests(file):
|
|
|
5 |
# upload file to firebase storage
|
6 |
-
blob = firebase_bucket.blob(
|
7 |
blob.upload_from_file(file.file)
|
8 |
# return gs link
|
9 |
-
return f"gs://{firebase_bucket.name}/{
|
10 |
|
11 |
def remove_file_question_tests(file_url):
|
12 |
# remove file from firebase storage using "gs://" link
|
|
|
1 |
+
import uuid
|
2 |
from app.configs.database import firebase_bucket, firebase_db
|
3 |
|
4 |
# CRUD operation
|
5 |
def upload_file_question_tests(file):
|
6 |
+
re_name_file = str(uuid.uuid4()).replace("-","_") + "_" + file.filename
|
7 |
# upload file to firebase storage
|
8 |
+
blob = firebase_bucket.blob(re_name_file)
|
9 |
blob.upload_from_file(file.file)
|
10 |
# return gs link
|
11 |
+
return f"gs://{firebase_bucket.name}/{re_name_file}"
|
12 |
|
13 |
def remove_file_question_tests(file_url):
|
14 |
# remove file from firebase storage using "gs://" link
|
app/modules/question_tests_retrieval/__init__.py
CHANGED
@@ -2,7 +2,7 @@ from fastapi import APIRouter, UploadFile, File
|
|
2 |
from typing import Annotated
|
3 |
|
4 |
from app.modules.question_tests_retrieval.models.jd2text import jobdes2text
|
5 |
-
from app.modules.question_tests_retrieval.models.text2tector import text2vector
|
6 |
from app.modules.question_tests_retrieval.models.question_tests_logic import get_question_test
|
7 |
|
8 |
qtretrieval_router = APIRouter(prefix="/qtretrieval", tags=["qtretrieval"])
|
@@ -15,15 +15,12 @@ async def index():
|
|
15 |
# only upload .txt file
|
16 |
async def send_jd(txt_file: Annotated[UploadFile, File(..., description="The JD file", media_type=["text/plain"])]):
|
17 |
try:
|
18 |
-
#
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
text = "Job Title is Senior AI Engineer, Level is Senior, and Brief summary of required skills is NLP, experiencing in using containers"
|
25 |
-
if get_question_test(text):
|
26 |
-
return {"message": "Send JD successfully and get question test successfully"}
|
27 |
else:
|
28 |
return {"message": "Error"}
|
29 |
except Exception as e:
|
|
|
2 |
from typing import Annotated
|
3 |
|
4 |
from app.modules.question_tests_retrieval.models.jd2text import jobdes2text
|
5 |
+
# from app.modules.question_tests_retrieval.models.text2tector import text2vector
|
6 |
from app.modules.question_tests_retrieval.models.question_tests_logic import get_question_test
|
7 |
|
8 |
qtretrieval_router = APIRouter(prefix="/qtretrieval", tags=["qtretrieval"])
|
|
|
15 |
# only upload .txt file
|
16 |
async def send_jd(txt_file: Annotated[UploadFile, File(..., description="The JD file", media_type=["text/plain"])]):
|
17 |
try:
|
18 |
+
# read the txt file with format
|
19 |
+
jobdes = txt_file.file.read().decode("utf-8")
|
20 |
+
sumaryjd_text = jobdes2text(jobdes)
|
21 |
+
if get_question_test(sumaryjd_text):
|
22 |
+
return {"message": "Send JD successfully and get question test successfully",
|
23 |
+
"sumaryjd_text": sumaryjd_text}
|
|
|
|
|
|
|
24 |
else:
|
25 |
return {"message": "Error"}
|
26 |
except Exception as e:
|
app/modules/question_tests_retrieval/models/question_tests_logic.py
CHANGED
@@ -25,10 +25,8 @@ def compare_vector(vector_extract, vector_des):
|
|
25 |
for item in vector_des:
|
26 |
two_object = (vector_extract, item)
|
27 |
x = gemini_evaluator.evaluate_strings(prediction=two_object[0], reference=two_object[1])
|
28 |
-
# print(f"item and score: {item} - {x.get('score')}")
|
29 |
if x.get('score') < maxnimun_value:
|
30 |
maxnimun_value = x.get('score')
|
31 |
-
print("maxnimun_value: ", maxnimun_value)
|
32 |
des_item_choose = item
|
33 |
if maxnimun_value == 2:
|
34 |
return False
|
|
|
25 |
for item in vector_des:
|
26 |
two_object = (vector_extract, item)
|
27 |
x = gemini_evaluator.evaluate_strings(prediction=two_object[0], reference=two_object[1])
|
|
|
28 |
if x.get('score') < maxnimun_value:
|
29 |
maxnimun_value = x.get('score')
|
|
|
30 |
des_item_choose = item
|
31 |
if maxnimun_value == 2:
|
32 |
return False
|
data/question_tests/{Question_AI_Senior_1.json → 6322f947_dc0d_41d2_a321_e0f0e57bf070_Question_AI_Senior_1.json}
RENAMED
File without changes
|