fixed: API key

app.py

@@ -86,18 +86,19 @@ def get_answer(question):
     response = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
         messages=[
-            {"role": "system", "content": "You are a rhythm generator. You generate rhythm patterns with the resolution of the 8th note"},
-            {"role": "user", "content": "Please generate a rhythm pattern. You use the following drums. Kick drum:BD, Snare drum:SD, Closed-hihat:CH, Open-hihat:OH, Low-tom:LT, Mid-tom:MT, High-tom:HT。You need to write the time resolution first."},
+            {"role": "system", "content": "You are a rhythm generator."},
+            {"role": "user", "content": "Please generate a rhythm pattern. You use the following drums. Kick drum:BD, Snare drum:SD, Closed-hihat:CH, Open-hihat:OH, Low-tom:LT, Mid-tom:MT, High-tom:HT. use 'x' for an accented beat, 'o' for a weak beat. You need to write the time resolution first."},
             {"role": "assistant", "content": markdown_table_sample2},
-            # {"role": "user", "content": "4分音符単位で生成して下さい. ドラムはBD, SD, CH, OH, LT, MT, HTを使います。最初に時間解像度の逆数を書いて下さい"},
-            # {"role": "assistant", "content": markdown_table_sample},
+            {"role": "user", "content": "Please generate a rhythm pattern. The resolution is the fourth note. You use the following drums. Kick drum:BD, Snare drum:SD, Closed-hihat:CH, Open-hihat:OH, Low-tom:LT, Mid-tom:MT, High-tom:HT. use 'x' for an accented beat, 'o' for a weak beat. You need to write the time resolution first."},
+            {"role": "assistant", "content": markdown_table_sample},
             {"role": "user", "content": question}
         ]
     )
     return response["choices"][0]["message"]["content"]
 
 def generate_rhythm(query, state):
-    if state["gen_count"] > MAX_QUERY or len(state["user_token"]) > 0:
+    print(state)
+    if state["gen_count"] > MAX_QUERY and 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,6 +124,7 @@ def generate_rhythm(query, state):
 # %%
 
 def on_token_change(user_token, state):
+    print(user_token)
     openai.api_key = user_token or os.environ.get("OPENAI_API_KEY")
     state["user_token"] = user_token
     return state