Update app.py

Update `app.py` to Use Tools from `tools/`

app.py

@@ -1,99 +1,31 @@
-from smolagents import CodeAgent, HfApiModel, tool
-import datetime
-import requests
-import pytz
+from smolagents import CodeAgent, HfApiModel
 import yaml
-from transformers import pipeline
-from keybert import KeyBERT
-import wikipediaapi
-from tools.final_answer import FinalAnswerTool
+from tools import FinalAnswerTool, VisitWebpageTool, DuckDuckGoSearchTool
 from Gradio_UI import GradioUI
 
-@tool # Text Summarizer
-def text_summarizer(text: str, max_length: int = 50) -> str:
-    """Summarizes the input text.
-    
-    Args:
-        text (str): The text to summarize.
-        max_length (int): The maximum length of the summary.
-    
-    Returns:
-        str: The summarized version of the input text or an error message.
-    """
-    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-    try:
-        summary = summarizer(text, max_length=max_length, min_length=20, do_sample=False)
-        return summary[0]['summary_text']
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-@tool # Keyword Extractor
-def keyword_extractor(text: str, num_keywords: int = 5) -> str:
-    """Extracts keywords from the input text.
-    
-    Args:
-        text (str): The text to extract keywords from.
-        num_keywords (int): The number of keywords to extract.
-    
-    Returns:
-        str: A string containing the extracted keywords or an error message.
-    """
-    kw_model = KeyBERT()
-    try:
-        keywords = kw_model.extract_keywords(text, keyphrase_ngram_range=(1, 2), top_n=num_keywords)
-        return ", ".join([kw[0] for kw in keywords])
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-@tool # Wikipedia Lookup
-def wikipedia_lookup(query: str, sentences: int = 3) -> str:
-    """Fetches a summary from Wikipedia.
-    
-    Args:
-        query (str): The topic to search for.
-        sentences (int): The number of sentences for the summary.
-    
-    Returns:
-        str: A summary of the topic or an error message.
-    """
-    wiki = wikipediaapi.Wikipedia('en', headers={"User-Agent": "raul-padua/HF_Agents_Course"})
-    page = wiki.page(query)
-    return page.summary[:500] + '...' if page.exists() else "No page found."
-
-@tool # Live Weather Fetching
-def weather_fetcher(city: str) -> str:
-    """Fetches current weather conditions for a city.
-    
-    Args:
-        city (str): The city name to get weather data for.
-    
-    Returns:
-        str: A string describing the weather or an error message.
-    """
-    try:
-        api_key = "b648dc2263850640175b2c0b26c1a974"
-        url = f"http://api.openweathermap.org/data/2.5/weather?q={city}&appid={api_key}&units=metric"
-        response = requests.get(url).json()
-        return f"{response['weather'][0]['description']}, {response['main']['temp']}°C"
-    except Exception as e:
-        return f"Error: {str(e)}"
+# Load prompts
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
 
+# Initialize tools
 final_answer = FinalAnswerTool()
+web_search = DuckDuckGoSearchTool(max_results=5)
+visit_webpage = VisitWebpageTool()
+
+# Configure model
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct'
 )
 
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
-
+# Create agent
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, text_summarizer, keyword_extractor, wikipedia_lookup, weather_fetcher],
+    tools=[final_answer, web_search, visit_webpage],
     max_steps=6,
     verbosity_level=1,
     prompt_templates=prompt_templates
 )
 
-GradioUI(agent).launch()
+# Launch Gradio UI