dump cached sample paths to JSON file and load it on Space reset

app/sample_caching.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import itertools
 import random
+import json
 from typing import List, Tuple, Set, Dict
 from hashlib import md5, sha1
 import spaces
@@ -17,8 +18,26 @@ class Sample:
         self.transcript = transcript
         self.modelName = modelName
 
+    def to_dict(self):
+        return {
+            'filename': self.filename,
+            'transcript': self.transcript,
+            'modelName': self.modelName,
+        }
+
 # cache audio samples for quick voting
 cached_samples: List[Sample] = []
+# cached_samples.append(Sample("audio1.mp3", "Hello, how are you?", "model1"))
+# cached_samples.append(Sample("audio2.mp3", "Hello, how are you?", "model2"))
+
+# load temporary samples
+json_data = ''
+try:
+    with open("_cached_samples.json", "r") as read:
+        loaded_samples = json.load(read)
+    cached_samples = [Sample(**json_data) for json_data in loaded_samples]
+except:
+    pass
 
 @spaces.GPU(duration=10)
 def asr_cached_for_dataset():

app/synth.py

@@ -288,6 +288,14 @@ def synthandreturn(text, autoplay, request: gr.Request):
             print('Error when trying to cache sample')
             return False
 
+        # save list to JSON file
+        cached_sample_dict = [cached_sample.to_dict() for cached_sample in cached_samples]
+        try:
+            with open("_cached_samples.json", "w") as write:
+                json.dump( cached_sample_dict , write )
+        except:
+            pass
+
     mdl1k = mdl1
     mdl2k = mdl2
     print(mdl1k, mdl2k)
@@ -331,7 +339,7 @@ def synthandreturn(text, autoplay, request: gr.Request):
     print(f"Retrieving models {mdl1k} and {mdl2k} from API")
     return (
         text,
-        "Synthesize",
+        "Synthesize 🐢",
         gr.update(visible=True), # r2
         mdl1, # model1
         mdl2, # model2
@@ -430,7 +438,7 @@ def synthandreturn_battle(text, mdl1, mdl2, autoplay):
     print(f"Retrieving models {mdl1k} and {mdl2k} from API")
     return (
         text,
-        "Synthesize",
+        "Synthesize 🐢",
         gr.update(visible=True), # r2
         mdl1, # model1
         mdl2, # model2
@@ -450,7 +458,7 @@ def randomsent_battle():
 def clear_stuff():
     return [
         gr.update(visible=True, value="", elem_classes=[]),
-        "Synthesize",
+        "Synthesize 🐢",
         gr.update(visible=False), # r2
         '', # model1
         '', # model2