Update app.py

app.py

@@ -10,11 +10,14 @@ ner_pipeline = pipeline("ner", model="Davlan/xlm-roberta-base-ner-hrl", grouped_
 # Translation models cache
 translation_models = {}
 
-# Get Wikidata entity info via SPARQL
+# Get enriched Wikidata info via SPARQL
 def get_wikidata_info(entity, lang="en"):
     query = f'''
-    SELECT ?item ?itemLabel ?itemDescription WHERE {{
+    SELECT ?item ?itemLabel ?itemDescription ?coordinate ?website ?sitelink WHERE {{
       ?item rdfs:label "{entity}"@{lang}.
+      OPTIONAL {{ ?item wdt:P625 ?coordinate. }}
+      OPTIONAL {{ ?item wdt:P856 ?website. }}
+      OPTIONAL {{ ?sitelink schema:about ?item; schema:isPartOf <https://{lang}.wikipedia.org/>. }}
       SERVICE wikibase:label {{ bd:serviceParam wikibase:language "{lang}". }}
     }} LIMIT 1
     '''
@@ -27,10 +30,13 @@ def get_wikidata_info(entity, lang="en"):
             item = data['results']['bindings'][0]
             label = item.get('itemLabel', {}).get('value', entity)
             description = item.get('itemDescription', {}).get('value', '')
-            return label, description
+            coord = item.get('coordinate', {}).get('value', '')
+            website = item.get('website', {}).get('value', '')
+            wiki = item.get('sitelink', {}).get('value', '')
+            return label, description, coord, website, wiki
     except:
         pass
-    return entity, ""
+    return entity, "", "", "", ""
 
 # Get Wikipedia description as fallback
 def get_wikipedia_summary(entity, lang="en"):
@@ -54,7 +60,7 @@ def translate_text(text, src_lang, tgt_lang):
     except:
         return text  # Return untranslated if model fails
 
-# Combined NER + Wikidata + fallback Wikipedia + translation
+# Combined NER + Wikidata + fallback Wikipedia + translation + links + map
 def multilingual_entity_info(text, output_lang):
     try:
         detected_lang = detect(text)
@@ -69,11 +75,27 @@ def multilingual_entity_info(text, output_lang):
         name = ent['word'].strip()
         if name not in seen and name.isalpha():
             seen.add(name)
-            label, desc = get_wikidata_info(name, lang=detected_lang)
+            label, desc, coord, website, wiki = get_wikidata_info(name, lang=detected_lang)
             if not desc:
                 desc = get_wikipedia_summary(name, lang=detected_lang)
             translated_desc = translate_text(desc, detected_lang, output_lang)
-            result += f"\n---\n\n## 🔎 {label}\n\n{translated_desc}\n"
+
+            links = ""
+            if wiki:
+                links += f"🔗 [Wikipedia]({wiki})  "
+            if website:
+                links += f"🌐 [Official Site]({website})"
+
+            map_embed = ""
+            if coord:
+                try:
+                    latlon = coord.replace('Point(', '').replace(')', '').split(' ')
+                    lon, lat = latlon[0], latlon[1]
+                    map_embed = f"\n<iframe width='100%' height='300' frameborder='0' scrolling='no' marginheight='0' marginwidth='0' src='https://www.openstreetmap.org/export/embed.html?bbox={lon}%2C{lat}%2C{lon}%2C{lat}&layer=mapnik&marker={lat}%2C{lon}'></iframe>"
+                except:
+                    pass
+
+            result += f"\n---\n\n## 🔎 {label}\n\n{translated_desc}\n\n{links}\n{map_embed}\n"
 
     return result if seen else "No named entities found."
 
@@ -85,9 +107,9 @@ iface = gr.Interface(
         gr.Dropdown(label="Select Output Language", choices=["en", "hi", "es", "fr", "de", "ta", "zh"], value="en")
     ],
     outputs=gr.Markdown(),
-    title="🌐 Multilingual NER + Wikidata + Wikipedia",
-    description="Detects entities in any language, fetches descriptions from Wikidata (or Wikipedia), and translates the output into your chosen language."
+    title="🌐 Multilingual NER + Wikidata + Wikipedia + Maps",
+    description="Detects entities in any language, fetches enriched Wikidata info, falls back to Wikipedia, translates the description, and embeds maps + links."
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()