test

app.py

@@ -15,7 +15,7 @@ from langchain_core.messages import SystemMessage
 
 # --- Custom Tools ---
 from wikipedia_tool import wikipedia_revision_by_year_keyword
-from count_max_bird_species_tool import count_max_distinct_bird_species_in_video
+from count_max_bird_species_tool import count_max_bird_species_in_video
 from image_to_text_tool import image_to_text
 from internet_search_tool import internet_search
 from botanical_classification_tool import get_botanical_classification
@@ -36,7 +36,7 @@ class LangChainAgent:
 
         tools = [
             wikipedia_revision_by_year_keyword,
-            count_max_distinct_bird_species_in_video,
+            count_max_bird_species_in_video,
             image_to_text,
             internet_search,
             get_botanical_classification,
@@ -137,7 +137,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         try:
             file_name = item.get("file_name")
             #provisorio
-            if not file_name:
+            if not file_name and "birds" not in question_text:
                 continue
             question_text_for_agent = question_text
             if file_name:

count_max_bird_species_tool.py

@@ -4,7 +4,7 @@ from langchain_core.tools import tool
 from utils import decode_base64_to_frame
 
 @tool
-def count_max_distinct_bird_species_in_video(youtube_url: str) -> int:
+def count_max_bird_species_in_video(youtube_url: str) -> int:
     """
     Count the maximum number of bird species in a YouTube video.
 
@@ -16,9 +16,9 @@ def count_max_distinct_bird_species_in_video(youtube_url: str) -> int:
     """
 
     frames = sample_youtube_video(youtube_url, 5)
-    return count_max_distinct_bird_species(frames)
+    return count_max_bird_species(frames)
 
-def count_max_distinct_bird_species(framesStr: list[str]) -> int:
+def count_max_dbird_species(framesStr: list[str]) -> int:
     """
     Count the maximum number of bird species in a single frame considering a list of frames.
 

requirements.txt

@@ -17,4 +17,5 @@ whisper
 yt-dlp
 wikipedia-api
 google-generativeai
-openpyxl
+openpyxl
+tabulate