Save generation number

app.py

@@ -13,6 +13,10 @@ import gradio as gr
 from TTS.api import TTS
 from TTS.utils.manage import ModelManager
 
+default_local_storage = {
+        "language-id": "en",
+        "generation-number-id": 1,
+    }
 max_64_bit_int = 2**63 - 1
 model_names = TTS().list_models()
 print(model_names.__dict__)
@@ -39,12 +43,18 @@ def save_preferences(preferences, value):
     preferences["language-id"] = value
     return preferences
 
+def save_preferences_generation_number(preferences, value):
+    preferences["generation-number-id"] = value
+    return preferences
+
 def load_preferences(saved_prefs):
+    saved_prefs = init_preferences(saved_prefs)
+    return [saved_prefs["language-id"], saved_prefs["generation-number-id"]]
+
+def init_preferences(saved_prefs):
     if saved_prefs is None:
-        saved_prefs = {
-            "language-id": "en",
-            }
-    return saved_prefs["language-id"]
+        saved_prefs = default_local_storage
+    return saved_prefs
 
 def update_output(output_number):
     return [
@@ -200,8 +210,7 @@ def predict_on_gpu(
     )
 
 with gr.Blocks() as interface:
-    local_storage = gr.BrowserState({ #"user_preferences":
-        "language-id": "en",})
+    local_storage = gr.BrowserState(default_local_storage)
     gr.HTML(
         """
         <h1><center>XTTS</center></h1>
@@ -251,6 +260,7 @@ Leave a star on the Github <a href="https://github.com/coqui-ai/TTS">TTS</a>, wh
             ],
             max_choices=1,
             elem_id = "language-id",
+            value = "en",
             )
             gr.HTML("More languages <a href='https://huggingface.co/spaces/Brasd99/TTS-Voice-Cloner'>here</a>")
         gender = gr.Radio(
@@ -396,6 +406,13 @@ You can also install XTTS on your computer using docker but it's more complicate
     ], outputs = [
        local_storage
     ])
+
+    generation_number.change(fn = save_preferences_generation_number, inputs = [
+        local_storage,
+        generation_number,
+    ], outputs = [
+       local_storage
+    ])
     
     submit.click(fn = update_output, inputs = [
         generation_number
@@ -542,7 +559,8 @@ You can also install XTTS on your computer using docker but it's more complicate
         fn=load_preferences, inputs = [
         local_storage
     ], outputs = [
-       language
+       language,
+       generation_number
     ]
     )