update app

app.py

@@ -21,10 +21,6 @@ def get_wikipedia_page_props(input_str: str):
     Returns:
         str: The QID or "NIL" if the QID is not found.
     """
-    # # Check if the result is already in the cache
-    # if input_str in cache:
-    #     return cache[input_str]
-
     try:
         # Preprocess the input string
         page_name, language = input_str.split(" >> ")
@@ -57,14 +53,11 @@ def get_wikipedia_page_props(input_str: str):
                 if "wikibase_item" in page_props:
                     return page_props["wikibase_item"]
                 else:
-                    # If no page properties found, fall back to OpenRefine Wikidata API
-                    return qid  # fallback_to_openrefine(page_name, language)
+                    return qid
             else:
-                return qid  # fallback_to_openrefine(page_name, language)
-
+                return qid
     except Exception as e:
-        # save_cache(cache, cache_file)
-        return qid  # fallback_to_openrefine(page_name, language)
+        return qid
 
 
 def get_wikipedia_title(qid, language="en"):
@@ -89,7 +82,6 @@ def get_wikipedia_title(qid, language="en"):
 
 
 def disambiguate_sentence(sentence):
-    entities = []
     # Generate model outputs for the sentence
     outputs = model.generate(
         **tokenizer([sentence], return_tensors="pt"),
@@ -99,18 +91,25 @@ def disambiguate_sentence(sentence):
     )
     decoded = tokenizer.batch_decode(outputs, skip_special_tokens=True)
     print(f"Decoded: {decoded}")
-    wikipedia_name = decoded[0]  # Assuming QID is the last token in the output
+    wikipedia_name = decoded[0]  # Assuming the entity name is in the output
     qid = get_wikipedia_page_props(wikipedia_name)
-    print(f"qid: {qid}")
-    #
-    # # Get Wikipedia title and URL
+    print(f"QID: {qid}")
+
+    # Get Wikipedia title and URL
     title, url = get_wikipedia_title(qid)
-    #
-    entity_info = f"QID: {qid}, Title: {title}, URL: {url}"
-    entities.append(entity_info)
 
-    print(f"Entities: {entities}")
-    return {"text": sentence, "entities": entities}
+    if qid == "NIL":
+        return "No entity found."
+
+    # Create an HTML output with a clickable link
+    entity_info = f"""
+    <div>
+        <strong>Entity:</strong> {title} <br>
+        <strong>QID:</strong> {qid} <br>
+        <a href="{url}" target="_blank">Wikipedia Page</a>
+    </div>
+    """
+    return entity_info
 
 
 def nel_app_interface():
@@ -122,7 +121,7 @@ def nel_app_interface():
         "entity should be surrounded by `[START]` and `[END]`. // "
         "!Only one entity per sentence is supported at the moment!",
     )
-    output_entities = gr.Textbox(label="Linked Entities")
+    output_entities = gr.HTML(label="Linked Entity")
 
     # Interface definition
     interface = gr.Interface(