Update modules/utils/subtitle_manager.py
Browse files
modules/utils/subtitle_manager.py
CHANGED
|
@@ -51,6 +51,11 @@ def get_csv(segments):
|
|
| 51 |
|
| 52 |
# Check if speakers are identified and get speaker id & text
|
| 53 |
temp = re.search(r'SPEAKER_[0-9][0-9]: ',segment['text'])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 54 |
if temp != None:
|
| 55 |
temp_string = str(temp.group())
|
| 56 |
speaker_id = temp_string.replace(': ','')
|
|
|
|
| 51 |
|
| 52 |
# Check if speakers are identified and get speaker id & text
|
| 53 |
temp = re.search(r'SPEAKER_[0-9][0-9]: ',segment['text'])
|
| 54 |
+
|
| 55 |
+
# check if speaker 'None' is identified for e.g. music
|
| 56 |
+
if temp == None:
|
| 57 |
+
temp = re.search('None: ',segment['text'])
|
| 58 |
+
|
| 59 |
if temp != None:
|
| 60 |
temp_string = str(temp.group())
|
| 61 |
speaker_id = temp_string.replace(': ','')
|