From 2646178a2842dcd3568dfc0200a5d7c9a041aa80 Mon Sep 17 00:00:00 2001
From: mtuszowski <mtuszowski@gmail.com>
Date: Thu, 19 Sep 2024 17:30:43 +0200
Subject: [PATCH] fx

---
 voice_recognition.py | 19 ++++++++-----------
 1 file changed, 8 insertions(+), 11 deletions(-)

diff --git a/voice_recognition.py b/voice_recognition.py
index 2ef7ca7..892b321 100644
--- a/voice_recognition.py
+++ b/voice_recognition.py
@@ -93,6 +93,9 @@ class WaveWriter:
         self.file_name_future.setframerate(SAMPLING_RATE)
         # send present to transcription
 
+    def writeframes(self, pcmdata):
+        self.transcript_file_present.writeframes(pcmdata)
+
     def cleanup(self):
         self.transcript_file_present.close()
         self.transcript_file_future.close()
@@ -103,20 +106,13 @@ class SRBuffer(voice_recv.AudioSink):
     """Endpoint AudioSink that generates a wav file.
     Best used in conjunction with a silence generating sink. (TBD)
     """
-
-
     # on member join dodajemytypa do listy
     # on member disconnect - dropujemy go
-
     def __init__(self, destination):
         super().__init__()
-
-        self._file: wave.Wave_write = wave.open(destination, "wb")
-        self._file.setnchannels(CHANNELS)
-        self._file.setsampwidth(SAMPLE_WIDTH)
-        self._file.setframerate(SAMPLING_RATE)
-        self.user_list = {}
-        self.wavewriter = {}
+        self.wavewriter = {
+            "polishhammer" : WaveWriter("polishhammer")
+        }
 
     def on_user_connect(self, username):
         self.wavewriter[username] = WaveWriter(username)
@@ -130,10 +126,11 @@ class SRBuffer(voice_recv.AudioSink):
     def write(self, user, data) -> None:
         logger.info("writing to file %s", self._file)
         logger.info("user: %s", user)
-        self._file.writeframes(data.pcm)
+        self.wavewriter[user].writeframes(data.pcm)
 
     def cleanup(self) -> None:
         try:
+            logger.info("Cleanup for SRBuffer")
             self._file.close()
         except Exception:
             logger.warning("WaveSink got error closing file on cleanup", exc_info=True)