wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

utils: move exp_smoothing to utils module

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-05-17 13:02:05 -07:00
parent 537e72a05f
commit f2cefeeedc
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/pipecat/services/ai_services.py
View File

@@ -22,6 +22,7 @@ from pipecat.frames.frames import (
VisionImageRawFrame, VisionImageRawFrame,
) )
from pipecat.processors.frame_processor import FrameDirection, FrameProcessor from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
from pipecat.utils.utils import exp_smoothing
class AIService(FrameProcessor): class AIService(FrameProcessor):
@@ -115,16 +116,13 @@ class STTService(AIService):
ww.setframerate(self._sample_rate) ww.setframerate(self._sample_rate)
return (content, ww) return (content, ww)
def _exp_smoothing(self, value: float, prev_value: float, factor: float) -> float:
return prev_value + factor * (value - prev_value)
def _get_smoothed_volume(self, audio: bytes, prev_rms: float, factor: float) -> float: def _get_smoothed_volume(self, audio: bytes, prev_rms: float, factor: float) -> float:
# https://docs.python.org/3/library/array.html # https://docs.python.org/3/library/array.html
audio_array = array.array('h', audio) audio_array = array.array('h', audio)
squares = [sample**2 for sample in audio_array] squares = [sample**2 for sample in audio_array]
mean = sum(squares) / len(audio_array) mean = sum(squares) / len(audio_array)
rms = math.sqrt(mean) rms = math.sqrt(mean)
return self._exp_smoothing(rms, prev_rms, factor) return exp_smoothing(rms, prev_rms, factor)
async def _append_audio(self, frame: AudioRawFrame): async def _append_audio(self, frame: AudioRawFrame):
# Try to filter out empty background noise # Try to filter out empty background noise

4
src/pipecat/utils/utils.py
View File

@@ -29,3 +29,7 @@ def obj_count(obj) -> int:
else: else:
_COUNTS[name] += 1 _COUNTS[name] += 1
return _COUNTS[name] return _COUNTS[name]
def exp_smoothing(value: float, prev_value: float, factor: float) -> float:
return prev_value + factor * (value - prev_value)