Last approach

app.py

@@ -19,7 +19,14 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Enhanced Custom Tools ---
 @tool
 def serper_search(query: str) -> str:
-    """Search the web using Serper API with advanced result filtering"""
+    """Search the web using Serper API with advanced result filtering
+    
+    Args:
+        query (str): The search query to execute
+        
+    Returns:
+        str: Formatted search results
+    """
     try:
         api_key = os.getenv("SERPER_API_KEY")
         if not api_key:
@@ -62,7 +69,14 @@ def serper_search(query: str) -> str:
 
 @tool
 def wikipedia_search(query: str) -> str:
-    """Wikipedia search with full content extraction"""
+    """Wikipedia search with full content extraction
+    
+    Args:
+        query (str): The Wikipedia search query
+        
+    Returns:
+        str: Wikipedia search results with content
+    """
     try:
         # Clean query for Wikipedia
         clean_query = query.replace(" ", "_")
@@ -114,7 +128,14 @@ def wikipedia_search(query: str) -> str:
 
 @tool
 def enhanced_youtube_analyzer(url: str) -> str:
-    """YouTube analyzer with transcript extraction and pattern matching"""
+    """YouTube analyzer with transcript extraction and pattern matching
+    
+    Args:
+        url (str): YouTube video URL to analyze
+        
+    Returns:
+        str: Detailed video analysis
+    """
     try:
         # Extract video ID
         video_id_match = re.search(r'(?:v=|\/)([0-9A-Za-z_-]{11}).*', url)
@@ -132,7 +153,7 @@ def enhanced_youtube_analyzer(url: str) -> str:
             data = response.json()
             result = f"Title: {data.get('title', '')}\nAuthor: {data.get('author_name', '')}\n"
             
-        # NEW: Try to get transcript
+        # Try to get transcript
         try:
             transcript_url = f"https://youtubetranscript.com/?server_vid={video_id}"
             transcript_res = requests.get(transcript_url, timeout=20)
@@ -156,7 +177,15 @@ def enhanced_youtube_analyzer(url: str) -> str:
 
 @tool
 def text_processor(text: str, operation: str = "analyze") -> str:
-    """Text processing with enhanced operations"""
+    """Text processing with enhanced operations
+    
+    Args:
+        text (str): Text to process
+        operation (str): Processing operation to perform
+        
+    Returns:
+        str: Processed text result
+    """
     try:
         if operation == "reverse":
             return text[::-1]
@@ -177,7 +206,16 @@ def text_processor(text: str, operation: str = "analyze") -> str:
 
 @tool
 def discography_analyzer(artist: str, start_year: int = None, end_year: int = None) -> str:
-    """Discography analyzer with chart data verification"""
+    """Discography analyzer with chart data verification
+    
+    Args:
+        artist (str): Artist name to analyze
+        start_year (int): Optional start year filter
+        end_year (int): Optional end year filter
+        
+    Returns:
+        str: Discography analysis results
+    """
     try:
         # Search for discography information
         query = f"{artist} discography studio albums"
@@ -218,7 +256,7 @@ def discography_analyzer(artist: str, start_year: int = None, end_year: int = No
         for year, album in albums:
             result += f"{year}: {album}\n"
         
-        # NEW: Verify with official chart data
+        # Verify with official chart data
         try:
             chart_url = f"https://musicbrainz.org/ws/2/release-group?artist={artist}&type=album&fmt=json"
             chart_res = requests.get(chart_url, headers={'User-Agent': 'GAIA Agent'}, timeout=15)
@@ -246,7 +284,15 @@ def discography_analyzer(artist: str, start_year: int = None, end_year: int = No
 
 @tool
 def data_extractor(source: str, target: str) -> str:
-    """Enhanced data extractor with expanded classifications"""
+    """Enhanced data extractor with expanded classifications
+    
+    Args:
+        source (str): Source data to extract from
+        target (str): Target data to extract
+        
+    Returns:
+        str: Extracted data results
+    """
     try:
         if "botanical" in target.lower():
             # EXPANDED classification dictionary
@@ -293,7 +339,14 @@ def data_extractor(source: str, target: str) -> str:
 
 @tool
 def chess_analyzer(description: str) -> str:
-    """Chess analyzer with position evaluation"""
+    """Chess analyzer with position evaluation
+    
+    Args:
+        description (str): Description of chess position
+        
+    Returns:
+        str: Chess analysis and recommendations
+    """
     try:
         if "black" in description.lower() and "turn" in description.lower():
             analysis = "Position Analysis (Black to move):\n"
@@ -307,7 +360,7 @@ def chess_analyzer(description: str) -> str:
             elif "attack" in description.lower():
                 analysis += "\nAttacking Strategy:\n- Target weak squares around enemy king\n- Sacrifice material for initiative\n"
             
-            # NEW: Recommend common defenses
+            # Recommend common defenses
             analysis += "\nCommon Defensive Resources:\n"
             analysis += "- Pinning attacker pieces\n- Counter-sacrifices\n- Deflection tactics\n"
             
@@ -540,7 +593,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             submitted_answer = None
             for attempt in range(2):
                 try:
-                    submitted_answer = EnhancedGAIAAgent()(question_text)
+                    submitted_answer = agent(question_text)
                     break
                 except Exception as e:
                     print(f"Attempt {attempt + 1} failed: {e}")