| import json
|
| from pathlib import Path
|
| from typing import Dict, Any, Optional
|
| import os
|
|
|
| class Settings:
|
| def __init__(self):
|
|
|
| project_root = Path(__file__).parent.parent
|
|
|
| self.settings_file = project_root / ".framepack" / "settings.json"
|
| self.settings_file.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
|
| self.default_settings = {
|
| "save_metadata": True,
|
| "gpu_memory_preservation": 6,
|
| "output_dir": str(project_root / "outputs"),
|
| "metadata_dir": str(project_root / "outputs"),
|
| "lora_dir": str(project_root / "loras"),
|
| "gradio_temp_dir": str(project_root / "temp"),
|
| "input_files_dir": str(project_root / "input_files"),
|
| "auto_save_settings": True,
|
| "gradio_theme": "default",
|
| "mp4_crf": 16,
|
| "clean_up_videos": True,
|
| "cleanup_temp_folder": False,
|
| "override_system_prompt": False,
|
| "system_prompt_template": "{\"template\": \"<|start_header_id|>system<|end_header_id|>\\n\\nDescribe the video by detailing the following aspects: 1. The main content and theme of the video.2. The color, shape, size, texture, quantity, text, and spatial relationships of the objects.3. Actions, events, behaviors temporal relationships, physical movement changes of the objects.4. background environment, light, style and atmosphere.5. camera angles, movements, and transitions used in the video:<|eot_id|><|start_header_id|>user<|end_header_id|>\\n\\n{}<|eot_id|>\", \"crop_start\": 95}",
|
| "startup_model_type": "None",
|
| "startup_preset_name": None
|
| }
|
| self.settings = self.load_settings()
|
|
|
| def load_settings(self) -> Dict[str, Any]:
|
| """Load settings from file or return defaults"""
|
| if self.settings_file.exists():
|
| try:
|
| with open(self.settings_file, 'r') as f:
|
| loaded_settings = json.load(f)
|
|
|
| settings = self.default_settings.copy()
|
| settings.update(loaded_settings)
|
| return settings
|
| except Exception as e:
|
| print(f"Error loading settings: {e}")
|
| return self.default_settings.copy()
|
| return self.default_settings.copy()
|
|
|
| def save_settings(self, **kwargs):
|
| """Save settings to file. Accepts keyword arguments for any settings to update."""
|
|
|
| self.settings.update(kwargs)
|
|
|
| for k, v in self.default_settings.items():
|
| self.settings.setdefault(k, v)
|
|
|
|
|
| for dir_key in ["output_dir", "metadata_dir", "lora_dir", "gradio_temp_dir"]:
|
| dir_path = self.settings.get(dir_key)
|
| if dir_path:
|
| os.makedirs(dir_path, exist_ok=True)
|
|
|
|
|
| with open(self.settings_file, 'w') as f:
|
| json.dump(self.settings, f, indent=4)
|
|
|
| def get(self, key: str, default: Any = None) -> Any:
|
| """Get a setting value"""
|
| return self.settings.get(key, default)
|
|
|
| def set(self, key: str, value: Any) -> None:
|
| """Set a setting value"""
|
| self.settings[key] = value
|
| if self.settings.get("auto_save_settings", True):
|
| self.save_settings()
|
|
|
| def update(self, settings: Dict[str, Any]) -> None:
|
| """Update multiple settings at once"""
|
| self.settings.update(settings)
|
| if self.settings.get("auto_save_settings", True):
|
| self.save_settings()
|
|
|
