gitea/conjurer
1
0
Fork 0
mirror of https://github.com/migatu/conjurer.git synced 2026-07-18 07:42:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

ff

Browse Source
This commit is contained in:
mtuszowski
2024-09-19 19:31:01 +02:00
parent 520d995553
commit 28db612fad
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
voice_recognition.py
+8 -8
View File
@@ -46,7 +46,7 @@ class WaveWriter:
self.present_file_id = 0 self.present_file_id = 0
self.file_name_past = None self.file_name_past = None
self.file_name_present = location + self.username + "_" + str(self.present_file_id) self.file_name_present = location + self.username + "_" + str(self.present_file_id) + ".mp3"
self.transcript_file_present : wave.Wave_write = wave.open( self.transcript_file_present : wave.Wave_write = wave.open(
self.file_name_present, "wb" self.file_name_present, "wb"
) )
@@ -54,7 +54,7 @@ class WaveWriter:
self.transcript_file_present.setsampwidth(SAMPLE_WIDTH) self.transcript_file_present.setsampwidth(SAMPLE_WIDTH)
self.transcript_file_present.setframerate(SAMPLING_RATE) self.transcript_file_present.setframerate(SAMPLING_RATE)
self.file_name_future = location + self.username + "_" + str(self.present_file_id + 1) self.file_name_future = location + self.username + "_" + str(self.present_file_id + 1) + ".mp3"
self.transcript_file_future : wave.Wave_write = wave.open( self.transcript_file_future : wave.Wave_write = wave.open(
self.file_name_future, "wb" self.file_name_future, "wb"
) )
@@ -69,7 +69,7 @@ class WaveWriter:
self.present_file_id += 1 self.present_file_id += 1
self.file_name_present = self.file_name_future self.file_name_present = self.file_name_future
self.transcript_file_present = self.transcript_file_future self.transcript_file_present = self.transcript_file_future
self.file_name_future = location + self.username + "_" + str(self.present_file_id + 1) self.file_name_future = location + self.username + "_" + str(self.present_file_id + 1) + ".mp3"
self.transcript_file_future : wave.Wave_write = wave.open( self.transcript_file_future : wave.Wave_write = wave.open(
self.file_name_future, "wb" self.file_name_future, "wb"
) )
@@ -95,7 +95,7 @@ class SRBuffer(voice_recv.AudioSink):
""" """
# on member join dodajemytypa do listy # on member join dodajemytypa do listy
# on member disconnect - dropujemy go # on member disconnect - dropujemy go
def __init__(self, destination): def __init__(self):
super().__init__() super().__init__()
self.wavewriter = { self.wavewriter = {
@@ -103,11 +103,11 @@ class SRBuffer(voice_recv.AudioSink):
} }
def on_user_connect(self, username): def on_user_connect(self, username):
self.wavewriter[username] = WaveWriter(username) self.wavewriter[str(username)] = WaveWriter(username)
def on_user_disconnect(self,username): def on_user_disconnect(self,username):
self.wavewriter[username].cleanup() self.wavewriter[str(username)].cleanup()
self.wavewriter.pop(username) self.wavewriter.pop(str(username))
def rotate_user(self,username): def rotate_user(self,username):
self.wavewriter[str(username)].rotate() self.wavewriter[str(username)].rotate()
@@ -131,7 +131,7 @@ class Transcriber(commands.Cog):
def __init__(self, bot): def __init__(self, bot):
self.bot = bot self.bot = bot
self._last_member = None self._last_member = None
self.wsink = SRBuffer(destination="/home/pi/Conjurer/wav.wav") self.wsink = SRBuffer()
self.threads = [] self.threads = []
self.comm_queue = Queue() self.comm_queue = Queue()