let5see/AI-Video
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化config中video的读取逻辑

Browse Source
This commit is contained in:
puke
2025-11-12 21:35:46 +08:00
parent 9eb8e7f82d
commit 8a0e0dedd8
3 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pixelle_video/config/__init__.py
View File

@@ -29,7 +29,7 @@ Usage:
if config_manager.validate(): if config_manager.validate():
print("Config is valid!") print("Config is valid!")
""" """
from .schema import PixelleVideoConfig, LLMConfig, ComfyUIConfig, TTSSubConfig, ImageSubConfig from .schema import PixelleVideoConfig, LLMConfig, ComfyUIConfig, TTSSubConfig, ImageSubConfig, VideoSubConfig
from .manager import ConfigManager from .manager import ConfigManager
from .loader import load_config_dict, save_config_dict from .loader import load_config_dict, save_config_dict
@@ -42,6 +42,7 @@ __all__ = [
"ComfyUIConfig", "ComfyUIConfig",
"TTSSubConfig", "TTSSubConfig",
"ImageSubConfig", "ImageSubConfig",
"VideoSubConfig",
"ConfigManager", "ConfigManager",
"config_manager", "config_manager",
"load_config_dict", "load_config_dict",

4
pixelle_video/config/manager.py
View File

@@ -135,6 +135,10 @@ class ConfigManager:
"image": { "image": {
"default_workflow": self.config.comfyui.image.default_workflow, "default_workflow": self.config.comfyui.image.default_workflow,
"prompt_prefix": self.config.comfyui.image.prompt_prefix, "prompt_prefix": self.config.comfyui.image.prompt_prefix,
},
"video": {
"default_workflow": self.config.comfyui.video.default_workflow,
"prompt_prefix": self.config.comfyui.video.prompt_prefix,
} }
} }

10
pixelle_video/config/schema.py
View File

@@ -59,12 +59,22 @@ class ImageSubConfig(BaseModel):
) )
class VideoSubConfig(BaseModel):
"""Video-specific configuration (under comfyui.video)"""
default_workflow: Optional[str] = Field(default=None, description="Default video workflow (optional)")
prompt_prefix: str = Field(
default="Minimalist black-and-white matchstick figure style illustration, clean lines, simple sketch style",
description="Prompt prefix for all video generation"
)
class ComfyUIConfig(BaseModel): class ComfyUIConfig(BaseModel):
"""ComfyUI configuration (includes global settings and service-specific configs)""" """ComfyUI configuration (includes global settings and service-specific configs)"""
comfyui_url: str = Field(default="http://127.0.0.1:8188", description="ComfyUI Server URL") comfyui_url: str = Field(default="http://127.0.0.1:8188", description="ComfyUI Server URL")
runninghub_api_key: Optional[str] = Field(default=None, description="RunningHub API Key (optional)") runninghub_api_key: Optional[str] = Field(default=None, description="RunningHub API Key (optional)")
tts: TTSSubConfig = Field(default_factory=TTSSubConfig, description="TTS-specific configuration") tts: TTSSubConfig = Field(default_factory=TTSSubConfig, description="TTS-specific configuration")
image: ImageSubConfig = Field(default_factory=ImageSubConfig, description="Image-specific configuration") image: ImageSubConfig = Field(default_factory=ImageSubConfig, description="Image-specific configuration")
video: VideoSubConfig = Field(default_factory=VideoSubConfig, description="Video-specific configuration")
class TemplateConfig(BaseModel): class TemplateConfig(BaseModel):