added checkbox whether to add timestamp at the end of the filename

app.py

@@ -46,7 +46,9 @@ class App:
                                               value="Automatic Detection", label="Language")
                         dd_subformat = gr.Dropdown(["SRT", "WebVTT"], value="SRT", label="Subtitle Format")
                     with gr.Row():
-                        cb_translate = gr.Checkbox(value=False, label="Translate to English?", interactive=True)
+                        cb_translate = gr.Checkbox(value=False, label="Translate to English?", interactive=True, scale=1)
+                    with gr.Row():
+                        cb_timestamp = gr.Checkbox(value=True, label="Add a timestamp to the end of the filename", interactive=True, scale=1)
                     with gr.Row():
                         btn_run = gr.Button("GENERATE SUBTITLE FILE", variant="primary")
                     with gr.Row():
@@ -54,7 +56,7 @@ class App:
                         btn_openfolder = gr.Button('📂', scale=2)
 
                     btn_run.click(fn=self.whisper_inf.transcribe_file,
-                                  inputs=[input_file, dd_model, dd_lang, dd_subformat, cb_translate],
+                                  inputs=[input_file, dd_model, dd_lang, dd_subformat, cb_translate, cb_timestamp],
                                   outputs=[tb_indicator])
                     btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
                     dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
@@ -76,6 +78,9 @@ class App:
                         dd_subformat = gr.Dropdown(choices=["SRT", "WebVTT"], value="SRT", label="Subtitle Format")
                     with gr.Row():
                         cb_translate = gr.Checkbox(value=False, label="Translate to English?", interactive=True)
+                    with gr.Row():
+                        cb_timestamp = gr.Checkbox(value=True, label="Add a timestamp to the end of the filename",
+                                                   interactive=True)
                     with gr.Row():
                         btn_run = gr.Button("GENERATE SUBTITLE FILE", variant="primary")
                     with gr.Row():
@@ -83,7 +88,7 @@ class App:
                         btn_openfolder = gr.Button('📂', scale=2)
 
                     btn_run.click(fn=self.whisper_inf.transcribe_youtube,
-                                  inputs=[tb_youtubelink, dd_model, dd_lang, dd_subformat, cb_translate],
+                                  inputs=[tb_youtubelink, dd_model, dd_lang, dd_subformat, cb_translate, cb_timestamp],
                                   outputs=[tb_indicator])
                     tb_youtubelink.change(get_ytmetas, inputs=[tb_youtubelink],
                                           outputs=[img_thumbnail, tb_title, tb_description])
@@ -126,6 +131,9 @@ class App:
                                                              choices=self.nllb_inf.available_source_langs)
                             dd_nllb_targetlang = gr.Dropdown(label="Target Language",
                                                              choices=self.nllb_inf.available_target_langs)
+                        with gr.Row():
+                            cb_timestamp = gr.Checkbox(value=True, label="Add a timestamp to the end of the filename",
+                                                       interactive=True)
                         with gr.Row():
                             btn_run = gr.Button("TRANSLATE SUBTITLE FILE", variant="primary")
                         with gr.Row():
