Spaces:
Build error
Build error
Also output JSON of results.
Browse files
app.py
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
from datetime import datetime
|
|
|
|
| 2 |
import math
|
| 3 |
from typing import Iterator, Union
|
| 4 |
import argparse
|
|
@@ -384,11 +385,13 @@ class WhisperTranscriber:
|
|
| 384 |
print("Max line width " + str(languageMaxLineWidth))
|
| 385 |
vtt = self.__get_subs(result["segments"], "vtt", languageMaxLineWidth, highlight_words=highlight_words)
|
| 386 |
srt = self.__get_subs(result["segments"], "srt", languageMaxLineWidth, highlight_words=highlight_words)
|
|
|
|
| 387 |
|
| 388 |
output_files = []
|
| 389 |
output_files.append(self.__create_file(srt, output_dir, source_name + "-subs.srt"));
|
| 390 |
output_files.append(self.__create_file(vtt, output_dir, source_name + "-subs.vtt"));
|
| 391 |
output_files.append(self.__create_file(text, output_dir, source_name + "-transcript.txt"));
|
|
|
|
| 392 |
|
| 393 |
return output_files, text, vtt
|
| 394 |
|
|
|
|
| 1 |
from datetime import datetime
|
| 2 |
+
import json
|
| 3 |
import math
|
| 4 |
from typing import Iterator, Union
|
| 5 |
import argparse
|
|
|
|
| 385 |
print("Max line width " + str(languageMaxLineWidth))
|
| 386 |
vtt = self.__get_subs(result["segments"], "vtt", languageMaxLineWidth, highlight_words=highlight_words)
|
| 387 |
srt = self.__get_subs(result["segments"], "srt", languageMaxLineWidth, highlight_words=highlight_words)
|
| 388 |
+
json_result = json.dumps(result, indent=4, ensure_ascii=False)
|
| 389 |
|
| 390 |
output_files = []
|
| 391 |
output_files.append(self.__create_file(srt, output_dir, source_name + "-subs.srt"));
|
| 392 |
output_files.append(self.__create_file(vtt, output_dir, source_name + "-subs.vtt"));
|
| 393 |
output_files.append(self.__create_file(text, output_dir, source_name + "-transcript.txt"));
|
| 394 |
+
output_files.append(self.__create_file(json_result, output_dir, source_name + "-result.json"));
|
| 395 |
|
| 396 |
return output_files, text, vtt
|
| 397 |
|