Update app.py

app.py

@@ -1,15 +1,16 @@
 import os
 import gradio as gr
+import cohere
 from crewai import Agent, Task, Crew
 from langchain_groq import ChatGroq
 from langchain_community.tools import DuckDuckGoSearchRun, DuckDuckGoSearchResults
 from crewai_tools import tool, SeleniumScrapingTool, ScrapeWebsiteTool
 
 # Define the DuckDuckGoSearch tool using the decorator for tool registration
-@tool('DuckDuckGoSearch')
-def search(search_query: str):
-    """Search the web for information on a given topic."""
-    return DuckDuckGoSearchRun().run(search_query)
+# @tool('DuckDuckGoSearch')
+# def search(search_query: str):
+#     """Search the web for information on a given topic."""
+#     return DuckDuckGoSearchRun().run(search_query)
 
 # Define the DuckDuckGoSearch tool
 @tool('DuckDuckGoResults')
@@ -32,16 +33,39 @@ def search_results(search_query: str):
     """
     return DuckDuckGoSearchResults().run(search_query)
 
+# Retrieve the Cohere API key from environment variables
+cohere_api_key = os.getenv('COHERE_API_KEY')
+co = cohere.Client(cohere_api_key)
 
-# Define the WebScrapper tool
 @tool('WebScrapper')
-def web_scrapper(url: str):
+def web_scrapper(url: str, topic: str):
     """
-    A tool designed to extract and read the content of a specified link.
+    A tool designed to extract and read the content of a specified link and generate a summary on a specific topic.
     It is capable of handling various types of web pages by making HTTP requests and parsing the received HTML content.
-    This tool can be particularly useful for web scraping tasks, data collection, or extracting specific information from websites.
+    This tool uses Cohere's API to generate summaries, which can be particularly useful for web scraping tasks, data collection,
+    or extracting specific information from websites.
+    
+    Args:
+    - url (str): The URL from which to scrape content.
+    - topic (str): The specific topic on which to generate a summary.
     """
-    return ScrapeWebsiteTool(website_url=url)
+    # Scrape content from the specified URL
+    scraper = ScrapeWebsiteTool(website_url=url)
+    content = scraper.run()
+    
+    # Prepare the prompt for generating the summary
+    prompt = f"Generate a summary of the following content on the topic ## {topic} ### \n\nCONTENT:\n\n" + content
+    
+    # Generate the summary using Cohere
+    response = co.chat(
+        model='command-r-plus',
+        message=prompt,
+        temperature=0.2,
+        chat_history=[],
+        prompt_truncation='AUTO'
+    )
+    
+    return response.text
 
 def kickoff_crew(topic: str) -> dict:
     try:
@@ -53,13 +77,12 @@ def kickoff_crew(topic: str) -> dict:
     
         # Initialize the Groq large language model
         groq_llm_70b = ChatGroq(temperature=0, groq_api_key=groq_api_key, model_name="llama3-70b-8192")
-        groq_llm_8b = ChatGroq(temperature=0, groq_api_key=groq_api_key, model_name="llama3-8b-8192")
     
         # Define Agents with Groq LLM
         researcher = Agent(
             role='Researcher',
             goal='Collect detailed information on {topic}',
-            tools=[search, search_results, web_scrapper],
+            tools=[search_results, web_scrapper],
             llm=groq_llm_70b,  # Assigning the Groq LLM here
             backstory=(
                 "As a diligent researcher, you explore the depths of the internet to "
@@ -94,8 +117,7 @@ def kickoff_crew(topic: str) -> dict:
         # Define Tasks
         research_task = Task(
             description=(
-                "Use DuckDuckGoSearch tool to gather initial information about {topic}. "
-                "Next, employ DuckDuckGoResults tool to gather full details of the insights from search results, reading them carefully and preparing detailed summaries on {topic}. "
+                "Use DuckDuckGoResults tool to gather full details of the insights from search results about ## {topic} ##, reading them carefully and preparing detailed summaries on {topic}. "
                 "Utilize the WebScrapper tool to extract additional information and insights from all links or URLs that appear significant regarding {topic} after analyzing the snippets of the search results. "
                 "Compile your findings into an initial draft, ensuring to include all sources with their titles and links relevant to the topic. "
                 "Throughout this process, maintain a high standard of accuracy and ensure that no information is fabricated or misrepresented."