mirror of
https://github.com/HiMeditator/auto-caption.git
synced 2026-02-12 18:44:57 +08:00
23 lines
645 B
Python
23 lines
645 B
Python
from sysaudio.win import LoopbackStream, mergeStreamChannels
|
|
from audio2text.gummy import GummyTranslator
|
|
import sys
|
|
|
|
def main():
|
|
sys.stdout.reconfigure(line_buffering=True)
|
|
loopback = LoopbackStream()
|
|
loopback.openStream()
|
|
|
|
gummy = GummyTranslator(loopback.RATE, "zh", "en")
|
|
gummy.translator.start()
|
|
|
|
for _ in range(0, 400):
|
|
if not loopback.stream: continue
|
|
data = loopback.stream.read(loopback.CHUNK)
|
|
data = mergeStreamChannels(data, loopback.CHANNELS)
|
|
gummy.translator.send_audio_frame(data)
|
|
|
|
gummy.translator.stop()
|
|
loopback.closeStream()
|
|
|
|
if __name__ == "__main__":
|
|
main() |