Archives/fabric
2
0
Fork 0
mirror of https://github.com/danielmiessler/fabric synced 2024-11-08 07:11:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #304 from bpmcircuits/main

Browse Source 
Language choice option when pulling a transcript from yt
This commit is contained in:
Daniel Miessler 2024-04-01 20:45:48 -07:00 committed by GitHub
commit 24f44b41f2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
installer/client/cli/yt.py
View File

@ -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: