| |
| |
| |
| |
|
|
| import json |
| import os |
|
|
| |
| def read_json_file(file_path): |
| """ |
| Reads a JSON file's content and returns it. |
| Ensures content matches the expected format. |
| """ |
| if not os.access(file_path, os.R_OK): |
| print(f"Warning: No read permissions for file {file_path}") |
| return None |
|
|
| try: |
| with open(file_path, 'r', encoding='utf-8') as file: |
| content = json.load(file) |
| |
| if not all(['name' in item and 'prompt' in item and 'negative_prompt' in item for item in content]): |
| print(f"Warning: Invalid content in file {file_path}") |
| return None |
| return content |
| except Exception as e: |
| print(f"An error occurred while reading {file_path}: {str(e)}") |
| return None |
|
|
| |
| def load_styles_from_directory(directory): |
| """ |
| Loads styles from all JSON files in the directory. |
| Renames duplicate style names by appending a suffix. |
| """ |
| |
| json_files = [os.path.join(directory, file) for file in os.listdir(directory) if file.endswith('.json') and os.path.isfile(os.path.join(directory, file))] |
| combined_data = [] |
| seen = set() |
|
|
| for json_file in json_files: |
| json_data = read_json_file(json_file) |
| if json_data: |
| for item in json_data: |
| original_style = item['name'] |
| style = original_style |
| suffix = 1 |
| while style in seen: |
| style = f"{original_style}_{suffix}" |
| suffix += 1 |
| item['name'] = style |
| seen.add(style) |
| combined_data.append(item) |
|
|
| unique_style_names = [item['name'] for item in combined_data if isinstance(item, dict) and 'name' in item] |
| |
| return combined_data, unique_style_names |
|
|
| |
|
|
