Update app.py

app.py

@@ -104,30 +104,24 @@ def add_time_to_subtitle(input_file, hours, minutes, seconds):
         
         subs = pysrt.open(input_file)
         
-        # Chuyển đổi thời gian nhập vào thành mili giây
-        total_milliseconds = (int(hours) * 3600 + int(minutes) * 60 + int(seconds)) * 1000
+        # Chuyển đổi thời gian nhập vào thành mili giây (hỗ trợ số thập phân)
+        try:
+            seconds_float = float(seconds)
+        except ValueError:
+            seconds_float = 0
+            
+        total_milliseconds = int((int(hours) * 3600 + int(minutes) * 60 + seconds_float) * 1000)
         
-        # Thêm thời gian vào tất cả các phụ đề nếu có thời gian được chỉ định
+        # Thêm thời gian vào tất cả các phụ đề
         if total_milliseconds > 0:
             for sub in subs:
-                # Thêm thời gian vào start
-                start_ms = sub.start.ordinal + total_milliseconds
-                sub.start.hours = start_ms // 3600000
-                sub.start.minutes = (start_ms % 3600000) // 60000
-                sub.start.seconds = (start_ms % 60000) // 1000
-                sub.start.milliseconds = start_ms % 1000
-                
-                # Thêm thời gian vào end
-                end_ms = sub.end.ordinal + total_milliseconds
-                sub.end.hours = end_ms // 3600000
-                sub.end.minutes = (end_ms % 3600000) // 60000
-                sub.end.seconds = (end_ms % 60000) // 1000
-                sub.end.milliseconds = end_ms % 1000
+                sub.start.ordinal += total_milliseconds
+                sub.end.ordinal += total_milliseconds
         
         # Lưu file tạm
         output_path = tempfile.NamedTemporaryFile(suffix=".srt", delete=False).name
         subs.save(output_path, encoding='utf-8')
-        return output_path, f"Đã thêm {hours}h {minutes}m {seconds}s vào file gốc"
+        return output_path, f"Đã thêm {hours}h {minutes}m {seconds_float}s vào file gốc"
     
     except Exception as e:
         raise gr.Error(f"Có lỗi xảy ra khi thêm thời gian: {str(e)}")
@@ -147,27 +141,21 @@ def translate_subtitle(input_file, source_language, target_language, hours, minu
         
         subs = pysrt.open(input_file)
         
-        # Chuyển đổi thời gian nhập vào thành mili giây
-        total_milliseconds = (int(hours) * 3600 + int(minutes) * 60 + int(seconds)) * 1000
+        # Chuyển đổi thời gian nhập vào thành mili giây (hỗ trợ số thập phân)
+        try:
+            seconds_float = float(seconds)
+        except ValueError:
+            seconds_float = 0
+            
+        total_milliseconds = int((int(hours) * 3600 + int(minutes) * 60 + seconds_float) * 1000)
         
-        # Thêm thời gian vào tất cả các phụ đề nếu có thời gian được chỉ định
+        # Thêm thời gian vào tất cả các phụ đề
         if total_milliseconds > 0:
             for sub in subs:
-                # Thêm thời gian vào start
-                start_ms = sub.start.ordinal + total_milliseconds
-                sub.start.hours = start_ms // 3600000
-                sub.start.minutes = (start_ms % 3600000) // 60000
-                sub.start.seconds = (start_ms % 60000) // 1000
-                sub.start.milliseconds = start_ms % 1000
-                
-                # Thêm thời gian vào end
-                end_ms = sub.end.ordinal + total_milliseconds
-                sub.end.hours = end_ms // 3600000
-                sub.end.minutes = (end_ms % 3600000) // 60000
-                sub.end.seconds = (end_ms % 60000) // 1000
-                sub.end.milliseconds = end_ms % 1000
+                sub.start.ordinal += total_milliseconds
+                sub.end.ordinal += total_milliseconds
         
-        # Xử lý khác nhau nếu là model đơn hay model kép (dịch qua tiếng Anh)
+        # Xử lý dịch thuật
         if isinstance(model_info[0], tuple):
             # Dịch qua tiếng Anh
             model1, tokenizer1 = model_info[0]
@@ -175,9 +163,7 @@ def translate_subtitle(input_file, source_language, target_language, hours, minu
             
             for sub in tqdm(subs, desc="Đang dịch"):
                 if sub.text.strip():
-                    # Dịch sang tiếng Anh trước
                     en_text = translate_text(sub.text, model1, tokenizer1)
-                    # Dịch từ tiếng Anh sang ngôn ngữ đích
                     sub.text = translate_text(en_text, model2, tokenizer2)
         else:
             # Dịch trực tiếp
@@ -189,7 +175,7 @@ def translate_subtitle(input_file, source_language, target_language, hours, minu
         # Lưu file tạm
         output_path = tempfile.NamedTemporaryFile(suffix=".srt", delete=False).name
         subs.save(output_path, encoding='utf-8')
-        return output_path, f"Dịch từ {source_language} sang {target_language} thành công! Đã thêm {hours}h {minutes}m {seconds}s"
+        return output_path, f"Dịch từ {source_language} sang {target_language} thành công! Đã thêm {hours}h {minutes}m {seconds_float}s"
     
     except Exception as e:
         raise gr.Error(f"Có lỗi xảy ra: {str(e)}")
@@ -221,7 +207,7 @@ with gr.Blocks(title="Subtitle Translator Pro", theme="soft") as demo:
             with gr.Row():
                 hours = gr.Number(label="Giờ", value=0, precision=0, minimum=0)
                 minutes = gr.Number(label="Phút", value=0, precision=0, minimum=0, maximum=59)
-                seconds = gr.Number(label="Giây", value=0, precision=0, minimum=0, maximum=59)
+                seconds = gr.Number(label="Giây", value=0, minimum=0, step=0.1)
             
             with gr.Row():
                 add_time_btn = gr.Button("Chỉ thêm thời gian", variant="secondary")
@@ -268,7 +254,7 @@ with gr.Blocks(title="Subtitle Translator Pro", theme="soft") as demo:
     - Tự động phát hiện ngôn ngữ nguồn
     - Dịch giữa 24 ngôn ngữ khác nhau
     - Hỗ trợ dịch qua tiếng Anh nếu không có model trực tiếp
-    - Thêm thời gian vào tất cả phụ đề (giờ, phút, giây)
+    - Thêm thời gian vào tất cả phụ đề (hỗ trợ giây thập phân)
     - Có nút riêng để chỉ thêm thời gian trước khi dịch
     """)