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

Push gladia translations as a TranscriptionFrame

Browse Source
This commit is contained in:
Dan Berg
2025-04-15 16:02:22 +02:00
committed by Mark Backman
parent acb6abd761
commit be8788e4da
1 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/pipecat/services/gladia/stt.py
View File

@@ -384,16 +384,31 @@ class GladiaSTTService(STTService):
if content["type"] == "transcript": if content["type"] == "transcript":
utterance = content["data"]["utterance"] utterance = content["data"]["utterance"]
confidence = utterance.get("confidence", 0) confidence = utterance.get("confidence", 0)
language = utterance["language"]
transcript = utterance["text"] transcript = utterance["text"]
if confidence >= self._confidence: if confidence >= self._confidence:
if content["data"]["is_final"]: if content["data"]["is_final"]:
await self.push_frame( await self.push_frame(
TranscriptionFrame(transcript, "", time_now_iso8601()) TranscriptionFrame(transcript, "", time_now_iso8601(), language)
) )
else: else:
await self.push_frame( await self.push_frame(
InterimTranscriptionFrame(transcript, "", time_now_iso8601()) InterimTranscriptionFrame(
transcript, "", time_now_iso8601(), language
)
) )
elif content["type"] == "translation":
translated_utterance = content["data"]["translated_utterance"]
original_language = content["data"]["original_language"]
translated_language = translated_utterance["language"]
confidence = translated_utterance.get("confidence", 0)
translation = translated_utterance["text"]
if translated_language != original_language and confidence >= self._confidence:
await self.push_frame(
TranscriptionFrame(
translation, "", time_now_iso8601(), translated_language
)
)
except websockets.exceptions.ConnectionClosed: except websockets.exceptions.ConnectionClosed:
# Expected when closing the connection # Expected when closing the connection
pass pass