Update app.py

app.py

@@ -81,19 +81,21 @@ def get_state(messages):
             return "profile"
     return "info"
 
-
-
-memory = SqliteSaver.from_conn_string(":memory:")
-
-nodes = {k:k for k in ['info', 'profile', END]}
-workflow = MessageGraph()
-workflow.add_node("info", chain)
-workflow.add_node("profile", profile_gen_chain)
-workflow.add_conditional_edges("info", get_state, nodes)
-workflow.add_conditional_edges("profile", get_state, nodes)
-workflow.set_entry_point("info")
-graph = workflow.compile(checkpointer=memory)
-
+@st.cache_resource
+def get_graph():
+    memory = SqliteSaver.from_conn_string(":memory:")
+    
+    nodes = {k:k for k in ['info', 'profile', END]}
+    workflow = MessageGraph()
+    workflow.add_node("info", chain)
+    workflow.add_node("profile", profile_gen_chain)
+    workflow.add_conditional_edges("info", get_state, nodes)
+    workflow.add_conditional_edges("profile", get_state, nodes)
+    workflow.set_entry_point("info")
+    graph = workflow.compile(checkpointer=memory)
+    return graph
+    
+graph = get_graph()
 config = {"configurable": {"thread_id": str(uuid.uuid4())}}
 
 # Streamlit app layout