update --modified code

app.py

@@ -1,30 +1,85 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
-import datetime
 import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 from pytrends.request import TrendReq
+from typing import Dict, List, Optional
+import pandas as pd
+from datetime import datetime, timedelta
+import pycountry
 
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def get_trending_articles(topic: str, location: str)-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that fetches the most trending articles on a specified topic in specified location.
+def get_trending_topics(topic: str, location: str = "US", timeframe: str = "today 1-m") -> str:
+    """A tool that fetches trending topics and related queries for a specific topic in a given location.
+    
     Args:
-        topic: A string representing a topic that the user wants information about (e.g., 'Artificial Intelligence').
-        location: A string representing the location of users interest to find information on topic near/around that location (e.g., 'India/Mumbai').
+        topic: The topic to search for (e.g., 'Artificial Intelligence', 'Climate Change')
+        location: Country or region code (e.g., 'US', 'IN', 'GB'). Defaults to 'US'
+        timeframe: Time range for the trends (e.g., 'today 1-m', 'today 3-m', 'today 12-m'). Defaults to 'today 1-m'
     """
     try:
+        # Initialize pytrends
         pytrends = TrendReq(hl='en-US', tz=360)
-        kw_list = [topic]
-        pytrends.build_payload(kw_list, cat=0, timeframe='today 5-y', geo='', gprop='')
-        df = pytrends.trending_searches(pn=location) 
-        return f"Here is the article you requested on {topic}: {df}"
+        
+        # Try to convert full country name to code if provided
+        try:
+            country = pycountry.countries.search_fuzzy(location)[0]
+            location = country.alpha_2
+        except:
+            # If conversion fails, use the provided location as-is
+            pass
+        
+        # Build the payload
+        kw_list = [query]
+        pytrends.build_payload(
+            kw_list,
+            cat=0,
+            timeframe=timeframe,
+            geo=location,
+            gprop=''
+        )
+        
+        # Get related queries
+        related_queries = pytrends.related_queries()
+        
+        # Get interest over time
+        interest_df = pytrends.interest_over_time()
+        
+        # Format the results
+        result = f"Trending information for '{query}' in {location}:\n\n"
+        
+        # Add top related queries if available
+        if query in related_queries and related_queries[query]['top'] is not None:
+            top_queries = related_queries[query]['top'].head(5)
+            result += "Top Related Queries:\n"
+            for idx, row in top_queries.iterrows():
+                result += f"- {row['query']} (Score: {row['value']})\n"
+        
+        # Add rising queries if available
+        if query in related_queries and related_queries[query]['rising'] is not None:
+            rising_queries = related_queries[query]['rising'].head(5)
+            result += "\nRising Queries:\n"
+            for idx, row in rising_queries.iterrows():
+                result += f"- {row['query']} (Score: {row['value']}%)\n"
+        
+        # Add interest over time summary if available
+        if not interest_df.empty:
+            avg_interest = interest_df[query].mean()
+            max_interest = interest_df[query].max()
+            max_date = interest_df[query].idxmax().strftime('%Y-%m-%d')
+            
+            result += f"\nInterest Summary:\n"
+            result += f"- Average interest: {avg_interest:.1f}/100\n"
+            result += f"- Peak interest: {max_interest:.1f}/100 (on {max_date})\n"
+        
+        return result
+    
     except Exception as e:
-        return f"Error fetching information about '{topic}': {str(e)}"
+        return f"Error fetching trending information: {str(e)}\nPlease ensure you've provided a valid topic and location."
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: