🚧 download_youtube_comments

app.py

@@ -25,6 +25,7 @@ from tools import (
     fetch_file,
     normalize_place_name,
     is_drink,
+    download_youtube_comments,
 )
 
 # (Keep Constants as is)
@@ -90,6 +91,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
                 fetch_file,
                 normalize_place_name,
                 is_drink,
+                download_youtube_comments,
             ],
             verbosity_level=2,
             additional_authorized_imports=authorized_imports,

requirements.txt

@@ -6,3 +6,4 @@ smolagents[toolkit]
 torch
 transformers
 wikipedia-api
+youtube-comment-downloader

tools.py

@@ -7,6 +7,10 @@ import spaces
 from transformers import pipeline
 from pdfminer.high_level import extract_text
 import os
+from youtube_comment_downloader import (
+    YoutubeCommentDownloader,
+    SORT_BY_POPULAR,
+)
 
 
 @tool
@@ -353,3 +357,27 @@ def is_drink(item_name: str) -> bool:
     """
     drinks = ["soda", "cola", "coffee", "tea", "juice", "milkshake", "water"]
     return item_name.lower() in drinks
+
+
+@tool
+def download_youtube_comments(url: str, max_comments: int = 100) -> list[str]:
+    """
+    Download up to `max_comments` popular comments from a YouTube video.
+
+    Args:
+        url (str): Full URL of a YouTube video (e.g. "https://www.youtube.com/watch?v=abc123").
+        max_comments (int): Maximum number of comments to retrieve (default is 100).
+
+    Returns:
+        list[str]: List of plain-text comment strings. Returns an empty list if the video
+                   can't be accessed or no comments are found.
+    """
+    downloader = YoutubeCommentDownloader()
+    comments = []
+    for comment in downloader.get_comments_from_url(
+        url, sort_by=SORT_BY_POPULAR
+    ):
+        comments.append(comment["text"])
+        if len(comments) >= max_comments:
+            break
+    return comments