change

SkillExtractV1.py

@@ -13,7 +13,9 @@ class SkillExtractorDetailsV1:
         matches = annotations['results']['full_matches']+annotations['results']['ngram_scored']
         skill_Typelst = []
         skill_Scorelst = []
-
+        reuiredSkilllst = []
+        softskillslst = []
+        goodtohavelst = []
         for result in matches:
             skill_id = result['skill_id']
             skill_name1 = skill_extractor.skills_db[skill_id]['skill_name']
@@ -25,15 +27,26 @@ class SkillExtractorDetailsV1:
             if(result.empty):
                 if( skill_name in skills_list):
                     continue
-                skills_list.append(skill_name)
-                skill_Typelst.append(skill_type)
-                skill_Scorelst.append(skill_score)
+                if(skill_score > 99 and skill_type != 'Soft  Skill'):
+                    reuiredSkilllst.append(skill_name)
+                elif(skill_score > 50 and skill_type == 'Soft  Skill'):
+                    softskillslst.append(skill_name)
+                elif(skill_score <= 50 and skill_type == 'Soft  Skill'):
+                    goodtohavelst.append(skill_name)
+                elif(skill_score <= 99 and skill_type != 'Soft  Skill'):
+                    goodtohavelst.append(skill_name)
+
+                # skills_list.append(skill_name)
+                # skill_Typelst.append(skill_type)
+                # skill_Scorelst.append(skill_score)
             else:
                 print('----'+ result)    
-        list_of_tuples = list(zip(skills_list, skill_Typelst,skill_Scorelst))        
-        df = pd.DataFrame(list_of_tuples,
-                  columns=['Skill_Name', 'Skill_Type','Skill_Score'])
-
+        # list_of_tuples = list(zip(skills_list, skill_Typelst,skill_Scorelst))        
+        # df = pd.DataFrame(list_of_tuples,
+        #           columns=['Skill_Name', 'Skill_Type','Skill_Score'])
+        data = {'Data':['Required Skills', 'Soft Skills', 'Good to have Skills'], 'Values':[reuiredSkilllst, softskillslst, goodtohavelst]}
+        df = pd.DataFrame(data)
+        
         return df        
 
     def GetSkillDatafromDB(db_params):