from langchain.tools import Tool, tool
from langchain_community.tools.tavily_search import TavilySearchResults
import math
import os

# Tool 1: Web Search (Tavily)
search = TavilySearchResults(
    tavily_api_key=os.getenv("TAVILY_API_KEY"),
    max_results=3
)
search_tool = Tool(
    name="web_search",
    func=lambda q: str(search.invoke(q)),
    description="Useful for searching the internet for current events, facts, or any real-time information. Input should be a search query."
)

# Tool 2: Calculator
@tool
def calculator(expression: str) -> str:
    """Evaluates a mathematical expression.
    Input should be a valid math expression like '2 + 2' or 'sqrt(16)'."""
    try:
        allowed = {k: v for k, v in math.__dict__.items()
                   if not k.startswith("__")}
        result = eval(expression, {"__builtins__": {}}, allowed)
        return str(result)
    except Exception as e:
        return f"Error: {str(e)}"

# Tool 3: Text Summarizer
@tool
def summarize_text(text: str) -> str:
    """Summarizes a long piece of text into key points.
    Input should be the text you want summarized."""
    words = text.split()
    if len(words) < 50:
        return "Text is already short: " + text
    sentences = text.split('.')[:3]
    return "Key content: " + '. '.join(sentences)

tools = [search_tool, calculator, summarize_text]