Update app.py

app.py

@@ -8,8 +8,8 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 from tools.visit_webpage import VisitWebpageTool
 
-MAX_STEPS = 15
-MAX_SEARCH_RESULTS = 10 # should be smaller than MAX_STEPS, as we potentially visit each result, and each visit costs one step
+MAX_STEPS = 8
+MAX_SEARCH_RESULTS = 5 # should be smaller than MAX_STEPS, as we potentially visit each result, and each visit costs one step
 
 ddg_search_tool = DuckDuckGoSearchTool(max_results=MAX_SEARCH_RESULTS)
 
@@ -29,6 +29,31 @@ def find_job_websites(job:str, location:str)-> str:
     
     return results
 
+visit_webpage = VisitWebpageTool()
+
+
+# 2. Text Summarizer
+summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+def text_summarizer(text: str) -> str:
+    try:
+        summary = summarizer(text, max_length=1000, min_length=20, do_sample=False)
+        return summary[0]['summary_text']
+    except Exception as e:
+        return f"Error in summarization: {str(e)}"
+
+@tool
+def visit_webpage_summarization(url:str)-> str:
+    """A tool that visits a webpage at the given url and, reads its content as a markdown string,
+    and extracts relevant information. Use this to browse webpages.
+    Args:
+        url: The url of the webpage to visit.
+    Returns:
+        A summarization containing all relevant information about the webpage's content.
+    """
+    content = visit_webpage(url)
+    
+    return text_summarizer(content)
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -45,7 +70,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
-visit_webpage = VisitWebpageTool()
+
 final_answer = FinalAnswerTool()
 
 # 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:
@@ -59,9 +84,6 @@ custom_role_conversions=None,
 )
 
 
-# Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
@@ -69,7 +91,7 @@ agent = CodeAgent(
     model=model,
     tools=[
         find_job_websites,
-        visit_webpage,
+        visit_webpage_summarization,
         final_answer
     ],
     max_steps=MAX_STEPS,