Vaibhav84 commited on
Commit
4a9632c
·
1 Parent(s): b4636a3
Files changed (2) hide show
  1. SkillExtractV1.py +12 -8
  2. app.py +1 -1
SkillExtractV1.py CHANGED
@@ -11,7 +11,9 @@ class SkillExtractorDetailsV1:
11
  skills_list = []
12
  annotations = skill_extractor.annotate(inputData)
13
  matches = annotations['results']['full_matches']+annotations['results']['ngram_scored']
14
- skills_list = []
 
 
15
  for result in matches:
16
  skill_id = result['skill_id']
17
  skill_name1 = skill_extractor.skills_db[skill_id]['skill_name']
@@ -19,18 +21,20 @@ class SkillExtractorDetailsV1:
19
  skill_type = skill_extractor.skills_db[skill_id]['skill_type']
20
  skill_score = round(result['score'],2)
21
  result = getdbskills[getdbskills['skill'] == skill_name.upper()]
22
- print(result.empty)
23
  if(result.empty):
24
- print('----**')
25
- else:
26
- print('----'+ result)
27
- if(skill_type != 'Soft Skill'):
28
- #print(skill_name)
29
  if( skill_name in skills_list):
30
  continue
31
  skills_list.append(skill_name)
 
 
 
 
 
 
 
32
 
33
- return skills_list
34
 
35
  def GetSkillDatafromDB(db_params):
36
  conn = psycopg2.connect(**db_params)
 
11
  skills_list = []
12
  annotations = skill_extractor.annotate(inputData)
13
  matches = annotations['results']['full_matches']+annotations['results']['ngram_scored']
14
+ skill_Type = []
15
+ skill_Score = []
16
+
17
  for result in matches:
18
  skill_id = result['skill_id']
19
  skill_name1 = skill_extractor.skills_db[skill_id]['skill_name']
 
21
  skill_type = skill_extractor.skills_db[skill_id]['skill_type']
22
  skill_score = round(result['score'],2)
23
  result = getdbskills[getdbskills['skill'] == skill_name.upper()]
24
+
25
  if(result.empty):
 
 
 
 
 
26
  if( skill_name in skills_list):
27
  continue
28
  skills_list.append(skill_name)
29
+ skill_type.append(skill_type)
30
+ skill_score.append(skill_score)
31
+ else:
32
+ print('----'+ result)
33
+ list_of_tuples = list(zip(skills_list, skill_Type,skill_Score))
34
+ df = pd.DataFrame(list_of_tuples,
35
+ columns=['Skill_Name', 'Skill_Type','Skill_Score'])
36
 
37
+ return df
38
 
39
  def GetSkillDatafromDB(db_params):
40
  conn = psycopg2.connect(**db_params)
app.py CHANGED
@@ -69,7 +69,7 @@ def UploadJobDescriptionV1(file: bytes = File(...), FileName: str = "sample.pdf
69
  #details = returnSkills.split('@')
70
  #data = {'Data':['Required Skills', 'Soft Skills', 'Good to have Skills'], 'Values':[details[0], details[1], details[2]]}
71
  #df = pd.DataFrame(data)
72
- return returnSkills
73
 
74
  @app.get("/AllProfileMatchResults")
75
  def AllProfileMatchResults():
 
69
  #details = returnSkills.split('@')
70
  #data = {'Data':['Required Skills', 'Soft Skills', 'Good to have Skills'], 'Values':[details[0], details[1], details[2]]}
71
  #df = pd.DataFrame(data)
72
+ return parse_csv(returnSkills)
73
 
74
  @app.get("/AllProfileMatchResults")
75
  def AllProfileMatchResults():