From 163d4a6e4ad6397d6717a378a9d8addb14f001e9 Mon Sep 17 00:00:00 2001
From: Sergey Mankovsky <sergey@monadical.com>
Date: Mon, 20 Jan 2025 12:46:20 +0100
Subject: [PATCH] Refactor transcribe segment

---
 .../modal_deployments/reflector_transcriber.py    | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/server/gpu/modal_deployments/reflector_transcriber.py b/server/gpu/modal_deployments/reflector_transcriber.py
index bde22339..d95bda52 100644
--- a/server/gpu/modal_deployments/reflector_transcriber.py
+++ b/server/gpu/modal_deployments/reflector_transcriber.py
@@ -95,14 +95,13 @@ class Transcriber:
                     vad_parameters={"min_silence_duration_ms": 500},
                 )
 
-            text = ""
-            words = []
-            for segment in segments:
-                text += segment.text
-                words.extend(
-                    {"word": word.word, "start": word.start, "end": word.end}
-                    for word in segment.words
-                )
+            segments = list(segments)
+            text = "".join(segment.text for segment in segments)
+            words = [
+                {"word": word.word, "start": word.start, "end": word.end}
+                for segment in segments
+                for word in segment.words
+            ]
 
             return {"text": text, "words": words}