mirror of
https://github.com/Bubberstation/Bubberstation.git
synced 2025-12-26 09:32:21 +00:00
6c9be73f51
* Adds TTS to the game. Players can select their own voices in preferences. * [SEMI-MODULAR] [MIRROR FIX] Fixes the TTS PR. (#21267) Fixes the TTS PR. --------- Co-authored-by: Watermelon914 <37270891+Watermelon914@users.noreply.github.com> Co-authored-by: Iamgoofball <iamgoofball@gmail.com>
55 lines
1.9 KiB
Python
55 lines
1.9 KiB
Python
import os
|
|
import io
|
|
import gc
|
|
import subprocess
|
|
import requests
|
|
from flask import Flask, request, send_file, abort
|
|
|
|
app = Flask(__name__)
|
|
|
|
authorization_token = os.get_env("TTS_AUTHORIZATION_TOKEN", "coolio")
|
|
|
|
@app.route("/tts")
|
|
def text_to_speech():
|
|
if authorization_token != request.headers.get("Authorization", ""):
|
|
abort(401)
|
|
|
|
voice = request.args.get("voice", '')
|
|
text = request.json.get("text", '')
|
|
|
|
filter_complex = request.args.get("filter", '')
|
|
filter_complex = filter_complex.replace("\"", "")
|
|
|
|
response = requests.get(f"http://tts-container:5003/generate-tts", json={ 'text': text, 'voice': voice })
|
|
if response.status_code != 200:
|
|
abort(500)
|
|
|
|
ffmpeg_result = None
|
|
if filter_complex != "":
|
|
ffmpeg_result = subprocess.run(["ffmpeg", "-f", "wav", "-i", "pipe:0", "-filter_complex", filter_complex, "-c:a", "libvorbis", "-b:a", "64k", "-f", "ogg", "pipe:1"], input=response.content, capture_output = True)
|
|
else:
|
|
ffmpeg_result = subprocess.run(["ffmpeg", "-f", "wav", "-i", "pipe:0", "-c:a", "libvorbis", "-b:a", "64k", "-f", "ogg", "pipe:1"], input=response.content, capture_output = True)
|
|
print(f"ffmpeg result size: {len(ffmpeg_result.stdout)} stderr = \n{ffmpeg_result.stderr.decode()}")
|
|
|
|
return send_file(io.BytesIO(ffmpeg_result.stdout), as_attachment=True, download_name='identifier.ogg', mimetype="audio/ogg")
|
|
|
|
|
|
@app.route("/tts-voices")
|
|
def voices_list():
|
|
if authorization_token != request.headers.get("Authorization", ""):
|
|
abort(401)
|
|
|
|
response = requests.get(f"http://tts-container:5003/tts-voices")
|
|
return response.content
|
|
|
|
@app.route("/health-check")
|
|
def tts_health_check():
|
|
gc.collect()
|
|
return "OK", 200
|
|
|
|
if __name__ == "__main__":
|
|
if os.getenv('TTS_LD_LIBRARY_PATH', "") != "":
|
|
os.putenv('LD_LIBRARY_PATH', os.getenv('TTS_LD_LIBRARY_PATH'))
|
|
from waitress import serve
|
|
serve(app, host="0.0.0.0", port=5002, threads=2, backlog=8, connection_limit=24, channel_timeout=10)
|