Update modules/utils/subtitle_manager.py

modules/utils/subtitle_manager.py

@@ -43,19 +43,18 @@ def get_csv(segments):
         # Check if speakers are identified and get speaker id & text
         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"
-            output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{segment['text']};\n"
+            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:
             output += f"{i + 1};{timeformat_srt(segment['start'])};{timeformat_srt(segment['end'])};{segment['text']};\n"
-
+        
         # Add titles to csv file
         if bDiarization:
-            output = "Line;Start time;End time;Speaker;Text;\n" + output
+        output = "Line;Start time;End time;Speaker;Text;\n" + output
         else:    
-            output = "Line;Start time;End time;Text;\n" + output
+        output = "Line;Start time;End time;Text;\n" + output
 
 
     return output