Spaces:
Building
Building
File size: 2,997 Bytes
f6b56a2 e2d04e8 f6b56a2 e2d04e8 f6b56a2 e2d04e8 f6b56a2 e2d04e8 f6b56a2 e2d04e8 f6b56a2 |
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 |
from gyraudio.audio_separation.properties import CLEAN, NOISY, MIXED, PREDICTED, SAMPLING_RATE
from pathlib import Path
from gyraudio.io.audio import save_audio_tensor
from gyraudio import root_dir
from interactive_pipe import Control, KeyboardControl
from interactive_pipe import interactive
import logging
HERE = Path(__file__).parent
MUTE = "mute"
LOGOS = {
PREDICTED: HERE/"play_logo_pred.png",
MIXED: HERE/"play_logo_mixed.png",
CLEAN: HERE/"play_logo_clean.png",
# NOISY: HERE/"play_logo_noise.png",
# MUTE: HERE/"mute_logo.png",
}
ICONS = [it for key, it in LOGOS.items()]
KEYS = [key for key, it in LOGOS.items()]
ping_pong_index = 0
@interactive(
player=Control(PREDICTED, KEYS, icons=ICONS, name="Player selection"))
def audio_selector(sig, mixed, pred, global_params={}, player=MUTE):
global_params["selected_audio"] = player if player != MUTE else global_params.get("selected_audio", MIXED)
global_params[MUTE] = player == MUTE
if player == CLEAN:
audio_track = sig["buffers"][CLEAN]
elif player == NOISY:
audio_track = sig["buffers"][NOISY]
elif player == MIXED:
audio_track = mixed
elif player == PREDICTED:
audio_track = pred
else:
audio_track = mixed
return audio_track
# @interactive(
# loop=KeyboardControl(True, keydown="l"))
def audio_trim(audio_track, global_params={}, loop=True):
sampling_rate = global_params.get(SAMPLING_RATE, 8000)
if global_params.get("trim", False):
start, end = global_params["trim"]["start"], global_params["trim"]["end"]
remainder = (end-start) % 8
audio_trim = audio_track[..., start:end-remainder]
repeat_factor = int(sampling_rate*4./(end-start))
logging.debug(f"{repeat_factor}")
repeat_factor = max(1, repeat_factor)
if not loop:
repeat_factor = 1
audio_trim = audio_trim.repeat(1, repeat_factor)
logging.debug(f"{audio_trim.shape}")
else:
audio_trim = audio_track
return audio_trim
# @interactive(
# volume=(1000, [0, 1000], "volume"),
# )
def audio_player(audio_trim, global_params={}, volume=1000):
sampling_rate = global_params.get(SAMPLING_RATE, 8000)
try:
if global_params.get(MUTE, True):
global_params["__stop"]()
print("mute!")
else:
ping_pong_path = root_dir/"__ping_pong"
ping_pong_path.mkdir(exist_ok=True)
global ping_pong_index
audio_track_path = ping_pong_path/f"_tmp_{ping_pong_index}.wav"
ping_pong_index = (ping_pong_index + 1) % 10
save_audio_tensor(audio_track_path, volume/100.*audio_trim,
sampling_rate=global_params.get(SAMPLING_RATE, sampling_rate))
global_params["__set_audio"](audio_track_path)
global_params["__play"]()
except Exception as exc:
logging.warning(f"Exception in audio_player {exc}")
pass
|