mirror of
https://github.com/StanGirard/quivr.git
synced 2024-11-28 05:13:57 +03:00
26 lines
725 B
Python
26 lines
725 B
Python
|
from .common import process_file
|
||
|
|
from langchain.document_loaders import UnstructuredHTMLLoader
|
||
|
|
import requests
|
||
|
|
import re
|
||
|
|
import unicodedata
|
||
|
|
import tempfile
|
||
|
|
import os
|
||
|
|
from fastapi import UploadFile
|
||
|
|
|
||
|
|
|
||
|
|
def process_html(vector_store, file: UploadFile, stats_db):
|
||
|
|
return process_file(vector_store, file, UnstructuredHTMLLoader, ".html", stats_db=stats_db)
|
||
|
|
|
||
|
|
|
||
|
|
def get_html(url):
|
||
|
|
response = requests.get(url)
|
||
|
|
if response.status_code == 200:
|
||
|
|
return response.text
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def slugify(text):
|
||
|
|
text = unicodedata.normalize('NFKD', text).encode('ascii', 'ignore').decode('utf-8')
|
||
|
|
text = re.sub(r'[^\w\s-]', '', text).strip().lower()
|
||
|
|
text = re.sub(r'[-\s]+', '-', text)
|
||
|
|
return text
|