Update app.py

Added a function provided by the agent as a tool. Let's see if it can summarize information from a website, and incorporate that as an answer to a question, well :)

app.py

@@ -89,6 +89,50 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"The current local time in {timezone} is: {local_time}"
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
+@tool
+def summarize_and_answer_from_web(url, question):
+    '''
+    A tool which takes a pair of inputs, a website and a question, and finds all relevant information 
+    about that question from the website, and return both the summary and the answer
+    Args:
+        url: a string which says which website to get information from
+        question: a string of text asking for information from that website
+    '''
+    # Get the webpage content
+    webpage_content = webpage_contents_get(url=url)
+    
+    # Get links from the webpage content
+    links = webpage_contents_soup_links(response_content=webpage_content)
+    
+    # Get paragraphs from the webpage content
+    paragraphs = webpage_contents_soup_paragraphs(response_content=webpage_content)
+    
+    # Combine the extracted information into a summary
+    summary = f"Title: {paragraphs[0] if paragraphs else 'No title found'}\\n\\nHeadings:\\n"
+    headings = [para for para in paragraphs if len(para.split()) < 15]  # Heuristic for headings
+    if headings:
+        for heading in headings:
+            summary += f"- {heading}\\n"
+    else:
+        summary += "No headings found.\\n"
+    
+    summary += "\\nParagraphs:\\n"
+    for para in paragraphs:
+        summary += f"- {para}\\n"
+    
+    summary += "\\nLinks:\\n"
+    for link in links:
+        summary += f"- {link}\\n"
+    
+    # Answer the question based on the summary
+    # Simple heuristic: check if the question is in any of the paragraphs
+    answer = "No specific answer found."
+    for para in paragraphs:
+        if question.lower() in para.lower():
+            answer = para
+            break
+    
+    return summary, answer
 
 
 final_answer = FinalAnswerTool()
@@ -117,7 +161,8 @@ with open("prompts.yaml", 'r') as stream:
 agent = CodeAgent(
     model=model,
     tools=[final_answer, image_generation_tool,webpage_contents_get,
-          webpage_header_get, webpage_contents_soup_links, webpage_contents_soup_paragraphs], ## add your tools here (don't remove final answer)
+          webpage_header_get, webpage_contents_soup_links, webpage_contents_soup_paragraphs,
+          summarize_and_answer_from_web], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,