jhj0517 commited on
Commit
320b77a
·
1 Parent(s): d8a7425

Enable writer options

Browse files
Files changed (1) hide show
  1. modules/utils/subtitle_manager.py +3 -2
modules/utils/subtitle_manager.py CHANGED
@@ -339,7 +339,8 @@ def get_writer(
339
 
340
 
341
  def generate_file(
342
- output_format: str, output_dir: str, result: Union[dict, List[Segment]], output_file_name: str, add_timestamp: bool = True,
 
343
  ) -> Tuple[str, str]:
344
  output_format = output_format.strip().lower().replace(".", "")
345
 
@@ -349,7 +350,7 @@ def generate_file(
349
 
350
  file_path = os.path.join(output_dir, f"{output_file_name}.{output_format}")
351
  file_writer = get_writer(output_format=output_format, output_dir=output_dir)
352
- file_writer(result=result, output_file_name=output_file_name)
353
  content = read_file(file_path)
354
  return content, file_path
355
 
 
339
 
340
 
341
  def generate_file(
342
+ output_format: str, output_dir: str, result: Union[dict, List[Segment]], output_file_name: str,
343
+ add_timestamp: bool = True, **kwargs
344
  ) -> Tuple[str, str]:
345
  output_format = output_format.strip().lower().replace(".", "")
346
 
 
350
 
351
  file_path = os.path.join(output_dir, f"{output_file_name}.{output_format}")
352
  file_writer = get_writer(output_format=output_format, output_dir=output_dir)
353
+ file_writer(result=result, output_file_name=output_file_name, **kwargs)
354
  content = read_file(file_path)
355
  return content, file_path
356