Update app.py

app.py

@@ -148,6 +148,8 @@ with block:
     submit.click(chat, inputs=[message, state, agent_state], outputs=[chatbot, state])
     message.submit(chat, inputs=[message, state, agent_state], outputs=[chatbot, state])
 
+    agent_state = chain
+
     gr.Markdown("![visitor badge](https://visitor-badge.glitch.me/badge?page_id=nickmuchi-investor-chatchain)")
 
 block.launch(debug=True)