Update app.py

app.py

@@ -1,7 +1,9 @@
 from youtube_transcript_api import YouTubeTranscriptApi as yta
 import gradio as gr
+from googletrans import Translator
 
-def transcript_generator(link):
+opts = ['afrikaans', 'albanian', 'amharic', 'arabic', 'armenian', 'azerbaijani', 'basque', 'belarusian', 'bengali', 'bosnian', 'bulgarian', 'catalan', 'cebuano', 'chichewa', 'chinese (simplified)', 'chinese (traditional)', 'corsican', 'croatian', 'czech', 'danish', 'dutch', 'english', 'esperanto', 'estonian', 'filipino', 'finnish', 'french', 'frisian', 'galician', 'georgian', 'german', 'greek', 'gujarati', 'haitian creole', 'hausa', 'hawaiian', 'hebrew', 'hebrew', 'hindi', 'hmong', 'hungarian', 'icelandic', 'igbo', 'indonesian', 'irish', 'italian', 'japanese', 'javanese', 'kannada', 'kazakh', 'khmer', 'korean', 'kurdish (kurmanji)', 'kyrgyz', 'lao', 'latin', 'latvian', 'lithuanian', 'luxembourgish', 'macedonian', 'malagasy', 'malay', 'malayalam', 'maltese', 'maori', 'marathi', 'mongolian', 'myanmar (burmese)', 'nepali', 'norwegian', 'odia', 'pashto', 'persian', 'polish', 'portuguese', 'punjabi', 'romanian', 'russian', 'samoan', 'scots gaelic', 'serbian', 'sesotho', 'shona', 'sindhi', 'sinhala', 'slovak', 'slovenian', 'somali', 'spanish', 'sundanese', 'swahili', 'swedish', 'tajik', 'tamil', 'telugu', 'thai', 'turkish', 'ukrainian', 'urdu', 'uyghur', 'uzbek', 'vietnamese', 'welsh', 'xhosa', 'yiddish', 'yoruba', 'zulu']
+def transcript_generator(link,option):
 
   yt_link = str(link)
   yt_id = yt_link.partition("=")[2]
@@ -14,22 +16,27 @@ def transcript_generator(link):
       if key == 'text':
         transcript += val
 
-  return transcript
+  translator = Translator()
+  translation = translator.translate(transcript, dest = option).text
 
+  return transcript,translation
 
 with gr.Blocks() as demo:
 
 
-    input = gr.Textbox(label="Enter YouTube Link",value='',lines=1)
-    output = gr.Textbox(label="Transcript",lines=10)
+    input1 = gr.Textbox(label="Enter YouTube Link",value='',lines=1)
+    input2 = gr.Dropdown(opts)
+    with gr.Tab("Transcript"):
+      output1 = gr.Textbox(label="Transcript",lines=10)
+    with gr.Tab("Translated Transcript"):
+      output2 = gr.Textbox(label="Translation",lines=10)
     btn = gr.Button(value="Submit")
-    btn.click(transcript_generator, inputs=[input], outputs=[output])
+    btn.click(transcript_generator, inputs=[input1,input2], outputs=[output1,output2])
 
     gr.Examples(
-        [["https://www.youtube.com/watch?v=47dtFZ8CFo8"], ["https://www.youtube.com/watch?v=hT_nvWreIhg"],['https://www.youtube.com/watch?v=JP0PNEA_-X0']],
-        [input],
+        [["https://www.youtube.com/watch?v=47dtFZ8CFo8",'chinese (simplified)'], ["https://www.youtube.com/watch?v=hT_nvWreIhg",'french'],['https://www.youtube.com/watch?v=PMs76lrqiA4','urdu']],
+        [input1,input2],
         transcript_generator,
-        
     )
 
 if __name__ == "__main__":