08f155a551
This commit renames the website loader, chunker to web page, as it is loading and chunking a single url than the complete website.
30 lines
731 B
Python
30 lines
731 B
Python
import requests
|
|
|
|
from bs4 import BeautifulSoup
|
|
|
|
from embedchain.utils import clean_string
|
|
|
|
|
|
class WebPageLoader:
|
|
|
|
def load_data(self, url):
|
|
response = requests.get(url)
|
|
data = response.content
|
|
soup = BeautifulSoup(data, 'html.parser')
|
|
for tag in soup([
|
|
"nav", "aside", "form", "header",
|
|
"noscript", "svg", "canvas",
|
|
"footer", "script", "style"
|
|
]):
|
|
tag.string = " "
|
|
output = []
|
|
content = soup.get_text()
|
|
content = clean_string(content)
|
|
meta_data = {
|
|
"url": url,
|
|
}
|
|
output.append({
|
|
"content": content,
|
|
"meta_data": meta_data,
|
|
})
|
|
return output |