rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Merge pull request #111 from one-some/ui2-well-something-because-i-dont-know-what-im-going-to-work-on-but-i-just-feel-like-working

Browse Source 
Add ability to sort stories as you wish!
This commit is contained in:
ebolam
2022-09-06 20:44:10 -04:00
committed by GitHub
parent cd42856f64 109d16f29a
commit 486d92fc7c
3 changed files with 329 additions and 183 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
aiserver.py
View File

@@ -7291,34 +7291,47 @@ def UI_2_load_model(data):
def UI_2_load_story_list(data):
file_popup("Select Story to Load", "./stories", "load_story", upload=True, jailed=True, folder_only=False, renameable=True,
deleteable=True, show_breadcrumbs=True, item_check=valid_story,
valid_only=True, hide_extention=True, extra_parameter_function=get_story_length,
column_names=['Story Name', 'Action Count'], show_filename=False,
column_widths=['auto', '100px'], advanced_sort=story_sort,
valid_only=True, hide_extention=True, extra_parameter_function=get_story_listing_data,
column_names=['Story Name', 'Action Count', 'Last Loaded'], show_filename=False,
column_widths=['auto', '150px', '150px'], advanced_sort=story_sort,
sort="Modified", desc=True)
def get_story_length(item_full_path, item, valid_selection):
def get_story_listing_data(item_full_path, item, valid_selection):
title = ""
action_count = -1
last_loaded = ""
if not valid_selection:
return ["", ""]
return [title, action_count, last_loaded]
with open(item_full_path, "r") as f:
js = json.load(f)
title = js['story_name'] if 'story_name' in js else ".".join(item.split(".")[:-1])
if 'file_version' not in js:
return [title, len(js['actions'])]
if js['file_version'] == 1:
return [title, len(js['actions'])]
return [title, 0 if js['actions']['action_count'] == -1 else js['actions']['action_count'] ]
title = js['story_name'] if 'story_name' in js else ".".join(item.split(".")[:-1])
action_count = len(js["actions"])
if title in koboldai_vars._system_settings.story_loads:
# UNIX Timestamp
last_loaded = int(time.mktime(time.strptime(koboldai_vars._system_settings.story_loads[title], "%m/%d/%Y, %H:%M:%S")))
if js.get("file_version", 1) == 1:
return [title, action_count, last_loaded]
action_count = 0 if js['actions']['action_count'] == -1 else js['actions']['action_count']
return [title, action_count, last_loaded]
def valid_story(file):
if file.endswith(".json"):
with open(file, "r") as f:
try:
js = json.load(f)
except:
pass
return False
return 'actions' in js
if file.endswith(".json"):
with open(file, "r") as f:
try:
js = json.load(f)
except:
pass
return False
return 'actions' in js
def story_sort(base_path, desc=False):
files = {}