Update app.py

app.py

@@ -34,14 +34,18 @@ def get_wikidata_info(entity, lang="en"):
         pass
     return entity, "No description available.", "", ""
 
-# Get fallback Wikipedia link
-def get_wikipedia_url(entity, lang="en"):
+# Get Wikipedia details
+
+def get_wikipedia_details(entity, lang="en"):
     try:
         wikipedia.set_lang(lang)
         page = wikipedia.page(entity, auto_suggest=True, redirect=True)
-        return page.url
+        categories = page.categories[:5]
+        links = page.links[:5]
+        url = page.url
+        return url, categories, links
     except:
-        return ""
+        return "", [], []
 
 # Enrich info with tags and intent
 def enrich_info(summary):
@@ -83,7 +87,7 @@ def ner_wikidata_lookup(text):
             seen.add(name)
 
             label, desc, coord, wikidata_url = get_wikidata_info(name, lang=detected_lang)
-            wiki_url = get_wikipedia_url(name, lang=detected_lang)
+            wiki_url, wiki_categories, wiki_links = get_wikipedia_details(name, lang=detected_lang)
 
             related_tags, detected_intent = enrich_info(desc)
 
@@ -104,6 +108,12 @@ def ner_wikidata_lookup(text):
             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>"
 
+            extra_info = ""
+            if wiki_categories:
+                extra_info += f"<p><b>Wikipedia Categories:</b> {', '.join(wiki_categories)}</p>"
+            if wiki_links:
+                extra_info += f"<p><b>Related Topics:</b> {', '.join(wiki_links)}</p>"
+
             result += f"""
             <hr><h3>🔎 {label}</h3>
             <p>{desc}</p>
@@ -111,6 +121,7 @@ def ner_wikidata_lookup(text):
             <p>{osm_link}</p>
             {tags_html}
             {intent_html}
+            {extra_info}
             """
 
     return result if seen else "No named entities found."
@@ -121,8 +132,8 @@ iface = gr.Interface(
     inputs=gr.Textbox(lines=4, placeholder="Type any sentence in any language..."),
     outputs=gr.HTML(),
     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."
+    description="Detects named entities, retrieves Wikidata descriptions, adds Wikipedia links, maps, and enriches output with semantic tags, intent detection, categories, and related topics."
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()