Update generic_nel.py

generic_nel.py

@@ -20,14 +20,25 @@ def get_wikipedia_page_props(input_str: str):
     Returns:
         str: The QID or "NIL" if the QID is not found.
     """
-    try:
+    # print(f"Input string: {input_str}")
+    if ">>" not in input_str:
+        page_name = input_str
+        language = "en"
+        print(
+            f"<< was not found in {input_str} so we are checking with these values: Page name: {page_name}, Language: {language}"
+        )
+    else:
         # Preprocess the input string
-        page_name, language = input_str.split(" >> ")
-        page_name = page_name.strip()
-        language = language.strip()
-    except ValueError:
-        return "Invalid input format. Use 'page_name >> language'.", "None"
-
+        try:
+            page_name, language = input_str.split(">>")
+            page_name = page_name.strip()
+            language = language.strip()
+        except:
+            page_name = input_str
+            language = "en"
+            print(
+                f"<< was not found in {input_str} so we are checking with these values: Page name: {page_name}, Language: {language}"
+            )
     wikipedia_url = f"https://{language}.wikipedia.org/w/api.php"
     wikipedia_params = {
         "action": "query",
@@ -50,11 +61,14 @@ def get_wikipedia_page_props(input_str: str):
                 page_props = data["query"]["pages"][page_id]["pageprops"]
 
                 if "wikibase_item" in page_props:
+                    # print(page_props["wikibase_item"], language)
                     return page_props["wikibase_item"], language
                 else:
                     return qid, language
             else:
                 return qid, language
+        else:
+            return qid, language
     except Exception as e:
         return qid, language