From 8944f69e4bd1f6bc2509d4e1e0c423d1b0209de8 Mon Sep 17 00:00:00 2001
From: ebolam <ebolam@gmail.com>
Date: Wed, 27 Sep 2023 13:58:06 -0400
Subject: [PATCH] More Audio enhancements

---
 aiserver.py              | 8 +++++---
 koboldai_settings.py     | 2 ++
 templates/index_new.html | 1 +
 3 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/aiserver.py b/aiserver.py
index b656f646..2457cb3e 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -7923,9 +7923,11 @@ def UI_2_audio():
         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")
+    if os.path.exists(filename):
+        return send_file(
+                filename, 
+                mimetype="audio/ogg")
+    show_error_notification("Error generating audio chunk", f"Something happened. Maybe check the log?")
 
 
 #==================================================================#
diff --git a/koboldai_settings.py b/koboldai_settings.py
index b7408b93..aa5a39c3 100644
--- a/koboldai_settings.py
+++ b/koboldai_settings.py
@@ -2058,6 +2058,7 @@ class KoboldStoryRegister(object):
             if action_id is None:
                 action_id = self.action_count
 
+            logger.info("Got request to generate audio for {}".format(action_id))
             if self.tts_model is None:
                 language = 'en'
                 model_id = 'v3_en'
@@ -2071,6 +2072,7 @@ class KoboldStoryRegister(object):
             filename = os.path.join(self._koboldai_vars.save_paths.generated_audio, f"{action_id}.ogg")
             filename_slow = os.path.join(self._koboldai_vars.save_paths.generated_audio, f"{action_id}_slow.ogg")
                 
+            logger.info("Got request to generate audio for {}".format(filename))
             if overwrite or not os.path.exists(filename):
                 if action_id == -1:
                     self.make_audio_queue.put((self._koboldai_vars.prompt, filename))
diff --git a/templates/index_new.html b/templates/index_new.html
index 0f2fecaf..9bfd08dd 100644
--- a/templates/index_new.html
+++ b/templates/index_new.html
@@ -109,6 +109,7 @@
 			<span class="tts_controls hidden var_sync_alt_story_gen_audio">
 					<button type="button" class="btn action_button" style="width: 30px; padding: 0px;" onclick='play_pause_tts()' aria-label="play"><span id="play_tts" class="material-icons-outlined" style="font-size: 1.4em;">play_arrow</span></button>
 					<button type="button" class="btn action_button" style="width: 30px; padding: 0px;" onclick='stop_tts()' aria-label="play"><span id="stop_tts" class="material-icons-outlined" style="font-size: 1.4em;">stop</span></button>
+					<button type="button" class="btn action_button" style="width: 30px; padding: 0px;" onclick='download_tts()' aria-label="play"><span id="stop_tts" class="material-icons-outlined" style="font-size: 1.4em;">download</span></button>
 			</span>
 			<button type="button" class="btn action_button submit var_sync_alt_system_aibusy" system_aibusy=False id="btnsubmit" onclick="storySubmit();" context-menu="submit-button">Submit</button>
 			<button type="button" class="btn action_button submited var_sync_alt_system_aibusy"  system_aibusy=False id="btnsent"><img id="thinking" src="static/thinking.gif" class="force_center" onclick="socket.emit('abort','');"></button>