Changes

app.py

@@ -49,6 +49,12 @@ class SkillDetails(BaseModel):
     requiredSkills: str
     softSkills: str
     goodToHaveSkills: str   
+
+class SkillData(BaseModel):
+    filename: str
+    requiredSkills: str
+    softSkills: str
+    goodToHaveSkills: str       
         
 class FileResponse(BaseModel):
     fileid: int
@@ -73,44 +79,34 @@ def parse_csv(df):
     parsed = json.loads(res)
     return parsed
     
-@app.post("/uploadJobDescriptionPDF/")
-def process_pdf_file(file: bytes =  File(...)):
-    # Save file locally for processing
-    text =''
-    reserve_pdf_on_memory = io.BytesIO(file)
-    load_pdf = PyPDF2.PdfReader(reserve_pdf_on_memory)
-    for page in load_pdf.pages:
-                text += page.extract_text()
-    print(text)
-    # Process saved file
-    return {"content": text}    
+ 
 
-@app.post("/uploadJobDescriptionPDF_Fname/")
-def process_pdf_file1(file: bytes =  File(...), FileName: str = str):
+@app.post("/UploadJobDescription/")
+def UploadJobDescription(file: bytes =  File(...), FileData: SkillData = SkillData):
     # Save file locally for processing
     text =''
     print(text)
-    if FileName.endswith("pdf"):
+    if FileData.FileName.endswith("pdf"):
         reserve_pdf_on_memory = io.BytesIO(file)
         load_pdf = PyPDF2.PdfReader(reserve_pdf_on_memory)
         for page in load_pdf.pages:
             text += page.extract_text()
        
-    elif FileName.endswith("doc") or FileName.endswith("docx"):
-        text = docx2txt.process(file)
+    elif FileData.FileName.endswith("doc") or FileData.FileName.endswith("docx"):
+        text = docx2txt.process(io.BytesIO(file))
         text = text.read()
     
     else:
         text = file.decode('utf-8')
 
-    returnID = UploadOpenFile.uploadFile(text,FileName,db_params)
+    returnID = UploadOpenFile.uploadFile(text,FileData.FileName,db_params)
     returnSkills = SkillExtractorDetails.SkillExtract(db_params,skill_extractor,returnID)     
     details = returnSkills.split('@')
-    #skill_data.requiredSkills = details[0]
-    #skill_data.softSkills = details[1]
-    #skill_data.goodToHaveSkills = details[1]    
+    FileData.requiredSkills = details[0]
+    FileData.softSkills = details[1]
+    FileData.goodToHaveSkills = details[1]    
     # Process saved file
-    return {"content": details}  
+    return {"content": FileData}  
 @app.get("/ProfileMatch") 
 def ProfileMatchResults():
    dbQuery = "select * from profilematch"