From 4d48f299eeab32d57a3fa26aaefb211b6cfed3c9 Mon Sep 17 00:00:00 2001 From: Bartosz Pokrywka Date: Wed, 27 Mar 2024 13:20:11 +0100 Subject: [PATCH] modified: installer/client/cli/yt.py --- installer/client/cli/yt.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/installer/client/cli/yt.py b/installer/client/cli/yt.py index b0449a0..0fff358 100644 --- a/installer/client/cli/yt.py +++ b/installer/client/cli/yt.py @@ -88,11 +88,11 @@ def main_function(url, options): # Get video transcript try: - transcript_list = YouTubeTranscriptApi.get_transcript(video_id) + transcript_list = YouTubeTranscriptApi.get_transcript(video_id, languages=[options.lang]) transcript_text = " ".join([item["text"] for item in transcript_list]) transcript_text = transcript_text.replace("\n", " ") except Exception as e: - transcript_text = f"Transcript not available. ({e})" + transcript_text = f"Transcript not available in the selected language ({options.lang}). ({e})" # Get comments if the flag is set comments = [] @@ -126,7 +126,8 @@ def main(): parser.add_argument('--duration', action='store_true', help='Output only the duration') parser.add_argument('--transcript', action='store_true', help='Output only the transcript') parser.add_argument('--comments', action='store_true', help='Output the comments on the video') - + parser.add_argument('--lang', default='en', help='Language for the transcript (default: English)') + args = parser.parse_args() if args.url is None: