Update ✨Entity Linking Application✨.py

✨Entity Linking Application✨.py

@@ -100,7 +100,7 @@ async def mains(name, single, combi):
             json_suggestion = await fetch_json(suggested_url, session)
             results = json_suggestion.get('query', {}).get('search')
             for i in results:
-                data.add(int(i.get('title')))
+                    data.add(i.get('title'))
 
         # Handle disambiguation links
         if data != {0}:
@@ -140,9 +140,12 @@ async def mains(name, single, combi):
             try:
                 wikibase_url = f"https://en.wikipedia.org/w/api.php?action=query&titles={ids}&prop=pageprops&format=json"
                 json_qid = await fetch_json(wikibase_url, session)
-                wikidata_qid = json_qid.get('query', {}).get('pages', {}).get(str(ids), {}).get('pageprops', {}).get('wikibase_item', {})
-                if wikidata_qid:
-                    qids.add(wikidata_qid)
+                wikidata_qid = json_qid.get('query', {}).get('pages', {})
+                for page_id, page_data in wikidata_qid.items():
+                    page_props = page_data.get('pageprops', {})
+                    wikibase_item = page_props.get('wikibase_item', None)
+                    if wikibase_item:
+                        qids.add(wikibase_item)
             except:
                 pass