Alexandre Gazola commited on
Commit
68cc240
·
1 Parent(s): 0c2f97b

workaround para leitura video youtube a partir de arquivo local

Browse files
count_max_bird_species_tool.py CHANGED
@@ -15,7 +15,11 @@ def count_max_bird_species_in_video(youtube_url: str) -> int:
15
  - int: The maximum number of bird species detected in the YouTube video.
16
  """
17
 
18
- frames = sample_youtube_video(youtube_url, 5)
 
 
 
 
19
  return count_max_bird_species(frames)
20
 
21
  def count_max_bird_species(framesStr: list[str]) -> int:
@@ -66,6 +70,9 @@ def count_max_bird_species(framesStr: list[str]) -> int:
66
  # Return the maximum number of distinct bird species detected in any frame
67
  return max_bird_species_count
68
 
 
 
 
69
 
70
  if __name__ == "__main__":
71
  sampled_frames = sample_youtube_video.invoke({
 
15
  - int: The maximum number of bird species detected in the YouTube video.
16
  """
17
 
18
+ # frames = sample_youtube_video(youtube_url, 5)
19
+ youtube_video_in_base_64 = os.path.join(BASE_DIR, 'files', frames_youtube_birds_video.b64)
20
+ print('youtube video path: ' + youtube_video_in_base_64)
21
+ frames = read_list_from_txt(youtube_video_in_base_64)
22
+
23
  return count_max_bird_species(frames)
24
 
25
  def count_max_bird_species(framesStr: list[str]) -> int:
 
70
  # Return the maximum number of distinct bird species detected in any frame
71
  return max_bird_species_count
72
 
73
+ def read_list_from_txt(file_path: str) -> list[str]:
74
+ with open(file_path, 'r', encoding='utf-8') as f:
75
+ return [line.rstrip('\n') for line in f]
76
 
77
  if __name__ == "__main__":
78
  sampled_frames = sample_youtube_video.invoke({
files/frames_youtube_birds_video.b64 ADDED
The diff for this file is too large to render. See raw diff
 
sample_youtube_video.py CHANGED
@@ -26,6 +26,7 @@ def sample_youtube_video(youtube_url: str, sample_rate: int) -> List[str]:
26
  'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/mp4', # Get the best quality video and audio in mp4 format
27
  'outtmpl': '%(title)s.%(ext)s', # Save with the video title
28
  'quiet': True, # Suppress console output
 
29
  }
30
 
31
  # Download the video
 
26
  'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/mp4', # Get the best quality video and audio in mp4 format
27
  'outtmpl': '%(title)s.%(ext)s', # Save with the video title
28
  'quiet': True, # Suppress console output
29
+ 'cookiefile': 'youtube_cookies.txt'
30
  }
31
 
32
  # Download the video