cahnges

app.py

@@ -1,26 +1,17 @@
 #Fast APi Packages
-from fastapi import FastAPI,File, UploadFile
+from fastapi import FastAPI,File
 from pydantic import BaseModel
 import json
-from typing_extensions import Annotated
 from fastapi.encoders import jsonable_encoder
 from fastapi.responses import JSONResponse
 
 #SkillExtraction Packages
-import PyPDF2
-from PyPDF2 import PdfReader
+
 import psycopg2
-from psycopg2 import sql
 import pandas as pd
-from datetime import date
 import numpy as np
 import spacy
-import re
-import docx2txt 
-from sentence_transformers import SentenceTransformer, util
-from sklearn.feature_extraction.text import CountVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
-import io  
 from spacy.matcher import PhraseMatcher
 from skillNer.general_params import SKILL_DB
 from skillNer.skill_extractor_class import SkillExtractor
@@ -29,13 +20,13 @@ register_adapter(np.int64, AsIs)
 import warnings
 warnings.filterwarnings('ignore')
 
-from io import BytesIO
-import requests
+
 #Custom Classes for endpoints
 from DbConnection import DbConnection
 from UploadFile import UploadOpenFile
 from SkillExtract import SkillExtractorDetails
 from ExtractContentsFromFile import ExtractContentFromFile
+from RemoveSkills import RemoveSkill
 import os
 os.environ['HF_HOME'] = '/hug/cache/'
 
@@ -109,15 +100,7 @@ def ExtractSkillsByJobID(skill_data: SkillDetails):
    return skill_data
 
 @app.post("/RemoveSkillsByName/")
-def RemoveSkills(SkillName : str):
-    conn = psycopg2.connect(**db_params)
-    cursor = conn.cursor()  
-    SkillName = data.split(':')[1]
-    print("Removing Skills " +  SkillName)
-    query = "update skillmaster set weightage = 0 where skilldetails = (%s)"        
-    params = (SkillName,)  
-    cursor.execute(query, params)
-    conn.commit()
-    cursor.close()
-    conn.close() 
+def RemoveSkills(SkillName : str):    
+    RemoveSkill.RemoveSkillDetails(db_params,SkillName)
+    return "Skill Removed Successfully"
 #return JSONResponse(content={"message": "Here's your interdimensional portal." , "mes1":"data2"})