rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Full Audio status message

Browse Source
This commit is contained in:
ebolam
2023-09-27 14:56:26 -04:00
parent e76679feb6
commit 30110888dd
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
aiserver.py
View File

@@ -7940,6 +7940,7 @@ def UI_2_audio_full():
if args.no_ui:
return redirect('/api/latest')
logger.info("Downloading complete audio file")
combined_audio = None
for action_id in range(-1, koboldai_vars.actions.action_count+1):
filename = os.path.join(koboldai_vars.save_paths.generated_audio, f"{action_id}.ogg")
@@ -7957,6 +7958,7 @@ def UI_2_audio_full():
else:
combined_audio = combined_audio + AudioSegment.from_file(filename, format="ogg")
else:
logger.info("Action {} has no audio. Generating now".format(action_id))
koboldai_vars.actions.gen_audio(action_id)
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)
@@ -7968,6 +7970,7 @@ def UI_2_audio_full():
else:
show_error_notification("Error generating audio chunk", f"Something happened. Maybe check the log?")
logger.info("Sending audio file")
file_handle = combined_audio.export(complete_filename, format="ogg")
return send_file(