Handle client-side refresh

app.py

@@ -120,15 +120,16 @@ def self_play():
     return self_play_history
 
 
+# Reset state upon client-side refresh
+reset_ds_state()
+reset_us_state()
+reset_self_play_state()
+
+
 # Initialise demo render
 block = gr.Blocks()
 
 with block:
-    # Reset state when app is refreshed client-side
-    ds_history = []
-    us_history = []
-    self_play_history = []
-
     gr.Markdown("# 💬 Jointly Optimized Task-Oriented Dialogue System And User Simulator 💬")
     gr.Markdown(
         "Created by [Alistair McLeay](https://alistairmcleay.com) for the [Masters in Machine Learning & Machine Intelligence at Cambridge University](https://www.mlmi.eng.cam.ac.uk/). <br/>\