Vaibhav84 commited on
Commit
d21275b
·
1 Parent(s): 71694b0
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -44,7 +44,10 @@ class FileDetails(BaseModel):
44
 
45
 
46
  class SkillDetails(BaseModel):
47
- skillid: int
 
 
 
48
 
49
  class FileResponse(BaseModel):
50
  fileid: int
@@ -84,4 +87,9 @@ def UploadFileDetails(file_data: FileDetails):
84
 
85
  @app.post("/ExtractSkills/")
86
  def ExtractSkills(skill_data: SkillDetails):
87
- return SkillExtractorDetails.SkillExtract(db_params,skill_extractor,skill_data.skillid)
 
 
 
 
 
 
44
 
45
 
46
  class SkillDetails(BaseModel):
47
+ skillid: int
48
+ requiredSkills: str
49
+ softSkills: str
50
+ goodToHaveSkills: str
51
 
52
  class FileResponse(BaseModel):
53
  fileid: int
 
87
 
88
  @app.post("/ExtractSkills/")
89
  def ExtractSkills(skill_data: SkillDetails):
90
+ returnSkills = SkillExtractorDetails.SkillExtract(db_params,skill_extractor,skill_data.skillid)
91
+ details = returnSkills.split('@')
92
+ skill_data.requiredSkills = details[0]
93
+ skill_data.softSkills = details[1]
94
+ skill_data.goodToHaveSkills = details[1]
95
+ return skill_data