jhj0517 commited on
Commit
28dc833
·
1 Parent(s): 2efa275

Fix time format and translation case

Browse files
Files changed (1) hide show
  1. modules/utils/subtitle_manager.py +11 -4
modules/utils/subtitle_manager.py CHANGED
@@ -14,7 +14,7 @@ from .files_manager import read_file
14
 
15
  def format_timestamp(
16
  seconds: float, always_include_hours: bool = True, decimal_marker: str = ","
17
- ):
18
  assert seconds >= 0, "non-negative timestamp expected"
19
  milliseconds = round(seconds * 1000.0)
20
 
@@ -34,10 +34,17 @@ def format_timestamp(
34
 
35
 
36
  def time_str_to_seconds(time_str: str, decimal_marker: str = ",") -> float:
37
- hours, minutes, rest = time_str.split(":")
 
 
 
 
 
 
 
 
38
  seconds, fractional = rest.split(decimal_marker)
39
 
40
- hours = int(hours)
41
  minutes = int(minutes)
42
  seconds = int(seconds)
43
  fractional_seconds = float("0." + fractional)
@@ -401,7 +408,7 @@ def generate_file(
401
  file_path = os.path.join(output_dir, f"{output_file_name}.{output_format}")
402
  file_writer = get_writer(output_format=output_format, output_dir=output_dir)
403
 
404
- if isinstance(file_writer, WriteLRC) and kwargs["highlight_words"]:
405
  kwargs["highlight_words"], kwargs["align_lrc_words"] = False, True
406
 
407
  file_writer(result=result, output_file_name=output_file_name, **kwargs)
 
14
 
15
  def format_timestamp(
16
  seconds: float, always_include_hours: bool = True, decimal_marker: str = ","
17
+ ) -> str:
18
  assert seconds >= 0, "non-negative timestamp expected"
19
  milliseconds = round(seconds * 1000.0)
20
 
 
34
 
35
 
36
  def time_str_to_seconds(time_str: str, decimal_marker: str = ",") -> float:
37
+ times = time_str.split(":")
38
+
39
+ if len(times) == 3:
40
+ hours, minutes, rest = times
41
+ hours = int(hours)
42
+ else:
43
+ hours = 0
44
+ minutes, rest = times
45
+
46
  seconds, fractional = rest.split(decimal_marker)
47
 
 
48
  minutes = int(minutes)
49
  seconds = int(seconds)
50
  fractional_seconds = float("0." + fractional)
 
408
  file_path = os.path.join(output_dir, f"{output_file_name}.{output_format}")
409
  file_writer = get_writer(output_format=output_format, output_dir=output_dir)
410
 
411
+ if isinstance(file_writer, WriteLRC) and kwargs.get("highlight_words", False):
412
  kwargs["highlight_words"], kwargs["align_lrc_words"] = False, True
413
 
414
  file_writer(result=result, output_file_name=output_file_name, **kwargs)