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 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(file.filename)
7
  blob.upload_from_file(file.file)
8
  # return gs link
9
- return f"gs://{firebase_bucket.name}/{file.filename}"
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
- # # read the txt file with format
19
- # jobdes = txt_file.file.read().decode("utf-8")
20
- # sumaryjd_text = jobdes2text(jobdes)
21
- # print("sumaryjd_text: ", sumaryjd_text)
22
- # sumaryjd_vector = text2vector(sumaryjd_text)
23
- # print("sumaryjd_vector: ", sumaryjd_vector)
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