Db Connection added

app.py

@@ -5,6 +5,10 @@ from psycopg2 import sql
 import pandas as pd
 import numpy as np
 from pydantic import BaseModel
+import spacy
+from spacy.matcher import PhraseMatcher
+from skillNer.general_params import SKILL_DB
+from skillNer.skill_extractor_class import SkillExtractor
 from psycopg2.extensions import register_adapter, AsIs
 register_adapter(np.int64, AsIs)
 import warnings
@@ -13,14 +17,20 @@ warnings.filterwarnings('ignore')
 #Custom Classes for endpoints
 from DbConnection import DbConnection
 from UploadFile import UploadOpenFile
+from SkillExtract import SkillExtractor
 
 app = FastAPI()
 
 class FileDetails(BaseModel):
     filecontents: str
     filename: str
-    
 
+class SkillDetails(BaseModel):
+    skillid: int    
+    
+nlp = spacy.load("en_core_web_lg")
+    # init skill extractor
+skill_extractor = SkillExtractor(nlp, SKILL_DB, PhraseMatcher)
 
 @app.get("/")
 async def root():
@@ -41,4 +51,8 @@ def ProfileMatchResults():
 
 @app.post("/UploadFile/")
 def UploadFileDetails(file_data: FileDetails):
-   return UploadOpenFile.uploadFile(file_data.filecontents,file_data.filename,db_params)
+   return UploadOpenFile.uploadFile(file_data.filecontents,file_data.filename,db_params)
+
+@app.post("/ExtractSkills/")
+def ExtractSkills(skill_data: SkillDetails):
+   return SkillExtractor.SkillExtract(db_params,skill_extractor,skill_data.skillid)

requirements.txt

@@ -2,4 +2,7 @@ fastapi
 uvicorn
 numpy
 pandas
-psycopg2_binary
+psycopg2_binary
+skillNer
+spacy
+en-core-web-lg @https://github.com/explosion/spacy-models/releases/download/en_core_web_lg-3.7.1/en_core_web_lg-3.7.1-py3-none-any.whl