Update app.py

app.py

@@ -19,12 +19,12 @@ def summarize(result):
       summarized_text.append(out)
     st.write(summarized_text)
 
-def get_transcript(video_id):
+def get_transcript(video_id,language):
     #youtube_video = "https://www.youtube.com/watch?v=A4OmtyaBHFE"
     #video_id = youtube_video.split("=")[1]
     #video_id
     #st.write(YouTubeVideo(video_id))
-    transcript = YouTubeTranscriptApi.get_transcript(video_id)
+    transcript = YouTubeTranscriptApi.get_transcript(video_id,languages=[language], preserve_formatting=True)
     print(transcript[0:5])
     
     result = ""
@@ -36,6 +36,8 @@ def get_transcript(video_id):
 
 def main():
     video_id=st.text_input("enter video id ") 
-    if video_id:
-        get_transcript(video_id)
+    language=st.text_input("Language", value="en") 
+
+    if video_id and language:
+        get_transcript(video_id,language)
 main()