diff --git a/voice_recognition.py b/voice_recognition.py index 8c0b5fd..8986b1e 100644 --- a/voice_recognition.py +++ b/voice_recognition.py @@ -46,7 +46,7 @@ class WaveWriter: self.present_file_id = 0 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.file_name_present, "wb" ) @@ -54,7 +54,7 @@ class WaveWriter: self.transcript_file_present.setsampwidth(SAMPLE_WIDTH) 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.file_name_future, "wb" ) @@ -69,7 +69,7 @@ class WaveWriter: self.present_file_id += 1 self.file_name_present = self.file_name_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.file_name_future, "wb" ) @@ -95,7 +95,7 @@ class SRBuffer(voice_recv.AudioSink): """ # on member join dodajemytypa do listy # on member disconnect - dropujemy go - def __init__(self, destination): + def __init__(self): super().__init__() self.wavewriter = { @@ -103,11 +103,11 @@ class SRBuffer(voice_recv.AudioSink): } def on_user_connect(self, username): - self.wavewriter[username] = WaveWriter(username) + self.wavewriter[str(username)] = WaveWriter(username) def on_user_disconnect(self,username): - self.wavewriter[username].cleanup() - self.wavewriter.pop(username) + self.wavewriter[str(username)].cleanup() + self.wavewriter.pop(str(username)) def rotate_user(self,username): self.wavewriter[str(username)].rotate() @@ -131,7 +131,7 @@ class Transcriber(commands.Cog): def __init__(self, bot): self.bot = bot self._last_member = None - self.wsink = SRBuffer(destination="/home/pi/Conjurer/wav.wav") + self.wsink = SRBuffer() self.threads = [] self.comm_queue = Queue()