From f2c93d914f6c928f3599bb1ea5a9fd70c11e83fd Mon Sep 17 00:00:00 2001
From: ebolam <ebolam@gmail.com>
Date: Tue, 6 Dec 2022 08:20:03 -0500
Subject: [PATCH] Fix for requesting audio before it's done generating

---
 aiserver.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/aiserver.py b/aiserver.py
index 6a3aa4c1..18a1aa20 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -9891,8 +9891,12 @@ def UI_2_test_match():
 def UI_2_audio():
     action_id = int(request.args['id']) if 'id' in request.args else koboldai_vars.actions.action_count
     filename = os.path.join(koboldai_vars.save_paths.generated_audio, f"{action_id}.ogg")
+    
     if not os.path.exists(filename):
         koboldai_vars.actions.gen_audio(action_id)
+        start_time = time.time()
+        while not os.path.exists(filename) and time.time()-start_time < 60: #Waiting up to 60 seconds for the file to be generated
+            time.sleep(0.1)
     return send_file(
              filename, 
              mimetype="audio/ogg")