fixed: user token

app.py

@@ -96,7 +96,7 @@ def get_answer(question):
     return response["choices"][0]["message"]["content"]
 
 def generate_rhythm(query, state):
-    if state["gen_count"] > MAX_QUERY:
+    if state["gen_count"] > MAX_QUERY or len(state["user_token"]) > 0:
         return [None, "You need to set your ChatGPT API Key to try more than %d times" % MAX_QUERY] 
     state["gen_count"] = state["gen_count"] + 1
 
@@ -123,10 +123,10 @@ def generate_rhythm(query, state):
 
 def on_token_change(user_token):
     openai.api_key = user_token or os.environ.get("OPENAI_API_KEY")
-
+    state["user_token"] = user_token
 
 with gr.Blocks() as demo:
-    state = gr.State({"gen_count": 0})
+    state = gr.State({"gen_count": 0, "user_token":""})
 
     gr.Markdown("Ask ChatGPT to generate rhythm patterns")
     with gr.Row():
@@ -136,17 +136,9 @@ with gr.Blocks() as demo:
         with gr.Column():
             out_audio = gr.Audio()
             out_text = gr.Textbox(placeholder="ChatGPT output")
-    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)
+    with gr.Column():
+        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=[])
 demo.launch()
-# demo = gr.Interface(
-#     fn=generate_rhythm,
-# inputs=gr.Textbox(label="command",show_label=True, placeholder="Give me a dope beat!", visible=True).style(container=False),
-#     outputs=["audio", "text"]
-# )
-# demo.launch()
-# %%
-
-# %%