Update app.py

app.py

@@ -24,6 +24,13 @@ def get_note_text(prompt):
         stop=["###"]
     )
     return response.choices[0].text.strip()
+    
+def increment_count():
+    with open('count.txt', 'r') as f:
+        count = int(f.read())
+    count += 1
+    with open('count.txt', 'w') as f:
+        f.write(str(count))
 
 def get_drummer_output(prompt, tempo):
     openai.api_key = os.environ['key']
@@ -37,6 +44,7 @@ def get_drummer_output(prompt, tempo):
     # bpm = int(reg.predict(prompt_enc)[0]) + 20
     audio = text_to_audio(note_text, tempo)
     audio = np.array(audio.get_array_of_samples(), dtype=np.float32)
+    increment_count()
     return (96000, audio)
 
 iface = gr.Interface(