Spaces:
Sleeping
Sleeping
| #!/usr/bin/python3 | |
| # -*- coding: utf-8 -*- | |
| import argparse | |
| import asyncio | |
| from functools import partial | |
| import json | |
| import logging | |
| from pathlib import Path | |
| import platform | |
| import re | |
| import tempfile | |
| from typing import List | |
| import uuid | |
| from project_settings import project_path, log_directory, edge_tts_record_file, edge_tts_temp_directory | |
| import log | |
| log.setup(log_directory=log_directory) | |
| import aiofiles | |
| import anyio | |
| import edge_tts | |
| import gradio as gr | |
| from toolbox.os.command import Command | |
| main_logger = logging.getLogger("main") | |
| def get_args(): | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument( | |
| "--example_wav_dir", | |
| default=(project_path / "data/examples").as_posix(), | |
| type=str | |
| ) | |
| args = parser.parse_args() | |
| return args | |
| async def edge_tts_get_speakers() -> List[str]: | |
| edge_tts_speakers_choices = list() | |
| voices = await edge_tts.list_voices() | |
| for voice in voices: | |
| short_name = voice["ShortName"] | |
| edge_tts_speakers_choices.append(short_name) | |
| return edge_tts_speakers_choices | |
| async def edge_tts_text_to_speech(text: str, speaker: str): | |
| communicate = edge_tts.Communicate(text, speaker) | |
| filename = edge_tts_temp_directory / "{}.wav".format(uuid.uuid4()) | |
| async with aiofiles.open(edge_tts_record_file.as_posix(), "a+", encoding="utf-8") as f: | |
| await f.write(json.dumps({ | |
| "text": text, | |
| "speaker": speaker, | |
| "filename": filename.as_posix(), | |
| }, ensure_ascii=False)) | |
| await communicate.save(filename) | |
| return filename | |
| def shell(cmd: str): | |
| return Command.popen(cmd) | |
| def main(): | |
| args = get_args() | |
| title = "## 电子书阅读." | |
| loop = asyncio.get_event_loop() | |
| edge_tts_speakers_choices = loop.run_until_complete(edge_tts_get_speakers()) | |
| # blocks | |
| with gr.Blocks() as blocks: | |
| gr.Markdown(value=title) | |
| with gr.Tabs(): | |
| with gr.TabItem("Ebook Reading"): | |
| e_book_reading_file = gr.File( | |
| value=(project_path / "data/e_book/confucianism/the_analects.txt").as_posix(), | |
| label="txt" | |
| ) | |
| e_book_reading_tts_engine = gr.Dropdown( | |
| choices=["Edge TTS"], value="Edge TTS", label="tts_engine" | |
| ) | |
| e_book_reading_button = gr.Button(value="e_book_reading", variant="primary") | |
| with gr.TabItem("Edge TTS"): | |
| edge_tts_text = gr.Textbox(value="学而时习之，不亦悦乎。", lines=4, max_lines=50, label="text") | |
| edge_tts_speaker = gr.Dropdown(choices=edge_tts_speakers_choices, value="zh-CN-XiaoxiaoNeural", label="speakers") | |
| edge_tts_audio = gr.Audio(type="filepath", label="audio", autoplay=True) | |
| edge_tts_button = gr.Button(value="edge_tts", variant="primary") | |
| edge_tts_button.click( | |
| edge_tts_text_to_speech, | |
| inputs=[ | |
| edge_tts_text, | |
| edge_tts_speaker, | |
| ], | |
| outputs=[ | |
| edge_tts_audio | |
| ], | |
| ) | |
| with gr.TabItem("shell"): | |
| shell_text = gr.Textbox(label="cmd") | |
| shell_button = gr.Button("run") | |
| shell_output = gr.Textbox(label="output") | |
| shell_button.click( | |
| shell, | |
| inputs=[ | |
| shell_text, | |
| ], | |
| outputs=[ | |
| shell_output | |
| ], | |
| ) | |
| launch = partial( | |
| blocks.queue().launch, | |
| share=False if platform.system() == "Windows" else False, | |
| server_name="127.0.0.1" if platform.system() == "Windows" else "0.0.0.0", | |
| server_port=7860, | |
| ) | |
| anyio.run( | |
| launch, | |
| backend="asyncio" | |
| ) | |
| return | |
| if __name__ == "__main__": | |
| main() | |