Update app.py

app.py

@@ -334,8 +334,12 @@ class DocumentRAG:
             ["story_generator"])
         graph.add_edge("story_generator", END)
 
+
         compiled = graph.compile(checkpointer=MemorySaver())
-        result = compiled.invoke({"topic": [topic], "context": [context]})
+        thread = {"configurable": {"thread_id": "storygraph-session"}}
+
+        result = compiled.invoke({"topic": [topic], "context": [context]}, thread)
+
         return result