Update generate_transcript.py

generate_transcript.py

@@ -6,6 +6,12 @@ import warnings
 from groq import Groq
 import os
 
+
+import re
+
+
+
+
 warnings.filterwarnings('ignore')
 
 
@@ -152,7 +158,13 @@ class TranscriptProcessor:
             pickle.dump(transcript, f)
         
         return self.transcript_output_path
-
+        
+    def extract_tuple(self,text):
+        match = re.search(r'\[.*\]', text)
+        if match:
+            return match.group(0)
+        return None
+        
     def rewrite_transcript(self):
         """
         Refines the transcript for TTS, adding expressive elements and saving as a list of tuples.
@@ -178,7 +190,7 @@ class TranscriptProcessor:
             model=self.model_name,
         )
         
-        rewritten_transcript = chat_completion.choices[0].message.content
+        rewritten_transcript =self.extract_tuple(chat_completion.choices[0].message.content)
         
         # Save the rewritten transcript as a pickle file
         with open(self.tts_output_path, 'wb') as f: