Update app.py

app.py

@@ -3,10 +3,20 @@ import datetime
 import requests
 import pytz
 import yaml
+import os
+from dotenv import load_dotenv
 from tools.final_answer import FinalAnswerTool
-
+import wikipediaapi
+from newsapi import NewsApiClient
+from forex_python.converter import CurrencyRates
+import sympy
+import random
+import string
 from Gradio_UI import GradioUI
 
+# Load environment variables
+load_dotenv()
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
@@ -33,6 +43,139 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def get_weather(city: str) -> str:
+    """Get current weather information for a specified city
+    Args:
+        city: Name of the city to get weather for
+    """
+    # Using OpenWeatherMap API
+    API_KEY = os.getenv("OPENWEATHERMAP_API_KEY")
+    if not API_KEY:
+        return "Error: OpenWeatherMap API key not found in environment variables"
+    
+    base_url = "http://api.openweathermap.org/data/2.5/weather"
+    
+    try:
+        params = {
+            'q': city,
+            'appid': API_KEY,
+            'units': 'metric'  # For Celsius
+        }
+        response = requests.get(base_url, params=params)
+        data = response.json()
+        
+        if response.status_code == 200:
+            temp = data['main']['temp']
+            humidity = data['main']['humidity']
+            description = data['weather'][0]['description']
+            return f"Weather in {city}: Temperature: {temp}°C, Humidity: {humidity}%, Conditions: {description}"
+        else:
+            return f"Error getting weather: {data.get('message', 'Unknown error')}"
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+@tool
+def convert_currency(amount: float, from_currency: str, to_currency: str) -> str:
+    """Convert an amount from one currency to another
+    Args:
+        amount: Amount to convert
+        from_currency: Source currency code (e.g., USD)
+        to_currency: Target currency code (e.g., EUR)
+    """
+    try:
+        c = CurrencyRates()
+        result = c.convert(from_currency.upper(), to_currency.upper(), amount)
+        return f"{amount} {from_currency.upper()} = {result:.2f} {to_currency.upper()}"
+    except Exception as e:
+        return f"Error converting currency: {str(e)}"
+
+@tool
+def get_news_headlines(topic: str, count: int = 5) -> str:
+    """Get latest news headlines for a specific topic
+    Args:
+        topic: Topic to search news for
+        count: Number of headlines to return (default: 5)
+    """
+    API_KEY = os.getenv("NEWSAPI_KEY")
+    if not API_KEY:
+        return "Error: NewsAPI key not found in environment variables"
+        
+    newsapi = NewsApiClient(api_key=API_KEY)
+    
+    try:
+        news = newsapi.get_everything(q=topic, language='en', sort_by='publishedAt', page_size=count)
+        if news['articles']:
+            headlines = []
+            for idx, article in enumerate(news['articles'][:count], 1):
+                headlines.append(f"{idx}. {article['title']} ({article['source']['name']})")
+            return "\n".join(headlines)
+        return f"No news found for topic: {topic}"
+    except Exception as e:
+        return f"Error fetching news: {str(e)}"
+
+@tool
+def get_wikipedia_summary(topic: str) -> str:
+    """Get a summary of a Wikipedia article
+    Args:
+        topic: Topic to get summary for
+    """
+    wiki = wikipediaapi.Wikipedia('en')
+    try:
+        page = wiki.page(topic)
+        if page.exists():
+            # Get first two sentences or first 500 characters, whichever is shorter
+            summary = page.summary[:500]
+            if len(summary) == 500:
+                summary = summary[:summary.rindex('.')] + '.'
+            return summary
+        return f"No Wikipedia article found for: {topic}"
+    except Exception as e:
+        return f"Error fetching Wikipedia summary: {str(e)}"
+
+@tool
+def solve_math_expression(expression: str) -> str:
+    """Solve a mathematical expression
+    Args:
+        expression: Mathematical expression as string
+    """
+    try:
+        # Convert string to sympy expression
+        expr = sympy.sympify(expression)
+        result = expr.evalf()
+        return f"Result: {result}"
+    except Exception as e:
+        return f"Error solving expression: {str(e)}"
+
+@tool
+def generate_password(length: int = 12, include_special: bool = True) -> str:
+    """Generate a secure random password
+    Args:
+        length: Length of password (default: 12)
+        include_special: Include special characters (default: True)
+    """
+    try:
+        if length < 8:
+            return "Password length must be at least 8 characters"
+        
+        chars = string.ascii_letters + string.digits
+        if include_special:
+            chars += string.punctuation
+            
+        password = ''.join(random.choice(chars) for _ in range(length))
+        
+        # Ensure password contains at least one of each required type
+        if not any(c.isupper() for c in password):
+            password = random.choice(string.ascii_uppercase) + password[1:]
+        if not any(c.islower() for c in password):
+            password = password[:-1] + random.choice(string.ascii_lowercase)
+        if not any(c.isdigit() for c in password):
+            pos = random.randint(1, len(password)-2)
+            password = password[:pos] + random.choice(string.digits) + password[pos+1:]
+            
+        return f"Generated password: {password}"
+    except Exception as e:
+        return f"Error generating password: {str(e)}"
 
 final_answer = FinalAnswerTool()
 
@@ -55,7 +198,15 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[
+        final_answer,
+        get_weather,
+        convert_currency,
+        get_news_headlines,
+        get_wikipedia_summary,
+        solve_math_expression,
+        generate_password
+    ],
     max_steps=6,
     verbosity_level=1,
     grammar=None,