Upload tools/get_youtube_viral.py with huggingface_hub

tools/get_youtube_viral.py

@@ -0,0 +1,86 @@
+import os
+import sys
+from googleapiclient.discovery import build
+from dotenv import load_dotenv
+
+load_dotenv()
+
+def get_youtube_viral(query=None, max_results=6, mode="search"):
+    """
+    mode='search' : search by query (niche research)
+    mode='trending': fetch YouTube trending videos (no query, uses chart=mostPopular)
+    """
+    api_key = os.getenv('YOUTUBE_API_KEY')
+    if not api_key:
+        print("Error: YOUTUBE_API_KEY not found in .env")
+        return []
+
+    try:
+        youtube = build('youtube', 'v3', developerKey=api_key)
+
+        if mode == "trending":
+            # Fetch actual YouTube trending
+            request = youtube.videos().list(
+                part='snippet,statistics',
+                chart='mostPopular',
+                regionCode='US',
+                maxResults=max_results
+            )
+            response = request.execute()
+            results = []
+            for item in response.get('items', []):
+                vid_id = item['id']
+                stats = item.get('statistics', {})
+                results.append({
+                    'title': item['snippet']['title'],
+                    'channel': item['snippet']['channelTitle'],
+                    'category': item['snippet'].get('categoryId', ''),
+                    'url': f"https://www.youtube.com/watch?v={vid_id}",
+                    'published_at': item['snippet']['publishedAt'],
+                    'views': stats.get('viewCount', '0'),
+                    'likes': stats.get('likeCount', '0'),
+                    'video_id': vid_id,
+                })
+            return results
+
+        else:
+            # Niche search
+            request = youtube.search().list(
+                q=query,
+                part='snippet',
+                type='video',
+                order='relevance',
+                maxResults=max_results,
+                publishedAfter='2026-03-01T00:00:00Z'
+            )
+            response = request.execute()
+            results = []
+            for item in response.get('items', []):
+                video_id = item['id']['videoId']
+                results.append({
+                    'title': item['snippet']['title'],
+                    'channel': item['snippet']['channelTitle'],
+                    'url': f"https://www.youtube.com/watch?v={video_id}",
+                    'published_at': item['snippet']['publishedAt'],
+                    'video_id': video_id,
+                })
+            return results
+
+    except Exception as e:
+        print(f"Error: {e}")
+        return []
+
+
+if __name__ == "__main__":
+    mode = sys.argv[1] if len(sys.argv) > 1 else "search"
+    query = sys.argv[2] if len(sys.argv) > 2 else "AI trends 2026"
+
+    videos = get_youtube_viral(query=query, mode=mode)
+    for v in videos:
+        print(f"TITLE: {v['title']}")
+        print(f"CHANNEL: {v['channel']}")
+        print(f"URL: {v['url']}")
+        print(f"PUBLISHED: {v['published_at']}")
+        if 'views' in v:
+            print(f"VIEWS: {v['views']}")
+        print("---")