Files
t6_mem0/embedchain/loaders/web_page.py
2023-07-11 08:31:42 +05:30

31 lines
774 B
Python

import requests
from bs4 import BeautifulSoup
from embedchain.utils import clean_string
class WebPageLoader:
def load_data(self, url):
''' Load data from a web page. '''
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