Update app.py

app.py

@@ -121,15 +121,19 @@ def resume_graph(human_input_from_ui: str, thread_id_state):
         print(f"Type of global_memory_saver before put_state: {type(global_memory_saver)}") # ADD THIS LINE
 
         # Use put_state() from the SqliteSaver instance
-        global_memory_saver.put_state(current_state_values, {"configurable": {"thread_id": thread_id_state}})
+        #global_memory_saver.put_state(current_state_values, {"configurable": {"thread_id": thread_id_state}})
 
-        for s in global_graph.stream(None, {"configurable": {"thread_id": thread_id_state}}):
+        # Update the state directly
+        config = {"configurable": {"thread_id": thread_id_state}}
+        global_graph.update_state(config, current_state_values)
+
+        for s in global_graph.stream(None, config):
             if "__end__" in s:
                 break
             else:
                 pass 
 
-        final_state_snapshot = global_graph.get_state({"configurable": {"thread_id": thread_id_state}})
+        final_state_snapshot = global_graph.get_state(config)
         final_state_values = final_state_snapshot.values
 
         final_message = final_state_values.get("final_response", "Graph finished without final response.")
@@ -139,6 +143,7 @@ def resume_graph(human_input_from_ui: str, thread_id_state):
         return (f"An error occurred during graph resumption: {e}",
                 gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=True), thread_id_state)
 
+
 # Gradio Interface setup
 with gr.Blocks() as demo:
     gr.Markdown("# LangGraph Human-in-the-Loop Demo (SqliteSaver)")