Spaces:
Running
Running
| #!/usr/bin/env python3 | |
| # -*- coding: utf-8 -*- | |
| """ | |
| Created on Tue Mar 11 19:08:55 2025 | |
| @author: blyss | |
| """ | |
| import os | |
| import json | |
| class SingletonMeta(type): | |
| """ | |
| The SingletonMeta class is useful for creating objects that persist as a single instance across the whole program. Basically a global class. | |
| """ | |
| _instances = {} | |
| def __call__(cls, *Parameters, **kwParameters): | |
| if cls not in cls._instances: | |
| cls._instances[cls] = super(SingletonMeta, cls).__call__(*Parameters, **kwParameters) | |
| return cls._instances[cls] | |
| class BlissfulSettings(metaclass=SingletonMeta): | |
| def __init__(self): | |
| """ | |
| Loads the settings from a 'settings.json' file, creating it with default settings if it doesn't exist. | |
| This method attempts to read the program's settings from a JSON file. If the file does not exist, | |
| it creates a new file with default settings. This ensures that the program can start with a known | |
| set of configurations and modify them as needed. | |
| This class is a SingletonMeta so even if we reinstantiate the class, this only happens the first time | |
| """ | |
| # These are globals that do not persist | |
| self.generating = 0 | |
| self.last_preview_file = "" | |
| default_settings = { | |
| "prompt": "a cat walks on the grass, realistic style", | |
| "resolution_x": 960, | |
| "resolution_y": 544, | |
| "fps": 24, | |
| "embedded_guidance": 6.0, | |
| "flow_shift": 7.0, | |
| "infer_steps": 50, | |
| "seed": 42, | |
| "video_length": 129, | |
| "attention": "sage", | |
| "blocks_to_swap": 0, | |
| "hidden_state_skip_layer": 2, | |
| "apply_final_norm": False, | |
| "reproduce": False, | |
| "fp8": True, | |
| "fp8_fast": False, | |
| "do_compile": False, | |
| "transformer_path": "", | |
| "text_encoder_1_path": "", | |
| "text_encoder_2_path": "", | |
| "vae_path": "", | |
| "lora_path": "", | |
| } | |
| if not os.path.exists("./settings.json"): | |
| with open("./settings.json", "w", encoding="utf-8") as file: | |
| json.dump(default_settings, file, indent=4) | |
| print("No existing settings found. Created default settings file.") | |
| with open("./settings.json", "r", encoding="utf-8") as file: | |
| data = json.load(file) | |
| for key, default_value in default_settings.items(): | |
| setattr(self, key, data.get(key, default_value)) | |
| def save_to_file(self): | |
| """ | |
| Saves the current settings to a JSON file named 'settings.json'. | |
| """ | |
| settings = { | |
| "prompt": self.prompt, | |
| "resolution_x": self.resolution_x, | |
| "resolution_y": self.resolution_y, | |
| "fps": self.fps, | |
| "embedded_guidance": self.embedded_guidance, | |
| "flow_shift": self.flow_shift, | |
| "infer_steps": self.infer_steps, | |
| "seed": self.seed, | |
| "video_length": self.video_length, | |
| "attention": self.attention, | |
| "blocks_to_swap": self.blocks_to_swap, | |
| "hidden_state_skip_layer": self.hidden_state_skip_layer, | |
| "apply_final_norm": self.apply_final_norm, | |
| "reproduce": self.reproduce, | |
| "fp8": self.fp8, | |
| "fp8_fast": self.fp8_fast, | |
| "do_compile": self.do_compile, | |
| "transformer_path": self.transformer_path, | |
| "text_encoder_1_path": self.text_encoder_1_path, | |
| "text_encoder_2_path": self.text_encoder_2_path, | |
| "vae_path": self.vae_path, | |
| "lora_path": self.lora_path, | |
| } | |
| with open("./settings.json", "w", encoding="utf-8") as file: | |
| json.dump(settings, file, indent=4) | |
| def update(self, option, value, label_target=None, label_value=None): | |
| """Method for updating various settings called via QT connection and may update an associated label/value""" | |
| setattr(self, option, value) | |
| if label_target is not None and label_value is not None: | |
| label_target.setText(str(label_value)) | |