Update app.py

app.py

@@ -29,17 +29,25 @@ 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
 
-        # 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:]
-                )
+        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:]
 
         # Display the highlighted text with HTML support
         st.markdown(highlighted_text, unsafe_allow_html=True)