Update app.py

app.py

@@ -319,6 +319,34 @@ class App:
                     with gr.Column():
                         gr.Markdown(device_info, label="Hardware info & installed packages")
 
+                with gr.TabItem("Translation"):  # tab3
+                    with gr.Row():
+                        file_subs = gr.Files(type="filepath", label="Upload Subtitle Files to translate here",file_types=['.vtt', '.srt'])
+
+                    with gr.Row():
+                        dd_model_size = gr.Dropdown(label="Model", value=nllb_params["model_size"],choices=self.nllb_inf.available_models)
+                        dd_source_lang = gr.Dropdown(label="Source Language", value=nllb_params["source_lang"],choices=self.nllb_inf.available_source_langs)
+                        dd_target_lang = gr.Dropdown(label="Target Language", value=nllb_params["target_lang"],choices=self.nllb_inf.available_target_langs)
+                    with gr.Row():
+                        nb_max_length = gr.Number(label="Max Length Per Line", value=nllb_params["max_length"],precision=0)
+                    with gr.Row():
+                        cb_timestamp = gr.Checkbox(value=translation_params["add_timestamp"], 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():
+                        tb_indicator = gr.Textbox(label="Output", scale=5)
+                        files_subtitles = gr.Files(label="Downloadable output file", scale=3)
+                        btn_openfolder = gr.Button('📂', scale=1)
+
+                    btn_run.click(fn=self.nllb_inf.translate_file,
+                                  inputs=[file_subs, dd_model_size, dd_source_lang, dd_target_lang,
+                                          nb_max_length, cb_timestamp],
+                                  outputs=[tb_indicator, files_subtitles])
+
+                    btn_openfolder.click(fn=lambda: self.open_folder(os.path.join(self.args.output_dir, "translations")),
+                                         inputs=None,
+                                         outputs=None)
+        
         # Launch the app with optional gradio settings
         args = self.args