Update app.py

app.py

@@ -76,9 +76,15 @@ def vis_network_html(relations, entities, type_colors):
             shape="box",
             color=color
         )
-    # Add edges for relations
+    # Add edges for relations, with tooltip as description
     for rel in relations:
-        net.add_edge(rel["subject"], rel["object"], label=rel["predicate"], arrows="to")
+        net.add_edge(
+            rel["subject"],
+            rel["object"],
+            label=rel["predicate"],
+            arrows="to",
+            title=rel.get("description", "")  # <-- Tooltip on hover
+        )
     html = net.generate_html()
     html_b64 = base64.b64encode(html.encode("utf-8")).decode("utf-8")
     return f'<iframe src="data:text/html;base64,{html_b64}" width="100%" height="350px" frameborder="0"></iframe>'
@@ -152,7 +158,7 @@ footer { display:none !important; }
 
 # Build the Gradio interface
 with gr.Blocks(css=css, fill_height=True, fill_width=True) as demo:
-    # gr.Markdown("# Data Explorer - [Dataset on Hugging Face](https://huggingface.co/datasets/nicpopovic/vital_articles_synthetic_information_extraction)")
+    #gr.Markdown("# Data Explorer - [Dataset on Hugging Face](https://huggingface.co/datasets/nicpopovic/vital_articles_synthetic_information_extraction)")
     idx = gr.Number(
         label="Row Index",
         value=0,