Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,6 +6,8 @@ import pandas as pd
|
|
| 6 |
import json
|
| 7 |
from urllib.parse import quote
|
| 8 |
from bs4 import BeautifulSoup
|
|
|
|
|
|
|
| 9 |
|
| 10 |
|
| 11 |
from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, load_tool, tool, FinalAnswerTool, OpenAIServerModel, VisitWebpageTool
|
|
@@ -45,28 +47,8 @@ def simple_web_search(query: str, num_results: int = 5) -> str:
|
|
| 45 |
num_results: Result count.
|
| 46 |
"""
|
| 47 |
try:
|
| 48 |
-
|
| 49 |
-
headers = {
|
| 50 |
-
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; rv:91.0) Gecko/20100101 Firefox/91.0'
|
| 51 |
-
}
|
| 52 |
-
response = requests.get(url, headers=headers, timeout=10)
|
| 53 |
-
response.raise_for_status()
|
| 54 |
-
|
| 55 |
-
soup = BeautifulSoup(response.text, 'html.parser')
|
| 56 |
-
results = []
|
| 57 |
-
|
| 58 |
-
for result in soup.select('.result')[:num_results]:
|
| 59 |
-
title = result.select_one('.result__title a')
|
| 60 |
-
snippet = result.select_one('.result__snippet')
|
| 61 |
-
if title and snippet:
|
| 62 |
-
results.append({
|
| 63 |
-
'title': title.get_text(strip=True),
|
| 64 |
-
'link': title['href'],
|
| 65 |
-
'snippet': snippet.get_text(strip=True)
|
| 66 |
-
})
|
| 67 |
-
|
| 68 |
return json.dumps(results if results else {"error": "No results found"})
|
| 69 |
-
|
| 70 |
except Exception as e:
|
| 71 |
return json.dumps({"error": str(e)})
|
| 72 |
#simple_web_search = simple_web_search()
|
|
|
|
| 6 |
import json
|
| 7 |
from urllib.parse import quote
|
| 8 |
from bs4 import BeautifulSoup
|
| 9 |
+
from duckduckgo_search import ddg
|
| 10 |
+
|
| 11 |
|
| 12 |
|
| 13 |
from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, load_tool, tool, FinalAnswerTool, OpenAIServerModel, VisitWebpageTool
|
|
|
|
| 47 |
num_results: Result count.
|
| 48 |
"""
|
| 49 |
try:
|
| 50 |
+
results = ddg(query, max_results=num_results)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
return json.dumps(results if results else {"error": "No results found"})
|
|
|
|
| 52 |
except Exception as e:
|
| 53 |
return json.dumps({"error": str(e)})
|
| 54 |
#simple_web_search = simple_web_search()
|