Update graph.py

graph.py

@@ -144,3 +144,30 @@ class EssayWriter:
             "essay": self.essay,
             "pdf_name": pdf_name,
         }
+        
+    def save_workflow_graph(self):
+        """Generate and save a Graphviz workflow visualization."""
+        try:
+            dot = graphviz.Digraph(format="png")
+            dot.attr(dpi='300')
+
+            # Define Nodes
+            dot.node("Router", "🔀 Router")
+            dot.node("Write Essay", "📝 Write Essay")
+            dot.node("Edit Essay", "✏️ Edit Essay")
+            dot.node("Answer", "💬 Answer")
+
+            # Define Edges
+            dot.edge("Router", "Write Essay")
+            dot.edge("Router", "Edit Essay")
+            dot.edge("Router", "Answer")
+            dot.edge("Write Essay", "✅ Done")
+            dot.edge("Edit Essay", "✅ Done")
+            dot.edge("Answer", "✅ Done")
+
+            # Save Graph
+            dot.render("graph")  # Saves as 'graph.png'
+
+        except Exception as e:
+            print(f"❌ Error generating workflow visualization: {e}")
+