Spaces:
Running
Running
File size: 2,829 Bytes
dbd33b2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 |
from youtube_transcript_api import YouTubeTranscriptApi
from googleapiclient.discovery import build
from googleapiclient.errors import HttpError
import re
import os
# Replace with your actual API key
API_KEY = os.environ.get('YOUTUBE_API_KEY', 'YOUR_API_KEY_HERE')
youtube = build('youtube', 'v3', developerKey=API_KEY)
def extract_video_id(url):
video_id_match = re.search(r"(?:v=|\/)([0-9A-Za-z_-]{11}).*", url)
if video_id_match:
return video_id_match.group(1)
return None
def get_video_metadata(video_id):
try:
request = youtube.videos().list(
part="snippet,contentDetails,statistics",
id=video_id
)
response = request.execute()
if 'items' in response and len(response['items']) > 0:
video = response['items'][0]
snippet = video['snippet']
return {
'title': snippet['title'],
'author': snippet['channelTitle'],
'upload_date': snippet['publishedAt'],
'view_count': video['statistics']['viewCount'],
'like_count': video['statistics'].get('likeCount', 'N/A'),
'comment_count': video['statistics'].get('commentCount', 'N/A'),
'duration': video['contentDetails']['duration']
}
else:
return None
except HttpError as e:
print(f"An HTTP error {e.resp.status} occurred: {e.content}")
return None
def get_transcript(video_id):
try:
transcript = YouTubeTranscriptApi.get_transcript(video_id)
metadata = get_video_metadata(video_id)
return {
'transcript': transcript,
'metadata': metadata
}
except Exception as e:
print(f"Error extracting transcript for video {video_id}: {str(e)}")
return None
def get_channel_videos(channel_id):
try:
request = youtube.search().list(
part="id,snippet",
channelId=channel_id,
type="video",
maxResults=50 # Adjust as needed
)
response = request.execute()
videos = []
for item in response['items']:
videos.append({
'video_id': item['id']['videoId'],
'title': item['snippet']['title'],
'description': item['snippet']['description'],
'published_at': item['snippet']['publishedAt']
})
return videos
except HttpError as e:
print(f"An HTTP error {e.resp.status} occurred: {e.content}")
return []
def process_videos(video_ids):
transcripts = {}
for video_id in video_ids:
transcript_data = get_transcript(video_id)
if transcript_data:
transcripts[video_id] = transcript_data
return transcripts |