Update workflow.py

workflow.py

@@ -14,8 +14,13 @@ from agents import (
     agent_3_integrated_node
 )
 
+# Global checkpointer to persist state between function calls
+_global_checkpointer = MemorySaver()
+
 def create_workflow():
     """Create the LangGraph workflow."""
+    global _global_checkpointer
+    
     workflow = StateGraph(WorkflowState)
     
     # Add Nodes
@@ -33,11 +38,8 @@ def create_workflow():
     workflow.add_edge("capture_website", "analysis")
     workflow.add_edge("analysis", END)
     
-    # Add persistence
-    checkpointer = MemorySaver()
-    
     return workflow.compile(
-        checkpointer=checkpointer,
+        checkpointer=_global_checkpointer,
         interrupt_before=["analysis"]
     )
 
@@ -80,4 +82,4 @@ def resume_workflow(thread_id, user_approval=True):
     for event in app.stream(None, config, stream_mode="values"):
         state = event
         
-    return app.get_state(config)
+    return app.get_state(config)