Update modules/utils/subtitle_manager.py

modules/utils/subtitle_manager.py

@@ -44,18 +44,15 @@ def get_srt(segments):
 def get_csv(segments):
     bDiarization = False    
     output = ""
-            
-    for i, segment in enumerate(segments):
-        if segment['text'].startswith(' '):
-            segment['text'] = segment['text'][1:]
 
-        # Check if speakers are identified and get speaker id & text
-        temp = re.search(r'_[0-9][0-9]: ',segment['text'])  
-        
-        if temp != None:
-            #temp_string = str(temp.group())
-            #speaker_id = temp_string.replace(': ','')
-            #speaker_text = (segment['text']).replace(temp_string,'')
+    # Check if speakers are identified
+    for segment in enumerate(segments):
+        if re.search(r'_[0-9][0-9]: ',segment['text']) != None:
+            bDiarization = True
+            break
+
+    for i, segment in enumerate(segments):
+        if bDiarization:
             speaker_id = ((segment['text']).split(":", 1)[0]).strip()
             speaker_text = ((segment['text']).split(":", 1)[1]).strip()
             output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_id};{speaker_text};\n"