4f5d79ed57
WebVTT isn't a very common format in the first place and attempting to make WordBoundary play nice with input text is very hard. Instead we now just display the word that the TTS is saying at a given time. In the future, we could try to enable SentenceBoundary but there is a risk that it will be banned by Microsoft as it is not used by Microsoft Edge itself. Closes: https://github.com/rany2/edge-tts/issues/118 Closes: https://github.com/rany2/edge-tts/issues/171 Closes: https://github.com/rany2/edge-tts/issues/229 Closes: https://github.com/rany2/edge-tts/issues/234 Signed-off-by: rany <rany2@riseup.net>
73 lines
2.0 KiB
Python
73 lines
2.0 KiB
Python
"""Main entrypoint for the edge-playback package."""
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
from shutil import which
|
|
|
|
|
|
def pr_err(msg: str) -> None:
|
|
"""Print to stderr."""
|
|
print(msg, file=sys.stderr)
|
|
|
|
|
|
def _main() -> None:
|
|
depcheck_failed = False
|
|
for dep in ("edge-tts", "mpv"):
|
|
if not which(dep):
|
|
pr_err(f"{dep} is not installed.")
|
|
depcheck_failed = True
|
|
|
|
if depcheck_failed:
|
|
pr_err("Please install the missing dependencies.")
|
|
sys.exit(1)
|
|
|
|
keep = os.environ.get("EDGE_PLAYBACK_KEEP_TEMP") is not None
|
|
mp3_fname = os.environ.get("EDGE_PLAYBACK_MP3_FILE")
|
|
srt_fname = os.environ.get("EDGE_PLAYBACK_SRT_FILE")
|
|
media, subtitle = None, None
|
|
try:
|
|
if not mp3_fname:
|
|
media = tempfile.NamedTemporaryFile(suffix=".mp3", delete=False)
|
|
media.close()
|
|
mp3_fname = media.name
|
|
|
|
if not srt_fname:
|
|
subtitle = tempfile.NamedTemporaryFile(suffix=".srt", delete=False)
|
|
subtitle.close()
|
|
srt_fname = subtitle.name
|
|
|
|
print(f"Media file: {mp3_fname}")
|
|
print(f"Subtitle file: {srt_fname}\n")
|
|
with subprocess.Popen(
|
|
[
|
|
"edge-tts",
|
|
f"--write-media={mp3_fname}",
|
|
f"--write-subtitles={srt_fname}",
|
|
]
|
|
+ sys.argv[1:]
|
|
) as process:
|
|
process.communicate()
|
|
|
|
with subprocess.Popen(
|
|
[
|
|
"mpv",
|
|
f"--sub-file={srt_fname}",
|
|
mp3_fname,
|
|
]
|
|
) as process:
|
|
process.communicate()
|
|
finally:
|
|
if keep:
|
|
print(f"\nKeeping temporary files: {mp3_fname} and {srt_fname}")
|
|
else:
|
|
if mp3_fname is not None and os.path.exists(mp3_fname):
|
|
os.unlink(mp3_fname)
|
|
if srt_fname is not None and os.path.exists(srt_fname):
|
|
os.unlink(srt_fname)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
_main()
|