time fixes

lang_ui.py

@@ -19,6 +19,7 @@ english_ui = {
     "input_video_warning": "Please submit a video",
     "start_time_warning": "Please provide a correct start time",
     "end_time_warning": "Please provide a correct end time",
+    "time_invalid": "Start time is greater than or equal to end time",
     "lang_radio_warning": "Please choose a language for the video",
     "model_dropdown_warning": "Please choose a model size for the video",
 
@@ -76,6 +77,7 @@ indonesia_ui = {
     "input_video_warning": "Mohon masukkan video",
     "start_time_warning": "Mohon berikan waktu mulai yang sesuai",
     "end_time_warning": "Mohon berikan waktu selesai yang sesuai",
+    "time_invalid": "Waktu mulai lebih besar dari atau sama dengan waktu selesai",
     "lang_radio_warning": "Mohon pilih bahasa yang digunakan dalam video",
     "model_dropdown_warning": "Mohon pilih ukuran model yang digunakan untuk video",
 

set_up.py

@@ -3,6 +3,7 @@ from file_name import *
 from diarization import start_diarization
 from transcribe import start_transcribe
 from video_tool import convert_video_to_audio, add_subtitle_to_video
+from datetime import datetime
 import gradio as gr
 import re
 import os
@@ -35,10 +36,12 @@ def prepare_input(input_file, start_time, end_time, lang, model_size, progress=g
     if validate_time_format(start_time) is False:
         gr.Warning(ui_lang["start_time_warning"])
         return [None, None, [None, None]]
-    if validate_time_format(end_time) is False or end_time is '99:99:99':
+    if validate_time_format(end_time) is False or end_time is '00:00:00':
         gr.Warning(ui_lang["end_time_warning"])
         return [None, None, [None, None]]
-    # Check if start time is lower than end time
+    if (check_if_time_invalid(start_time, end_time)):
+        gr.Warning(ui_lang["time_invalid"])
+        return [None, None, [None, None]]
     if lang is None:
         gr.Warning(ui_lang["lang_radio_warning"])
         return [None, None, [None, None]]
@@ -75,3 +78,15 @@ def prepare_input(input_file, start_time, end_time, lang, model_size, progress=g
 def validate_time_format(input_string):
     pattern = re.compile(r'^\d{2}:\d{2}:\d{2}$')
     return pattern.match(input_string) is not None
+
+
+def check_if_time_invalid(start_time, end_time):
+    start = get_total_seconds(start_time)
+    end = get_total_seconds(end_time)
+    return start >= end
+
+
+def get_total_seconds(time_string):
+    hours, minutes, seconds = map(int, time_string.split(":"))
+    total_seconds = hours * 3600 + minutes * 60 + seconds
+    return total_seconds

ui.py

@@ -24,11 +24,11 @@ url_download_button = gr.Button(
     ui_lang["download_button_value"], size='sm', interactive=True)
 input_video = gr.Video(
     label=ui_lang["input_video_label"], interactive=True)
-input_video_button = gr.Button(
-    ui_lang["input_video_button_value"], interactive=True)
+# input_video_button = gr.Button(
+#     ui_lang["input_video_button_value"], interactive=True)
 start_time = gr.Textbox("00:00:00", max_lines=1, placeholder="00:00:00",
                         label=ui_lang["start_time_label"], interactive=True)
-end_time = gr.Textbox("00:10:00", max_lines=1, placeholder="99:99:99",
+end_time = gr.Textbox("00:00:00", max_lines=1, placeholder="99:99:99",
                       label=ui_lang["end_time_label"], interactive=True)
 lang_radio = gr.Radio(
     ui_lang["lang_radio_choices"], label=ui_lang["lang_radio_label"], info=ui_lang["lang_radio_info"], type='index', interactive=True)