gitea/conjurer
1
0
Fork 0
mirror of https://github.com/migatu/conjurer.git synced 2026-07-14 21:38:38 +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.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()