enhance role description handling in extract_resume_roles method

handler.py

@@ -185,20 +185,28 @@ 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.