Update app.py

app.py

@@ -29,25 +29,17 @@ if st.button("Highlight Geo-Entities"):
     if user_input.strip():
         # Process the text using spaCy
         doc = nlp(user_input)
-        
-        # Generate highlighted text with different colors for each entity type
-        highlighted_text = ""
-        last_pos = 0
 
-        for ent in doc.ents:
-            color = COLOR_MAP.get(ent.label_, ('black', ''))[0]  # Default to black if label not in map
-
-            # Add text before the entity
-            highlighted_text += user_input[last_pos:ent.start_char]
-
-            # Add the highlighted entity text
-            highlighted_text += f"<span style='color:{color}; font-weight:bold'>{ent.text}</span>"
-
-            # Update the position
-            last_pos = ent.end_char
-
-        # Add any remaining text after the last entity
-        highlighted_text += user_input[last_pos:]
+        # Highlight geo-entities with different colors
+        highlighted_text = user_input
+        for ent in reversed(doc.ents):
+            if ent.label_ in COLOR_MAP:
+                color = COLOR_MAP[ent.label_][0]
+                highlighted_text = (
+                    highlighted_text[:ent.start_char] +
+                    f"<span style='color:{color}; font-weight:bold'>{ent.text}</span>" + 
+                    highlighted_text[ent.end_char:]
+                )
 
         # Display the highlighted text with HTML support
         st.markdown(highlighted_text, unsafe_allow_html=True)