Created
September 7, 2024 00:30
-
-
Save MattiooFR/acff7bd103bd4adffa23e5afb8943201 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, NoTranscriptFound | |
| def get_video_id(url): | |
| """ | |
| Extract the video ID from a YouTube URL | |
| """ | |
| if 'youtube.com/watch?v=' in url: | |
| return url.split('v=')[1] | |
| elif 'youtu.be/' in url: | |
| return url.split('be/')[1] | |
| else: | |
| raise ValueError("Invalid YouTube URL") | |
| def fetch_available_languages(video_id): | |
| """ | |
| Fetch available languages for the YouTube video transcript. | |
| """ | |
| try: | |
| transcript_list = YouTubeTranscriptApi.list_transcripts(video_id) | |
| languages = [] | |
| # Iterate over the transcript list to get the languages | |
| for transcript in transcript_list: | |
| languages.append((transcript.language, transcript.language_code)) | |
| return languages | |
| except TranscriptsDisabled: | |
| print("Transcripts are disabled for this video.") | |
| return [] | |
| except NoTranscriptFound: | |
| print("No transcripts found for this video.") | |
| return [] | |
| def choose_language(languages): | |
| """ | |
| Display available languages and allow the user to choose one. | |
| """ | |
| print("\nAvailable languages:") | |
| for idx, (language, code) in enumerate(languages): | |
| print(f"{idx + 1}. {language} ({code})") | |
| choice = int(input("\nEnter the number of the language you want to select: ")) - 1 | |
| if choice < 0 or choice >= len(languages): | |
| print("Invalid choice. Please run the program again.") | |
| exit() | |
| return languages[choice][1] | |
| def download_transcript(video_id, language_code, output_file): | |
| """ | |
| Download the YouTube transcript in the selected language and save it to a .txt file. | |
| """ | |
| try: | |
| transcript = YouTubeTranscriptApi.get_transcript(video_id, languages=[language_code]) | |
| with open(output_file, 'w', encoding='utf-8') as f: | |
| for entry in transcript: | |
| timestamp = format_time(entry['start']) | |
| f.write(f"{timestamp} - {entry['text']}\n") | |
| print(f"Transcript saved to {output_file}") | |
| except Exception as e: | |
| print(f"Error fetching transcript: {e}") | |
| def format_time(seconds): | |
| """ | |
| Format timestamp from seconds to HH:MM:SS. | |
| """ | |
| mins, secs = divmod(int(seconds), 60) | |
| hrs, mins = divmod(mins, 60) | |
| return f"{hrs:02}:{mins:02}:{secs:02}" | |
| def preview_transcript(output_file, lines=5): | |
| """ | |
| Display the first 'n' lines of the transcript. | |
| """ | |
| try: | |
| with open(output_file, 'r', encoding='utf-8') as f: | |
| transcript_lines = f.readlines() | |
| print("\nPreview of the transcript:") | |
| for line in transcript_lines[:lines]: | |
| print(line.strip()) | |
| except Exception as e: | |
| print(f"Error reading the transcript file: {e}") | |
| if __name__ == "__main__": | |
| url = input("Enter the YouTube video URL: ") | |
| video_id = get_video_id(url) | |
| # Fetch available languages | |
| languages = fetch_available_languages(video_id) | |
| if languages: | |
| # Ask user to select a language | |
| language_code = choose_language(languages) | |
| # Download transcript and save to file | |
| output_file = f"youtube_transcript_{video_id}_{language_code}.txt" | |
| download_transcript(video_id, language_code, output_file) | |
| # Preview the first 5 sentences of the transcript | |
| preview_transcript(output_file, 5) | |
| else: | |
| print("No available languages for this video.") |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment