File size: 2,192 Bytes
b36a86c |
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 |
from dotenv import find_dotenv, load_dotenv
import requests
import os
import shutil
import subprocess
from pathlib import Path
from typing import Iterator, Optional
load_dotenv(find_dotenv())
def get_voices():
url = "https://api.elevenlabs.io/v1/voices"
api_key = os.getenv("ELEVEN_API_KEY")
headers = {"Accept": "application/json", "xi-api-key": api_key}
response = requests.get(url, headers=headers)
print(response.text)
return response.text
def is_installed(lib_name: str) -> bool:
lib = shutil.which(lib_name)
if lib is None:
return False
global_path = Path(lib)
# else check if path is valid and has the correct access rights
return global_path.exists() and os.access(global_path, os.X_OK)
def play(audio: bytes, notebook: bool = False) -> None:
if notebook:
from IPython.display import Audio, display
display(Audio(audio, rate=44100, autoplay=True))
else:
if not is_installed("ffplay"):
raise ValueError("ffplay from ffmpeg not found, necessary to play audio.")
args = ["ffplay", "-autoexit", "-", "-nodisp"]
proc = subprocess.Popen(
args=args,
stdout=subprocess.PIPE,
stdin=subprocess.PIPE,
stderr=subprocess.PIPE,
)
_, err = proc.communicate(input=audio)
proc.poll()
def save(audio: bytes, filename: str) -> None:
with open(filename, "wb") as f:
f.write(audio)
def stream(audio_stream: Iterator[bytes]) -> bytes:
if not is_installed("mpv"):
raise ValueError("mpv not found, necessary to stream audio.")
mpv_command = ["mpv", "--no-cache", "--no-terminal", "--", "fd://0"]
mpv_process = subprocess.Popen(
mpv_command,
stdin=subprocess.PIPE,
stdout=subprocess.DEVNULL,
stderr=subprocess.DEVNULL,
)
audio = b""
for chunk in audio_stream:
if chunk is not None:
mpv_process.stdin.write(chunk) # type: ignore
mpv_process.stdin.flush() # type: ignore
audio += chunk
if mpv_process.stdin:
mpv_process.stdin.close()
mpv_process.wait()
return audio
|