Update app.py

app.py

@@ -114,6 +114,68 @@ def to_turtle(df: pd.DataFrame, base_ns: str) -> str:
     ttl = g.serialize(format="turtle")
     return ttl.decode("utf-8") if hasattr(ttl, "decode") else ttl
 
+# ---------- NEW: RDF → rainbow graph (PyVis) ----------
+import hashlib, tempfile, pathlib, html
+
+def _rainbow_color(idx: int) -> str:
+    # 12-color rainbow palette (hex)
+    palette = [
+        "#e6194b","#ffa31a","#ffe119","#bfef45","#3cb44b","#42d4f4",
+        "#4363d8","#911eb4","#f032e6","#a9a9a9","#fabed4","#ffd8b1"
+    ]
+    return palette[idx % len(palette)]
+
+def _color_for_label(label: str) -> str:
+    # deterministic color from label
+    h = int(hashlib.sha1(label.encode("utf-8")).hexdigest(), 16)
+    return _rainbow_color(h)
+
+def turtle_to_pyvis_html(ttl_text: str):
+    try:
+        from pyvis.network import Network
+    except Exception:
+        # friendly guidance if pyvis isn't installed
+        msg = (
+            "<div style='padding:12px;border:1px solid #eee'>"
+            "<b>PyVis not installed.</b> Add <code>pyvis</code> to requirements.txt to enable the interactive graph preview."
+            "</div>"
+        )
+        return msg
+
+    if not ttl_text.strip():
+        return "<div style='padding:12px;border:1px solid #eee'>No Turtle to render. Export RDF first.</div>"
+
+    g = Graph()
+    try:
+        g.parse(data=ttl_text, format="turtle")
+    except Exception as e:
+        return f"<pre style='white-space:pre-wrap;color:#b00'>Parse error: {html.escape(str(e))}</pre>"
+
+    net = Network(height="620px", width="100%", directed=True, notebook=False, cdn_resources="remote")
+    net.toggle_physics(True)
+
+    def compact(term):
+        try:
+            return g.namespace_manager.normalizeUri(term)
+        except Exception:
+            return str(term)
+
+    nodes_seen = set()
+    for s, p, o in g:
+        sn, pn, on = compact(s), compact(p), compact(o)
+        if sn not in nodes_seen:
+            net.add_node(sn, label=sn, color=_color_for_label(sn), shape="dot")
+            nodes_seen.add(sn)
+        if on not in nodes_seen:
+            net.add_node(on, label=on, color=_color_for_label(on), shape="dot")
+            nodes_seen.add(on)
+        net.add_edge(sn, on, label=pn, color=_color_for_label(pn), arrows="to")
+
+    tmpdir = pathlib.Path(tempfile.mkdtemp())
+    out_html = tmpdir / "graph.html"
+    net.show(str(out_html))
+    return f'<iframe src="file/{out_html}" style="width:100%;height:640px;border:0" allowfullscreen></iframe>'
+
 # ---- Gradio UI (tiny) ----
 EXAMPLE = (
     "Yas Etessam is a content architect who works with large enterprises. "
@@ -145,6 +207,11 @@ with gr.Blocks(title="Relationship Helper — Minimal") as demo:
 
     turtle_out = gr.Code(label="RDF/Turtle", value="")  # ✅ safe initial value
 
+    # ---------- NEW: preview controls at the bottom ----------
+    with gr.Row():
+        btn_preview = gr.Button("Preview Graph 🌈")
+    graph_html = gr.HTML(label="Knowledge Graph Preview (rainbow)")
+
     def on_extract(text: str):
         rows = extract_triples(text) if text.strip() else []
         return pd.DataFrame(rows, columns=["subject", "relation", "object", "confidence"])
@@ -157,10 +224,16 @@ with gr.Blocks(title="Relationship Helper — Minimal") as demo:
             return "# (No rows to export)"
         return to_turtle(data, base or "https://example.org/")
 
+    def on_preview(ttl_text: str):
+        return turtle_to_pyvis_html(ttl_text)
+
     btn.click(on_extract, inputs=text_in, outputs=df)
     btn_clear.click(on_clear, outputs=[text_in, df, turtle_out])
     btn_export.click(on_export, inputs=[df, base_ns], outputs=turtle_out)
 
+    # NEW: wire up preview
+    btn_preview.click(on_preview, inputs=turtle_out, outputs=graph_html)
+
 # For Hugging Face Spaces, expose on 0.0.0.0 and allow shareable link
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860, share=True)