include translation and fix wrong number of input params

app.py

@@ -12,7 +12,7 @@ class GradioInference():
         self.loaded_model = whisper.load_model(self.current_size)
         self.yt = None
     
-    def download_videos(link):
+    def download_videos(self, link):
         """Specify the yt-dlp parameters
 
         Args:
@@ -69,13 +69,14 @@ class GradioInference():
         transcribe_options = dict(task="transcribe", **options)
         translate_options = dict(task="translate", **options)
         results = self.loaded_model.transcribe(path, **transcribe_options)
+        translation_txt = self.loaded_model.transcribe(path, **translate_options)["text"]
 
         if subs == "None":
-            return results["text"]
+            return results["text"], translation_txt
         elif subs == ".srt":
-            return self.srt(results["segments"])
+            return self.srt(results["segments"]), translation_txt
         elif ".csv" == ".csv":
-            return self.csv(results["segments"])
+            return self.csv(results["segments"]), translation_txt
    
     def srt(self, segments):
         output = ""
@@ -131,9 +132,11 @@ with block:
             title = gr.Label(label="Video Title")
             with gr.Row().style(equal_height=True):
                 img = gr.Image(label="Thumbnail")
-                text = gr.Textbox(label="Transcription", placeholder="Transcription Output", lines=10)
+            with gr.Row().style(equal_height=True):
+                transcript = gr.Textbox(label="Transcription", placeholder="Transcription Output", lines=10)
+                translate = gr.Textbox(label="Translation", placeholder="Translation Output", lines=10)
             with gr.Row().style(equal_height=True): 
                 btn = gr.Button("Transcribe")       
-            btn.click(gio, inputs=[link, lang, sz, wt], outputs=[text])
+            btn.click(gio, inputs=[link, lang, sz, wt], outputs=[transcript, translate])
             link.change(gio.populate_metadata, inputs=[link], outputs=[img, title])
 block.launch()