Update app.py

app.py

@@ -3,11 +3,13 @@ from transformers import pipeline
 from langdetect import detect
 import requests
 import wikipedia
+import re
 
 # Load NER model
 ner_pipeline = pipeline("ner", model="Davlan/xlm-roberta-base-ner-hrl", grouped_entities=True)
 
 # Get Wikidata entity info
+
 def get_wikidata_info(entity, lang="en"):
     query = f'''
     SELECT ?item ?itemLabel ?itemDescription ?coordinate WHERE {{
@@ -41,6 +43,29 @@ def get_wikipedia_url(entity, lang="en"):
     except:
         return ""
 
+# Enrich info with tags and intent
+def enrich_info(summary):
+    related_info = []
+
+    if re.search(r'capital', summary, re.IGNORECASE):
+        related_info.append("🏛️ Capital city")
+    if re.search(r'tourism|attraction', summary, re.IGNORECASE):
+        related_info.append("🧳 Popular for tourism")
+    if re.search(r'population', summary, re.IGNORECASE):
+        related_info.append("👥 Densely populated")
+    if re.search(r'university|education', summary, re.IGNORECASE):
+        related_info.append("🎓 Educational hub")
+    if re.search(r'beach', summary, re.IGNORECASE):
+        related_info.append("🏖️ Known for beaches")
+
+    intent = "General knowledge inquiry"
+    if re.search(r'tourism|travel', summary, re.IGNORECASE):
+        intent = "Looking for travel guidance"
+    elif re.search(r'university|education', summary, re.IGNORECASE):
+        intent = "Seeking educational info"
+
+    return related_info, intent
+
 # Main combined function
 def ner_wikidata_lookup(text):
     try:
@@ -60,6 +85,8 @@ def ner_wikidata_lookup(text):
             label, desc, coord, wikidata_url = get_wikidata_info(name, lang=detected_lang)
             wiki_url = get_wikipedia_url(name, lang=detected_lang)
 
+            related_tags, detected_intent = enrich_info(desc)
+
             osm_link = ""
             if coord:
                 try:
@@ -74,11 +101,16 @@ def ner_wikidata_lookup(text):
             if wiki_url:
                 links += f"<a href='{wiki_url}' target='_blank'>📘 Wikipedia</a>"
 
+            tags_html = f"<p><b>Related Tags:</b> {' | '.join(related_tags)}</p>" if related_tags else ""
+            intent_html = f"<p><b>Intent:</b> {detected_intent}</p>"
+
             result += f"""
             <hr><h3>🔎 {label}</h3>
             <p>{desc}</p>
             <p>{links}</p>
             <p>{osm_link}</p>
+            {tags_html}
+            {intent_html}
             """
 
     return result if seen else "No named entities found."
@@ -88,8 +120,8 @@ iface = gr.Interface(
     fn=ner_wikidata_lookup,
     inputs=gr.Textbox(lines=4, placeholder="Type any sentence in any language..."),
     outputs=gr.HTML(),
-    title="🌐 NER with Wikidata + Wikipedia + Maps",
-    description="Detects named entities, retrieves Wikidata descriptions, adds Wikipedia links and map locations."
+    title="🌐 NER with Wikidata + Wikipedia + Smart Tags",
+    description="Detects named entities, retrieves Wikidata descriptions, adds Wikipedia links, maps, and enriches output with semantic tags and intent detection."
 )
 
 if __name__ == "__main__":