Update app.py

app.py

@@ -1,86 +1,45 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from duckduckgo_search import ddg
+from typing import List
+import time
 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_topics(query: 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.
+def search_internet(query: str, max_results: int = 3) -> str:
+    """A tool that searches the internet for information about a given query.
     
     Args:
-        query: 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'
+        query: The search query (e.g., 'latest AI news', 'climate change updates')
+        max_results: Maximum number of results to return (default: 3)
     """
     try:
-        # Initialize pytrends
-        pytrends = TrendReq(hl='en-US', tz=360)
-        
-        # 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()
+        # Perform the search
+        results = ddg(query, max_results=max_results)
         
-        # Get interest over time
-        interest_df = pytrends.interest_over_time()
+        if not results:
+            return f"No results found for '{query}'"
         
         # Format the results
-        result = f"Trending information for '{query}' in {location}:\n\n"
+        formatted_results = f"Search results for '{query}':\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')
+        for i, result in enumerate(results, 1):
+            formatted_results += f"{i}. {result['title']}\n"
+            formatted_results += f"   {result['link']}\n"
+            formatted_results += f"   {result['body']}\n\n"
             
-            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
+        return formatted_results
     
     except Exception as e:
-        return f"Error fetching trending information: {str(e)}\nPlease ensure you've provided a valid topic and location."
-
+        return f"Error performing search: {str(e)}"
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -118,7 +77,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_trending_topics, get_current_time_in_timezone,], ## add your tools here (don't remove final answer)
+    tools=[final_answer, search_internet, get_current_time_in_timezone,], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,