fixed: state

app.py

@@ -122,9 +122,10 @@ def generate_rhythm(query, state):
     return [(SR, audio_data), text_output] 
 # %%
 
-def on_token_change(user_token):
+def on_token_change(user_token, state):
     openai.api_key = user_token or os.environ.get("OPENAI_API_KEY")
     state["user_token"] = user_token
+    return state
 
 with gr.Blocks() as demo:
     state = gr.State({"gen_count": 0, "user_token":""})
@@ -141,5 +142,5 @@ with gr.Blocks() as demo:
         gr.Markdown("Enter your own OpenAI API Key to try out more than 5 times. You can get it [here](https://platform.openai.com/account/api-keys).", elem_id="label")
         user_token = gr.Textbox(placeholder="OpenAI API Key", type="password", show_label=False)
     btn.click(fn=generate_rhythm, inputs=[inp, state], outputs=[out_audio, out_text])
-    user_token.change(on_token_change, inputs=[user_token], outputs=[])
+    user_token.change(on_token_change, inputs=[user_token, state], outputs=[state])
 demo.launch()