Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-25 09:19:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add support for loading multiple URLs into superbig

Browse Source
This commit is contained in:
oobabooga 2023-05-07 11:07:16 -03:00
parent 04eca9b65b
commit befa307c42
1 changed files with 20 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
extensions/superbig/script.py
View File

@ -92,8 +92,13 @@ def feed_file_into_collector(file, chunk_len, chunk_count):
yield i
def feed_url_into_collector(url, chunk_len, chunk_count):
yield 'Loading the URL...'
def feed_url_into_collector(urls, chunk_len, chunk_count):
urls = urls.strip().split('\n')
all_text = ''
cumulative = ''
for url in urls:
cumulative += f'Loading {url}...\n\n'
yield cumulative
html = urlopen(url).read()
soup = BeautifulSoup(html, features="html.parser")
for script in soup(["script", "style"]):
@ -103,7 +108,9 @@ def feed_url_into_collector(url, chunk_len, chunk_count):
lines = (line.strip() for line in text.splitlines())
chunks = (phrase.strip() for line in lines for phrase in line.split(" "))
text = '\n\n'.join(chunk for chunk in chunks if chunk)
for i in feed_data_into_collector(text, chunk_len, chunk_count):
all_text += text
for i in feed_data_into_collector(all_text, chunk_len, chunk_count):
yield i
@ -173,7 +180,7 @@ def ui():
update_data = gr.Button('Apply')
with gr.Tab("URL input"):
url_input = gr.Textbox(lines=1, label='Input URL')
url_input = gr.Textbox(lines=10, label='Input URL', info='Enter one or more URLs separated by newline characters')
update_url = gr.Button('Apply')
with gr.Tab("File input"):