Spaces:

LAP-DEV
/

Demo

Running

LAP-DEV commited on Nov 22, 2024

Commit

cef3668

verified ·

1 Parent(s): 02296dc

Update modules/utils/subtitle_manager.py

Files changed (1) hide show

modules/utils/subtitle_manager.py CHANGED Viewed

@@ -44,7 +44,7 @@ def get_csv(segments):
         temp = re.search(r'SPEAKER_[0-9][0-9]: ',segment['text'])
         if temp != None:
             speaker_id = (temp.group()).replace(': ','')
-            speaker_text = test_input.replace(temp.group(),'')
             output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_id};{speaker_text};\n"
             bDiarization = True
         else:

         temp = re.search(r'SPEAKER_[0-9][0-9]: ',segment['text'])
         if temp != None:
             speaker_id = (temp.group()).replace(': ','')
+            speaker_text = temp.replace(temp.group(),'')
             output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_id};{speaker_text};\n"
             bDiarization = True
         else: