oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-12-06 07:12:10 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix the UI failing to launch if the Notebook prompt is too long

Browse source
This commit is contained in:
oobabooga 2025-08-30 08:42:26 -07:00
parent 08f90f4b64
commit a3eb67e466
4 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
modules/prompts.py
View file

@ -22,8 +22,7 @@ def load_prompt(fname):
if file_path.exists(): if file_path.exists():
with open(file_path, 'r', encoding='utf-8') as f: with open(file_path, 'r', encoding='utf-8') as f:
text = f.read() text = f.read()
if len(text) > 0 and text[-1] == '\n': text = text.rstrip()
text = text[:-1]
return text return text
else: else:

3
modules/ui_default.py
View file

@ -22,8 +22,7 @@ def create_ui():
with gr.Row(): with gr.Row():
with gr.Column(): with gr.Column():
with gr.Row(): with gr.Row():
initial_text = load_prompt(shared.settings['prompt-notebook']) shared.gradio['textbox-default'] = gr.Textbox(value="", lines=27, label='Input', elem_classes=['textbox_default', 'add_scrollbar'])
shared.gradio['textbox-default'] = gr.Textbox(value=initial_text, lines=27, label='Input', elem_classes=['textbox_default', 'add_scrollbar'])
shared.gradio['token-counter-default'] = gr.HTML(value="<span>0</span>", elem_id="default-token-counter") shared.gradio['token-counter-default'] = gr.HTML(value="<span>0</span>", elem_id="default-token-counter")
with gr.Row(): with gr.Row():

3
modules/ui_notebook.py
View file

@ -30,8 +30,7 @@ def create_ui():
with gr.Column(scale=4): with gr.Column(scale=4):
with gr.Tab('Raw'): with gr.Tab('Raw'):
with gr.Row(): with gr.Row():
initial_text = load_prompt(shared.settings['prompt-notebook']) shared.gradio['textbox-notebook'] = gr.Textbox(label="", value="", lines=27, elem_id='textbox-notebook', elem_classes=['textbox', 'add_scrollbar'])
shared.gradio['textbox-notebook'] = gr.Textbox(label="", value=initial_text, lines=27, elem_id='textbox-notebook', elem_classes=['textbox', 'add_scrollbar'])
shared.gradio['token-counter-notebook'] = gr.HTML(value="<span>0</span>", elem_id="notebook-token-counter") shared.gradio['token-counter-notebook'] = gr.HTML(value="<span>0</span>", elem_id="notebook-token-counter")
with gr.Tab('Markdown'): with gr.Tab('Markdown'):

8
server.py
View file

@ -6,6 +6,7 @@ from pathlib import Path
from modules import shared from modules import shared
from modules.block_requests import OpenMonkeyPatch, RequestBlocker from modules.block_requests import OpenMonkeyPatch, RequestBlocker
from modules.logging_colors import logger from modules.logging_colors import logger
from modules.prompts import load_prompt
# Set up Gradio temp directory path # Set up Gradio temp directory path
gradio_temp_path = Path('user_data') / 'cache' / 'gradio' gradio_temp_path = Path('user_data') / 'cache' / 'gradio'
@ -109,6 +110,13 @@ def create_interface():
'filter_by_loader': (shared.args.loader or 'All') if not shared.args.portable else 'llama.cpp' 'filter_by_loader': (shared.args.loader or 'All') if not shared.args.portable else 'llama.cpp'
}) })
if shared.settings['prompt-notebook']:
prompt = load_prompt(shared.settings['prompt-notebook'])
shared.persistent_interface_state.update({
'textbox-default': prompt,
'textbox-notebook': prompt
})
# Clear existing cache files # Clear existing cache files
for cache_file in ['pfp_character.png', 'pfp_character_thumb.png']: for cache_file in ['pfp_character.png', 'pfp_character_thumb.png']:
cache_path = Path(f"user_data/cache/{cache_file}") cache_path = Path(f"user_data/cache/{cache_file}")