Browse Source

Merge pull request #304 from bpmcircuits/main

Language choice option when pulling a transcript from yt
pull/325/head
Daniel Miessler 8 months ago committed by GitHub
parent
commit
24f44b41f2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 7
      installer/client/cli/yt.py

7
installer/client/cli/yt.py

@ -88,11 +88,11 @@ def main_function(url, options):
# Get video transcript # Get video transcript
try: 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 = " ".join([item["text"] for item in transcript_list])
transcript_text = transcript_text.replace("\n", " ") transcript_text = transcript_text.replace("\n", " ")
except Exception as e: 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 # Get comments if the flag is set
comments = [] comments = []
@ -126,7 +126,8 @@ def main():
parser.add_argument('--duration', action='store_true', help='Output only the duration') 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('--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('--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() args = parser.parse_args()
if args.url is None: if args.url is None:

Loading…
Cancel
Save