Update modules/utils/subtitle_manager.py

modules/utils/subtitle_manager.py

@@ -50,15 +50,12 @@ def get_csv(segments):
             segment['text'] = segment['text'][1:]
 
         # Check if speakers are identified and get speaker id & text
-        temp = re.search(r'SPEAKER_[0-9][0-9]: ',segment['text'])
-
-        # Check if speaker 'None' is identified for e.g. music    
-        if temp == None:
-            temp = re.search('None: ',segment['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_id = temp_string.replace('_ ',' ')
             speaker_text = (segment['text']).replace(temp_string,'')
             output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{speaker_id};{speaker_text};\n"
             bDiarization = True