Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import gradio as gr
 import cohere
+import requests
 from crewai import Agent, Task, Crew, Process
 
 from langchain_groq import ChatGroq
@@ -8,6 +9,7 @@ from langchain_cohere import ChatCohere
 
 from langchain_community.tools import DuckDuckGoSearchRun, DuckDuckGoSearchResults
 from crewai_tools import tool, SeleniumScrapingTool, ScrapeWebsiteTool
+from duckduckgo_search import DDGS  # Asegúrate de tener instalado el paquete duckduckgo_search
 
 # Ensure essential environment variables are set
 cohere_api_key = os.getenv('COHERE_API_KEY')
@@ -20,12 +22,19 @@ if not groq_api_key:
 # Initialize API clients
 co = cohere.Client(cohere_api_key)
 
-# 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)
+# Jina.ai API key
+jina_api_key = os.getenv('JINA_API_KEY', 'jina_e62a3177429a4ab199bbbce5d8a3efb2sZB9jmZTZme67gGtEeldrlBYO2KP')
 
+def fetch_content(url):
+    try:
+        response = requests.get(
+            f"https://r.jina.ai/{url}",
+            headers={"Authorization": f"Bearer {jina_api_key}"}
+        )
+        response.raise_for_status()
+        return response.text
+    except requests.exceptions.RequestException as e:
+        return f"Error fetching content: {e}"
 
 # Define the DuckDuckGoSearch tool
 @tool('DuckDuckGoSearchResults')
@@ -38,10 +47,9 @@ def search_results(search_query: str) -> dict:
     Returns:
     - list: A list of dictionaries, where each dictionary represents a search result. Each dictionary includes 'snippet' of the page and the 'link' with the url linking to it.
     """
-    search = DuckDuckGoSearchResults(max_results=10)
-    results = search.run(search_query)
-    
-    return results
+    results = DDGS().text(search_query, max_results=5, timelimit='m')
+    results_list = [{"title": result['title'], "snippet": result['body'], "link": result['href']} for result in results]
+    return results_list
 
 @tool('WebScrapper')
 def web_scrapper(url: str, topic: str) -> str:
@@ -57,8 +65,7 @@ def web_scrapper(url: str, topic: str) -> str:
     - summary (str): summary of the url on the topic
     """
     # Scrape content from the specified URL
-    scraper = ScrapeWebsiteTool(website_url=url)
-    content = scraper.run()
+    content = fetch_content(url)
     
     # 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
@@ -195,4 +202,4 @@ def main():
     demo.queue(api_open=False, max_size=3).launch()
 
 if __name__ == "__main__":
-    main()
+    main()