Vaibhav84 commited on
Commit
2e32f9d
·
1 Parent(s): 92cc8d6
Files changed (2) hide show
  1. UploadFile.py +1 -1
  2. app.py +9 -2
UploadFile.py CHANGED
@@ -23,5 +23,5 @@ class UploadOpenFile:
23
  cursor.close()
24
  conn.close()
25
  print("File Uploaded...")
26
- return "File Uploaded Successfully!!!"
27
 
 
23
  cursor.close()
24
  conn.close()
25
  print("File Uploaded...")
26
+ return MasterId
27
 
app.py CHANGED
@@ -41,6 +41,10 @@ class FileDetails(BaseModel):
41
  class SkillDetails(BaseModel):
42
  skillid: int
43
 
 
 
 
 
44
  nlp = spacy.load("en_core_web_lg")
45
  # init skill extractor
46
  skill_extractor = SkillExtractor(nlp, SKILL_DB, PhraseMatcher)
@@ -63,8 +67,11 @@ def ProfileMatchResults():
63
  return parse_csv(df)
64
 
65
  @app.post("/UploadFile/")
66
- def UploadFileDetails(file_data: FileDetails):
67
- return UploadOpenFile.uploadFile(file_data.filecontents,file_data.filename,db_params)
 
 
 
68
 
69
  @app.post("/ExtractSkills/")
70
  def ExtractSkills(skill_data: SkillDetails):
 
41
  class SkillDetails(BaseModel):
42
  skillid: int
43
 
44
+ class FileResponse(BaseModel):
45
+ fileid: int
46
+ message: str
47
+
48
  nlp = spacy.load("en_core_web_lg")
49
  # init skill extractor
50
  skill_extractor = SkillExtractor(nlp, SKILL_DB, PhraseMatcher)
 
67
  return parse_csv(df)
68
 
69
  @app.post("/UploadFile/")
70
+ def UploadFileDetails(file_data: FileDetails):
71
+ FileResponse.fileid = UploadOpenFile.uploadFile(file_data.filecontents,file_data.filename,db_params)
72
+ FileResponse.message = "File Uploaded Successfully!!!"
73
+ return
74
+
75
 
76
  @app.post("/ExtractSkills/")
77
  def ExtractSkills(skill_data: SkillDetails):