Fix file name and add `add_timestamp` parameter

app.py

@@ -257,6 +257,9 @@ class App:
                                                                   self.deepl_api.available_target_langs.keys()))
                         with gr.Row():
                             cb_deepl_ispro = gr.Checkbox(label="Pro User?", value=False)
+                        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():
@@ -266,7 +269,7 @@ class App:
 
                     btn_run.click(fn=self.deepl_api.translate_deepl,
                                   inputs=[tb_authkey, file_subs, dd_deepl_sourcelang, dd_deepl_targetlang,
-                                          cb_deepl_ispro],
+                                          cb_deepl_ispro, cb_timestamp],
                                   outputs=[tb_indicator, files_subtitles])
 
                     btn_openfolder.click(fn=lambda: self.open_folder(os.path.join("outputs", "translations")),

modules/translation/deepl_api.py

@@ -97,6 +97,7 @@ class DeepLAPI:
                         source_lang: str,
                         target_lang: str,
                         is_pro: bool,
+                        add_timestamp: bool,
                         progress=gr.Progress()) -> list:
         """
         Translate subtitle files using DeepL API
@@ -112,6 +113,8 @@ class DeepLAPI:
             Target language of the file to transcribe from gr.Dropdown()
         is_pro: str
             Boolean value that is about pro user or not from gr.Checkbox().
+        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.
 
@@ -141,10 +144,12 @@ class DeepLAPI:
                     progress(batch_end / len(parsed_dicts), desc="Translating..")
 
                 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(self.output_dir, "", f"{file_name}-{timestamp}.srt")
+                if add_timestamp:
+                    timestamp = datetime.now().strftime("%m%d%H%M%S")
+                    file_name += f"-{timestamp}"
+
+                output_path = os.path.join(self.output_dir, "", f"{file_name}.srt")
                 write_file(subtitle, output_path)
 
             elif file_ext == ".vtt":
@@ -161,11 +166,12 @@ class DeepLAPI:
                     progress(batch_end / len(parsed_dicts), desc="Translating..")
 
                 subtitle = get_serialized_vtt(parsed_dicts)
-                timestamp = datetime.now().strftime("%m%d%H%M%S")
 
-                file_name = file_name[:-9]
-                output_path = os.path.join(self.output_dir, "", f"{file_name}-{timestamp}.vtt")
+                if add_timestamp:
+                    timestamp = datetime.now().strftime("%m%d%H%M%S")
+                    file_name += f"-{timestamp}"
 
+                output_path = os.path.join(self.output_dir, "", f"{file_name}.vtt")
                 write_file(subtitle, output_path)
 
             files_info[file_name] = {"subtitle": subtitle, "path": output_path}