videosdk-plugins-rnnoise 0.0.31__py3-none-any.whl → 0.0.32__py3-none-any.whl

videosdk/plugins/rnnoise/denoise.py

@@ -4,12 +4,15 @@ from .rnnoise import RNN
 import numpy as np
 import resampy
 
+
 class RNNoise(Denoise):
     def __init__(self):
+        """Initialize the RNNoise denoise plugin.
+        """
         super().__init__()
         self.rnnoise = RNN()
         self._target_sample_rate = 48000
-        self._frame_duration_ms = 20 
+        self._frame_duration_ms = 20
         self._rnnoise_frame_size = 480
 
     async def denoise(self, audio_frames: bytes, **kwargs: Any) -> bytes:
@@ -18,29 +21,35 @@ class RNNoise(Denoise):
 
         audio_np = np.frombuffer(audio_frames, dtype=np.int16)
         num_samples = len(audio_np)
-        original_sample_rate = int(num_samples * 1000 / self._frame_duration_ms)
+        original_sample_rate = int(
+            num_samples * 1000 / self._frame_duration_ms)
 
         if original_sample_rate != self._target_sample_rate:
             audio_float = audio_np.astype(np.float32) / 32767.0
-            resampled_audio_float = resampy.resample(audio_float, sr_orig=original_sample_rate, sr_new=self._target_sample_rate)
-            resampled_audio_np = (resampled_audio_float * 32767.0).astype(np.int16)
+            resampled_audio_float = resampy.resample(
+                audio_float, sr_orig=original_sample_rate, sr_new=self._target_sample_rate)
+            resampled_audio_np = (resampled_audio_float *
+                                  32767.0).astype(np.int16)
         else:
             resampled_audio_np = audio_np
 
-        num_rnnoise_frames = len(resampled_audio_np) // self._rnnoise_frame_size
+        num_rnnoise_frames = len(
+            resampled_audio_np) // self._rnnoise_frame_size
         denoised_chunks = []
 
         for i in range(num_rnnoise_frames):
             start = i * self._rnnoise_frame_size
             end = start + self._rnnoise_frame_size
             chunk = resampled_audio_np[start:end]
-            
+
             if len(chunk) != self._rnnoise_frame_size:
                 continue
 
             chunk_bytes = chunk.tobytes()
-            _vod_prob, denoised_chunk_bytes = self.rnnoise.process_frame(chunk_bytes)
-            denoised_chunk_np = np.frombuffer(denoised_chunk_bytes, dtype=np.int16)
+            _vod_prob, denoised_chunk_bytes = self.rnnoise.process_frame(
+                chunk_bytes)
+            denoised_chunk_np = np.frombuffer(
+                denoised_chunk_bytes, dtype=np.int16)
             denoised_chunks.append(denoised_chunk_np)
 
         if not denoised_chunks:
@@ -50,7 +59,8 @@ class RNNoise(Denoise):
 
         if original_sample_rate != self._target_sample_rate:
             denoised_float = denoised_audio_np.astype(np.float32) / 32767.0
-            original_format_float = resampy.resample(denoised_float, sr_orig=self._target_sample_rate, sr_new=original_sample_rate)
+            original_format_float = resampy.resample(
+                denoised_float, sr_orig=self._target_sample_rate, sr_new=original_sample_rate)
             final_audio_np = (original_format_float * 32767.0).astype(np.int16)
         else:
             final_audio_np = denoised_audio_np
@@ -59,4 +69,4 @@ class RNNoise(Denoise):
 
     async def aclose(self) -> None:
         self.rnnoise.destroy()
-        await super().aclose()
+        await super().aclose()

videosdk/plugins/rnnoise/version.py

@@ -1 +1 @@
-__version__ = "0.0.31"
+__version__ = "0.0.32"

{videosdk_plugins_rnnoise-0.0.31.dist-info → videosdk_plugins_rnnoise-0.0.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: videosdk-plugins-rnnoise
-Version: 0.0.31
+Version: 0.0.32
 Summary: VideoSDK Agent Framework plugin for RNNoise.
 Author: videosdk
 License-Expression: Apache-2.0
@@ -14,7 +14,7 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.11
 Requires-Dist: numpy>=1.27
 Requires-Dist: resampy
-Requires-Dist: videosdk-agents>=0.0.31
+Requires-Dist: videosdk-agents>=0.0.32
 Description-Content-Type: text/markdown
 
 # VideoSDK RNNoise Plugin

{videosdk_plugins_rnnoise-0.0.31.dist-info → videosdk_plugins_rnnoise-0.0.32.dist-info}/RECORD

@@ -1,10 +1,10 @@
 videosdk/plugins/rnnoise/__init__.py,sha256=W1ioncb06WtjaxKXUuhQMsr6JHZmNTPeUOXqnRkQiuw,51
 videosdk/plugins/rnnoise/build_rnnoise.py,sha256=2vJDXgS3rl5w98e6XmxhMrSzvBgoYcAf03f4G_iZD98,3480
-videosdk/plugins/rnnoise/denoise.py,sha256=t0Y8SjdRE2iVlrP3gfwmD_ArXwcBLBCi1V9-JryBJFI,2385
+videosdk/plugins/rnnoise/denoise.py,sha256=Ait9K-rY4GLNRz2uz71O8WQaqqvBjSLVuqoOiRdbjjk,2564
 videosdk/plugins/rnnoise/rnnoise.py,sha256=HJ5qx8ZFfM0TW_zeioOPoAOqDKcdB0BxCgh0MzS214Y,1359
-videosdk/plugins/rnnoise/version.py,sha256=YRrMDApG1V6fNZdI4BA631DmsqtamSXgI1yu0tnt4h0,23
+videosdk/plugins/rnnoise/version.py,sha256=WKDnjJM7gYpD9fIwhK2qAZICJAT2ndquQ6VcOar074Y,23
 videosdk/plugins/rnnoise/files/librnnoise.0.dylib,sha256=4nvKWEolh1ZfHeIeteAv0PA3bJEf8OH_dKAQNbsRtQ8,3659392
 videosdk/plugins/rnnoise/files/librnnoise.dylib,sha256=4nvKWEolh1ZfHeIeteAv0PA3bJEf8OH_dKAQNbsRtQ8,3659392
-videosdk_plugins_rnnoise-0.0.31.dist-info/METADATA,sha256=9TxtNmxqBu1kw3hDtjS50RVAby7Cznvl-kK5TkGRY1E,1192
-videosdk_plugins_rnnoise-0.0.31.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-videosdk_plugins_rnnoise-0.0.31.dist-info/RECORD,,
+videosdk_plugins_rnnoise-0.0.32.dist-info/METADATA,sha256=TgvwbPhayKmX_zW6T-yGT9m2Ygqh1956PRn1a68b_OU,1192
+videosdk_plugins_rnnoise-0.0.32.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+videosdk_plugins_rnnoise-0.0.32.dist-info/RECORD,,