Update graph.py

graph.py

@@ -150,10 +150,10 @@ class EssayWriter:
     import os
     import graphviz
 
-    def save_workflow_graph(self):
-        """Generate and save a Graphviz workflow visualization with logging."""
-        log_file = "/tmp/graph_debug.log"  
-        try:
+    #def save_workflow_graph(self):
+    #    """Generate and save a Graphviz workflow visualization with logging."""
+    #    log_file = "/tmp/graph_debug.log"  
+    #    try:
             output_path = "/tmp/graph"  
             dot = graphviz.Digraph(format="png")
             dot.attr(dpi='300')
@@ -194,4 +194,16 @@ class EssayWriter:
 
 
 
+    def save_workflow_graph(self):
+        """Generate and save a dynamic LangGraph visualization."""
+        try:
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmpfile:
+                with open(tmpfile.name, "wb") as f:
+                    f.write(self.graph.get_graph().draw_mermaid_png())
+
+                print(f"✅ Workflow visualization updated: {tmpfile.name}")
+        except Exception as e:
+            print(f"❌ Error generating graph: {e}")
+
+