| import os |
| import gradio as gr |
| import modules.localization as localization |
| import json |
|
|
|
|
| all_styles = [] |
|
|
|
|
| def try_load_sorted_styles(style_names, default_selected): |
| global all_styles |
|
|
| all_styles = style_names |
|
|
| try: |
| if os.path.exists('sorted_styles.json'): |
| with open('sorted_styles.json', 'rt', encoding='utf-8') as fp: |
| sorted_styles = [] |
| for x in json.load(fp): |
| if x in all_styles: |
| sorted_styles.append(x) |
| for x in all_styles: |
| if x not in sorted_styles: |
| sorted_styles.append(x) |
| all_styles = sorted_styles |
| except Exception as e: |
| print('Load style sorting failed.') |
| print(e) |
|
|
| unselected = [y for y in all_styles if y not in default_selected] |
| all_styles = default_selected + unselected |
|
|
| return |
|
|
|
|
| def sort_styles(selected): |
| global all_styles |
| unselected = [y for y in all_styles if y not in selected] |
| sorted_styles = selected + unselected |
| try: |
| with open('sorted_styles.json', 'wt', encoding='utf-8') as fp: |
| json.dump(sorted_styles, fp, indent=4) |
| except Exception as e: |
| print('Write style sorting failed.') |
| print(e) |
| all_styles = sorted_styles |
| return gr.CheckboxGroup.update(choices=sorted_styles) |
|
|
|
|
| def localization_key(x): |
| return x + localization.current_translation.get(x, '') |
|
|
|
|
| def search_styles(selected, query): |
| unselected = [y for y in all_styles if y not in selected] |
| matched = [y for y in unselected if query.lower() in localization_key(y).lower()] if len(query.replace(' ', '')) > 0 else [] |
| unmatched = [y for y in unselected if y not in matched] |
| sorted_styles = matched + selected + unmatched |
| return gr.CheckboxGroup.update(choices=sorted_styles) |
|
|
