jhj0517 commited on
Commit
0036278
·
1 Parent(s): 9cf3b54

Fix format and add `add_timestamp`

Browse files
Files changed (1) hide show
  1. modules/utils/subtitle_manager.py +8 -2
modules/utils/subtitle_manager.py CHANGED
@@ -6,6 +6,7 @@ import re
6
  import sys
7
  import zlib
8
  from typing import Callable, List, Optional, TextIO, Union, Dict
 
9
 
10
  from modules.whisper.data_classes import Segment
11
 
@@ -52,10 +53,15 @@ class ResultWriter:
52
  self.output_dir = output_dir
53
 
54
  def __call__(
55
- self, result: Union[dict, List[Segment]], output_file_name: str, options: Optional[dict] = None, **kwargs
 
56
  ):
57
  if isinstance(result, List) and result and isinstance(result[0], Segment):
58
- result = [seg.dict() for seg in result]
 
 
 
 
59
 
60
  output_path = os.path.join(
61
  self.output_dir, output_file_name + "." + self.extension
 
6
  import sys
7
  import zlib
8
  from typing import Callable, List, Optional, TextIO, Union, Dict
9
+ from datetime import datetime
10
 
11
  from modules.whisper.data_classes import Segment
12
 
 
53
  self.output_dir = output_dir
54
 
55
  def __call__(
56
+ self, result: Union[dict, List[Segment]], output_file_name: str, add_timestamp: bool = True,
57
+ options: Optional[dict] = None, **kwargs
58
  ):
59
  if isinstance(result, List) and result and isinstance(result[0], Segment):
60
+ result = {"segments": [seg.dict() for seg in result]}
61
+
62
+ if add_timestamp:
63
+ timestamp = datetime.now().strftime("%m%d%H%M%S")
64
+ output_file_name += timestamp
65
 
66
  output_path = os.path.join(
67
  self.output_dir, output_file_name + "." + self.extension