@@ -135,7 +143,7 @@ class App:
                             md_vram_table = gr.HTML(NLLB_VRAM_TABLE, elem_id="md_nllb_vram_table")
 
                     btn_run.click(fn=self.nllb_inf.translate_file,
-                                  inputs=[file_subs, dd_nllb_model, dd_nllb_sourcelang, dd_nllb_targetlang],
+                                  inputs=[file_subs, dd_nllb_model, dd_nllb_sourcelang, dd_nllb_targetlang, cb_timestamp],
                                   outputs=[tb_indicator])
                     btn_openfolder.click(fn=lambda: self.open_folder(os.path.join("outputs", "translations")),
                                          inputs=None,

modules/nllb_inference.py

@@ -33,6 +33,7 @@ class NLLBInference(BaseInterface):
                        model_size: str,
                        src_lang: str,
                        tgt_lang: str,
+                       add_timestamp: bool,
                        progress=gr.Progress()):
         """
         Translate subtitle file from source language to target language
@@ -47,6 +48,8 @@ class NLLBInference(BaseInterface):
             Source language of the file to translate from gr.Dropdown()
         tgt_lang: str
             Target language of the file to translate from gr.Dropdown()
+        add_timestamp: bool
+            Boolean value from gr.Checkbox() that determines whether to add a timestamp at the end of the filename.
         progress: gr.Progress
             Indicator to show progress directly in gradio.
             I use a forked version of whisper for this. To see more info : https://github.com/jhj0517/jhj0517-whisper/tree/add-progress-callback
@@ -85,8 +88,10 @@ class NLLBInference(BaseInterface):
                     subtitle = get_serialized_srt(parsed_dicts)
 
                     timestamp = datetime.now().strftime("%m%d%H%M%S")
-                    file_name = file_name[:-9]
-                    output_path = os.path.join("outputs", "translations", f"{file_name}-{timestamp}")
+                    if add_timestamp:
+                        output_path = os.path.join("outputs", "translations", f"{file_name}-{timestamp}")
+                    else:
+                        output_path = os.path.join("outputs", "translations", f"{file_name}")
 
                     write_file(subtitle, f"{output_path}.srt")
 
@@ -100,8 +105,10 @@ class NLLBInference(BaseInterface):
                     subtitle = get_serialized_vtt(parsed_dicts)
 
                     timestamp = datetime.now().strftime("%m%d%H%M%S")
-                    file_name = file_name[:-9]
-                    output_path = f"outputs/translations/{file_name}-{timestamp}"
+                    if add_timestamp:
+                        output_path = os.path.join("outputs", "translations", f"{file_name}-{timestamp}")
+                    else:
+                        output_path = os.path.join("outputs", "translations", f"{file_name}")
 
                     write_file(subtitle, f"{output_path}.vtt")
 

modules/whisper_Inference.py

@@ -24,6 +24,7 @@ class WhisperInference(BaseInterface):
                         lang: str,
                         subformat: str,
                         istranslate: bool,
+                        add_timestamp: bool,
                         progress=gr.Progress()):
         """
         Write subtitle file from Files
@@ -41,6 +42,8 @@ class WhisperInference(BaseInterface):
         istranslate: bool
             Boolean value from gr.Checkbox() that determines whether to translate to English.
             It's Whisper's feature to translate speech from another language directly into English end-to-end.
+        add_timestamp: bool
+            Boolean value from gr.Checkbox() that determines whether to add a timestamp at the end of the filename.
         progress: gr.Progress
             Indicator to show progress directly in gradio.
             I use a forked version of whisper for this. To see more info : https://github.com/jhj0517/jhj0517-whisper/tree/add-progress-callback
@@ -75,10 +78,12 @@ class WhisperInference(BaseInterface):
                 progress(1, desc="Completed!")
 
                 file_name, file_ext = os.path.splitext(os.path.basename(fileobj.orig_name))
-                file_name = file_name[:-9]
                 file_name = safe_filename(file_name)
                 timestamp = datetime.now().strftime("%m%d%H%M%S")
-                output_path = os.path.join("outputs", f"{file_name}-{timestamp}")
+                if add_timestamp:
+                    output_path = os.path.join("outputs", f"{file_name}-{timestamp}")
+                else:
+                    output_path = os.path.join("outputs", f"{file_name}")
 
                 if subformat == "SRT":
                     subtitle = get_srt(result["segments"])
@@ -108,6 +113,7 @@ class WhisperInference(BaseInterface):
                            lang: str,
                            subformat: str,
                            istranslate: bool,
+                           add_timestamp: bool,
                            progress=gr.Progress()):
         """
         Write subtitle file from Youtube
@@ -125,6 +131,8 @@ class WhisperInference(BaseInterface):
         istranslate: bool
             Boolean value from gr.Checkbox() that determines whether to translate to English.
             It's Whisper's feature to translate speech from another language directly into English end-to-end.
+        add_timestamp: bool
+            Boolean value from gr.Checkbox() that determines whether to add a timestamp at the end of the filename.
         progress: gr.Progress
             Indicator to show progress directly in gradio.
             I use a forked version of whisper for this. To see more info : https://github.com/jhj0517/jhj0517-whisper/tree/add-progress-callback
@@ -157,7 +165,10 @@ class WhisperInference(BaseInterface):
 
             file_name = safe_filename(yt.title)
             timestamp = datetime.now().strftime("%m%d%H%M%S")
-            output_path = os.path.join("outputs", f"{file_name}-{timestamp}")
+            if add_timestamp:
+                output_path = os.path.join("outputs", f"{file_name}-{timestamp}")
+            else:
+                output_path = os.path.join("outputs", f"{file_name}")
 
             if subformat == "SRT":
                 subtitle = get_srt(result["segments"])