test

handler.py

@@ -185,28 +185,20 @@ class EndpointHandler():
 
             # If element is a description, directly append to the last role in the array
             if self.Resume_labels[item]['value'] == "EDC":
-                # If description is stuck between EJT, most likely it is an EJT
-                if i - 1 > 0 and i + 1 < len(predictions) and self.Resume_labels[predictions[i - 1]]['value'] == "EJT" and self.Resume_labels[predictions[i + 1]]['value'] == "EJT":
-                    roles[len(roles) - 1]["title"].append(lines[i])
-                    continue
                 if roles:
                     roles[-1]["description"].append(lines[i])
                 else:
                     # Optionally, log the error or create a default role
                     print("Warning: Description found but no role header exists. Skipping this description.")
 
-            # If element is not an EJT or EDC but stuck between EDC, most likely it is an EDC
-            if self.Resume_labels[item]['value'] != "EDC" and self.Resume_labels[item]['value'] != "EJT":
-                if i - 1 > 0 and i + 1 < len(predictions) and self.Resume_labels[predictions[i - 1]]['value'] == "EDC" and self.Resume_labels[predictions[i + 1]]['value'] == "EDC":
-                    roles[-1]["description"].append(lines[i])
-
         # Cleaning description
         for item in roles:
             sentences = self.capture_sentences(item['description'])
             item['description'] = sentences
 
         return roles
-    
+
+
     def parse_date(self, date_str):
         """Tries multiple formats to parse a date string into a datetime object.