SmokeyBandit commited on
Commit
ceb828a
·
verified ·
1 Parent(s): 49e76f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  import requests
3
  from bs4 import BeautifulSoup
4
- from duckduckgo_search import ddg
5
  from transformers import pipeline
6
 
7
  # ---- Initialize Summarization Pipeline ----
@@ -18,12 +18,16 @@ def web_scrape_agent(url):
18
  except Exception as e:
19
  return f"Error during scraping: {e}"
20
 
21
- # ---- Research Agent using DuckDuckGo Search ----
22
  def research_agent(query):
23
  try:
24
- results = ddg(query, max_results=5)
 
 
25
  if results:
26
- response = "\n".join([f"{result['title']}:\n{result['href']}" for result in results])
 
 
27
  return response
28
  else:
29
  return "No results found."
 
1
  import gradio as gr
2
  import requests
3
  from bs4 import BeautifulSoup
4
+ from duckduckgo_search import DDGS # Updated import for the new API
5
  from transformers import pipeline
6
 
7
  # ---- Initialize Summarization Pipeline ----
 
18
  except Exception as e:
19
  return f"Error during scraping: {e}"
20
 
21
+ # ---- Research Agent using DuckDuckGo Search with new API ----
22
  def research_agent(query):
23
  try:
24
+ # Use the DDGS class as a context manager to perform a text search.
25
+ with DDGS() as ddgs:
26
+ results = ddgs.text(query, max_results=5)
27
  if results:
28
+ response = "\n\n".join(
29
+ [f"{result.get('title', 'No Title')}:\n{result.get('href', 'No URL')}" for result in results]
30
+ )
31
  return response
32
  else:
33
  return "No results found."