changes

AddSkillDetails.py

@@ -0,0 +1,34 @@
+import psycopg2
+from psycopg2 import sql
+
+class AddSkill:
+    def AddSkillDetails(db_params, skills):
+        returnMsg=''
+        details = skills.split(',')
+        skill_details = details[0]
+        skill_type = details [1]
+        skill_score1 = details[2]
+        weightage = -2
+        is_active = True
+        conn = psycopg2.connect(**db_params)
+        cursor = conn.cursor()
+        print("Adding Skill " + skill_details)
+        query = "SELECT skillid FROM skillmaster WHERE upper(skillDetails) IN (%s)"
+        params = (skill_details.upper(),)  # Replace 'Test' with your actual variable or user input
+        cursor.execute(query, params)
+        if cursor.rowcount == 0:    
+
+            insert_query = sql.SQL("""INSERT INTO SkillMaster (SkillDetails, SkillType, Weightage, IsActive, skill_score) 
+                        VALUES (%s, %s, %s, %s, %s) RETURNING SkillID""")
+            cursor.execute(insert_query, (skill_details, skill_type, weightage, is_active, skill_score1))
+            conn.commit()
+            returnMsg = 'Skill Added successfully'
+        else:            
+            returnMsg = 'Skill Already in DB'
+             
+        # Close the cursor and connection
+        cursor.close()
+        # Close the connection
+        conn.close()
+        print(returnMsg)   
+        return returnMsg

ClassModals.py

@@ -0,0 +1,6 @@
+from pydantic import BaseModel
+class ClassModals:
+    class AddSkillDetails(BaseModel):    
+        SkillName: str = 'SkillName'
+        SkillType: str = 'Soft Skill'
+        SkillScore: int = 10   

app.py

@@ -24,6 +24,8 @@ from UploadFile import UploadOpenFile
 from SkillExtract import SkillExtractorDetails
 from ExtractContentsFromFile import ExtractContentFromFile
 from RemoveSkills import RemoveSkill
+from AddSkillDetails import AddSkill
+import ClassModals
 import os
 os.environ['HF_HOME'] = '/hug/cache/'
 
@@ -41,6 +43,11 @@ class SkillDetails(BaseModel):
     softSkills: str
     goodToHaveSkills: str  
 
+class AddSkillDetails1(BaseModel):    
+    SkillName: str = 'SkillName'
+    SkillType: str = 'Soft Skill'
+    SkillScore: int = 10     
+
 class SkillData(BaseModel):
     filename: str 
          
@@ -100,5 +107,13 @@ def ExtractSkillsByJobID(skill_data: SkillDetails):
 def RemoveSkills(SkillName : str):    
     RemoveSkill.RemoveSkillDetails(db_params,SkillName)
     return "Skill Removed Successfully"
+
+@app.post("/AddSkillDeails/")
+def AddSkills(Skills : ClassModals.ClassModals.AddSkillDetails):    
+    skilldetailsStr = Skills.SkillName + ',' + Skills.SkillType + ',' + str(Skills.SkillScore)
+    AddSkill.AddSkillDetails(db_params,skilldetailsStr)
+    return "Skill Removed Successfully"
+
+
 #return JSONResponse(content={"message": "Here's your interdimensional portal." , "mes1":"data2"})
 #https://vaibhav84-resumeapi.hf.space/docs