Fix: Overlaping entities issue

app.py

@@ -78,12 +78,15 @@ def create_custom_entity_viz(data, full_text):
             end = dataentity["end"]
             
             if start < len(doc) and end <= len(doc):
-                span = Span(doc, start, end, label=node["type"])
-                
-                # print(span)
-                spans.append(span)
-                if node["type"] not in colors:
-                    colors[node["type"]] = get_random_light_color()
+                # Check for overlapping spans
+                overlapping = any(s.start < end and start < s.end for s in spans)
+                if not overlapping:                
+                    span = Span(doc, start, end, label=node["type"])
+                    
+                    # print(span)
+                    spans.append(span)
+                    if node["type"] not in colors:
+                        colors[node["type"]] = get_random_light_color()
 
     doc.set_ents(spans, default="unmodified")
     doc.spans["sc"] = spans

phi3_instruct_graph.py

@@ -109,5 +109,5 @@ class Phi3InstructGraph:
     def extract(self, text):
         messages = self._get_messages(text)
         pipe_output = self._generate(messages)
-        print("pipe_output json", pipe_output[0]["generated_text"])
+        # print("pipe_output json", pipe_output[0]["generated_text"])
         return pipe_output[0]["generated_text"]