File size: 3,686 Bytes
b0bcb9b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 |
# Copyright Jiaqi Liu
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import os
import tempfile
from typing import Literal
from typing import Union
import gradio as gr
from openai import OpenAI
server_name = os.getenv("SERVER_NAME", "0.0.0.0")
def latin_replacement(text: str):
with open("ancient-greek-phonemes.txt", "r") as mapping_file:
mapping = dict(
[(ancient_greek, latin) for ancient_greek, latin in
[line.rstrip().split("->") for line in mapping_file.readlines()]]
)
for key, value in mapping.items():
text = text.replace(key, value)
return text
def tts(
openai_key: str,
text: str,
model: Union[str, Literal["tts-1", "tts-1-hd"]],
voice: Literal["alloy", "echo", "fable", "onyx", "nova", "shimmer"],
output_file_format: Literal["mp3", "opus", "aac", "flac"] = "mp3",
speed: float = 1.0
):
if len(text) > 0:
text = latin_replacement(text)
try:
client = OpenAI(api_key=openai_key)
response = client.audio.speech.create(
model=model,
voice=voice,
input=text,
response_format=output_file_format,
speed=speed
)
except Exception as error:
print(str(error))
raise gr.Error(
"An error occurred while generating speech. Please check your API key and come back try again.")
with tempfile.NamedTemporaryFile(suffix=".mp3", delete=False) as temp_file:
temp_file.write(response.content)
temp_file_path = temp_file.name
return temp_file_path
else:
return "1-second-of-silence.mp3"
with gr.Blocks() as app:
gr.Markdown("# <center> AI Doesn't Speak Ancient Greek. Aristotle Does! </center>")
with gr.Row(variant="panel"):
model = gr.Dropdown(choices=["tts-1", "tts-1-hd"], label="Model", value="tts-1")
voice = gr.Dropdown(choices=["alloy", "echo", "fable", "onyx", "nova", "shimmer"], label="Voice Options",
value="alloy")
output_file_format = gr.Dropdown(choices=["mp3", "opus", "aac", "flac"], label="Output Options", value="mp3")
speed = gr.Slider(minimum=0.25, maximum=4.0, value=1.0, step=0.01, label="Speed")
openai_key = gr.Textbox(
label="OpenAI API key",
placeholder="We can get the key at https://platform.openai.com/api-keys"
)
text = gr.Textbox(
label="Ancient Greek text",
placeholder="Enter your text and then click on the \"Text-To-Speech\" button, or simply press the Enter key."
)
examples = gr.Examples([[" Ὁμώνυμα λέγεται ὧν ὄνομα μόνον κοινόν"]], text)
btn = gr.Button("Hear It!")
output_audio = gr.Audio(label="Speech Output")
text.submit(fn=tts, inputs=[openai_key, text, model, voice, output_file_format, speed], outputs=output_audio,
api_name="tts")
btn.click(fn=tts, inputs=[openai_key, text, model, voice, output_file_format, speed], outputs=output_audio,
api_name=False)
app.launch(server_name=server_name)
|