mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2026-03-09 15:13:56 +01:00
e2548f69a9
If --user-data-dir is not set, auto-detect: use ../user_data when ./user_data doesn't exist, making it easy to share user data across portable builds by placing it one folder up.
38 lines
1 KiB
Python
38 lines
1 KiB
Python
from pathlib import Path
|
|
|
|
from modules import shared, utils
|
|
from modules.text_generation import get_encoded_length
|
|
|
|
|
|
def load_prompt(fname):
|
|
if not fname:
|
|
# Create new file
|
|
new_name = utils.current_time()
|
|
prompt_path = shared.user_data_dir / "logs" / "notebook" / f"{new_name}.txt"
|
|
prompt_path.parent.mkdir(parents=True, exist_ok=True)
|
|
initial_content = "In this story,"
|
|
prompt_path.write_text(initial_content, encoding='utf-8')
|
|
|
|
# Update settings to point to new file
|
|
shared.settings['prompt-notebook'] = new_name
|
|
|
|
return initial_content
|
|
|
|
file_path = shared.user_data_dir / 'logs' / 'notebook' / f'{fname}.txt'
|
|
if file_path.exists():
|
|
with open(file_path, 'r', encoding='utf-8') as f:
|
|
text = f.read()
|
|
text = text.rstrip()
|
|
|
|
return text
|
|
else:
|
|
return ''
|
|
|
|
|
|
def count_tokens(text):
|
|
try:
|
|
tokens = get_encoded_length(text)
|
|
return str(tokens)
|
|
except Exception:
|
|
return '0'
|