Update graph.py

graph.py

@@ -174,23 +174,23 @@ class EssayWriter:
             # Generate and save the graph in /tmp/
             dot.render(output_path, format="png", cleanup=False)
 
-            # Check if the file exists in /tmp/
+            # Check if file exists
             graph_path = "/tmp/graph.png"
             if os.path.exists(graph_path):
                 with open(log_file, "w") as f:
                     f.write("✅ Graphviz successfully generated /tmp/graph.png\n")
                 print("✅ Graphviz successfully generated /tmp/graph.png")  
             else:
-                with open(log_file, "w") as f:
-                    f.write("❌ Graphviz failed to generate /tmp/graph.png\n")
-                print("❌ Graphviz failed to generate /tmp/graph.png")  # Debugging print
-                raise FileNotFoundError("❌ Graphviz failed to generate the workflow graph.")
+                raise FileNotFoundError("❌ Graphviz failed to generate /tmp/graph.png")
 
         except Exception as e:
-            error_message = f"❌ Error generating workflow visualization: {e}\n"
+            # Capture full traceback
+            error_message = f"❌ Error generating workflow visualization:\n{traceback.format_exc()}\n"
             with open(log_file, "w") as f:
                 f.write(error_message)
-            print(error_message)  
+            print(error_message)
+            return error_message  
+