Update app.py

app.py

@@ -4,27 +4,11 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-from bs4 import BeautifulSoup
-from duckduckgo_search import ddg
-import re
+from tools.visit_webpage import VisitWebpageTool
+from tools.web_search import DuckDuckGoSearchTool
 
 from Gradio_UI import GradioUI
 
-def generate_summary(text: str, num_sentences: int = 5) -> str:
-    """
-    A simple summarization function that extracts the first few sentences from the text.
-    
-    Args:
-        text: The full text to summarize.
-        num_sentences: The number of sentences to include in the summary.
-        
-    Returns:
-        A summary composed of the first few sentences.
-    """
-    # Split text into sentences based on punctuation
-    sentences = re.split(r'(?<=[.!?])\s+', text)
-    summary = " ".join(sentences[:num_sentences])
-    return summary
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -37,48 +21,6 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
-@tool
-def duckduck_tool(query: str, num_results: int = 5) -> str:
-    """
-    A tool that uses DuckDuckGo search to retrieve the latest information and webpage content for a given query.
-    
-    Args:
-        query: The search query string.
-        num_results: The number of search results to return (default is 5).
-        
-    Returns:
-        A string summarizing the latest information and webpage content related to the query.
-    """
-    # Perform DuckDuckGo search
-    results = ddg(query, max_results=num_results)
-    if not results:
-        return "No search results found."
-    
-    combined_content = ""
-    
-    # Iterate over search results to fetch content
-    for result in results:
-        url = result.get("href") or result.get("url")
-        if not url:
-            continue
-        try:
-            response = requests.get(url, timeout=5)
-            if response.status_code == 200:
-                soup = BeautifulSoup(response.text, "html.parser")
-                # Extract text from paragraph tags
-                paragraphs = soup.find_all("p")
-                page_text = " ".join(p.get_text() for p in paragraphs)
-                combined_content += page_text + "\n"
-        except Exception:
-            # Skip URLs that cause errors
-            continue
-
-    if not combined_content.strip():
-        return "No content could be retrieved from the search results."
-    
-    # Generate a summary using our custom summarization function
-    summary = generate_summary(combined_content, num_sentences=5)
-    return summary
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -97,6 +39,8 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 
 final_answer = FinalAnswerTool()
+visit_webpage = VisitWebpageTool()
+web_search = DuckDuckGoSearchTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
@@ -117,7 +61,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, duckduck_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, visit_webpage, web_search ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,