minimal circuit.

app.py

@@ -1,11 +1,12 @@
 import os
 import glob
 from pathlib import Path
-import traceback
 
 import gradio as gr
 import jpype
 import jpype.imports
+
+from graphviz import Digraph
 from jpype import JClass, getDefaultJVMPath
 
 def _find_psl_jars() -> list[str]:
@@ -20,7 +21,6 @@ def _find_psl_jars() -> list[str]:
     # 2) Local ./jars
     this_dir = Path(__file__).resolve().parent
     jars_dir = f"{this_dir}/.jars"
-    print(f"Jars dir is {jars_dir}")
     jars.extend(glob.glob(f"{jars_dir}/*.jar"))
 
     # Deduplicate while preserving order
@@ -171,30 +171,43 @@ def _pred_name(lit: str) -> str:
 def _is_negated(lit: str) -> bool:
     return lit.strip().startswith('!')
 
-def rules_to_graphviz_image_bytes(rules=RULES) -> bytes:
+def rules_to_graphviz_file(rules=RULES, basename: str = "rules_graph") -> str:
+    """
+    Render Graphviz to a PNG on disk and return the filepath.
+    Produces 'basename.png' next to your app.
+    """
     g = Digraph(name="CircuitMinimality", format="png", engine="dot")
     g.attr(rankdir="LR", fontname="Helvetica")
     g.node_attr.update(shape="box", style="rounded,filled", fillcolor="#f8f8f8", fontname="Helvetica")
 
+    def pred_name(lit: str) -> str:
+        return lit.lstrip('!').split('(')[0].strip()
+
+    def is_negated(lit: str) -> bool:
+        return lit.strip().startswith('!')
+
     # Nodes
     preds = set()
     for r in rules:
-        preds.update(_pred_name(x) for x in (r["body"] + [r["head"]]))
+        preds.update(pred_name(x) for x in (r["body"] + [r["head"]]))
     for p in sorted(preds):
         g.node(p)
 
     # Edges
     for r in rules:
         head_lit = r["head"]
-        head_pred = _pred_name(head_lit)
-        neg_head = _is_negated(head_lit)
+        head_pred = pred_name(head_lit)
+        neg_head = is_negated(head_lit)
         color = "#2ca02c" if not neg_head else "#d62728"
         style = "solid" if not neg_head else "dashed"
         label = f"{r['name']} ({'HARD' if r.get('hard', False) else r.get('weight', '')})".strip()
         for b in r["body"]:
-            g.edge(_pred_name(b), head_pred, color=color, fontcolor=color, style=style, penwidth="2", label=label)
+            g.edge(pred_name(b), head_pred, color=color, fontcolor=color, style=style, penwidth="2", label=label)
 
-    return g.pipe(format="png")
+    # Render to file: returns full path without extension; add '.png'
+    out = g.render(filename=basename, cleanup=True)  # writes basename.png
+    png_path = out if out.endswith(".png") else f"{out}.png"
+    return png_path
 
 def _rules_commentary_md(rules=RULES) -> str:
     lines = ["### Rule Commentary", ""]
@@ -227,14 +240,13 @@ add_rules(model, RULES)
 
 # 3) Build commentary and graph image
 COMMENTARY_MD = _rules_commentary_md(RULES)
-GRAPH_PNG = rules_to_graphviz_image_bytes(RULES)
+GRAPH_PATH = rules_to_graphviz_file(RULES, basename="rules_graph")
 
-# --- Static Gradio UI ---
-with gr.Blocks(title="PSL Minimal-Circuit Rules") as demo:
-    gr.Markdown("## PSL Minimal-Circuit • Overview")
+with gr.Blocks(title="PSL Minimal-Circuit Rules (Static)") as demo:
+    gr.Markdown("## PSL Minimal-Circuit • Static Overview")
     gr.Markdown(f"**JVM status:**\n\n{JVM_STATUS}")
     gr.Markdown(COMMENTARY_MD)
-    gr.Image(value=GRAPH_PNG, label="Rule Graph (green = positive head, red dashed = negated head)")
+    gr.Image(value=GRAPH_PATH, label="Rule Graph (green = positive head, red dashed = negated head)")
 
 if __name__ == "__main__":
     demo.launch()