| from __future__ import annotations |
|
|
| import gradio as gr |
|
|
| from . import common |
| from . import frame_planning as frames |
|
|
|
|
| def validate_user_process_definition(process_definition: dict | None, get_model_def) -> list[str]: |
| settings = process_definition.get("settings") if isinstance(process_definition, dict) else None |
| if not isinstance(settings, dict): |
| return ["The selected settings file could not be read."] |
| model_type = str(settings.get("model_type") or "").strip() |
| if len(model_type) == 0: |
| return ["The selected settings file does not define a model."] |
| try: |
| model_def = frames.require_model_def(model_type, get_model_def) |
| except gr.Error as exc: |
| return [common.get_error_message(exc) or "The selected model is not available."] |
| problems: list[str] = [] |
| image_mode = common.coerce_int(settings.get("image_mode"), 0) |
| if image_mode != 0 or bool(model_def.get("audio_only", False)) or bool(model_def.get("image_outputs", False)) or bool(settings.get("image_outputs", False)): |
| problems.append("The selected settings must generate a video, not images or audio only.") |
| video_prompt_type = str(settings.get("video_prompt_type") or "") |
| if "V" not in video_prompt_type: |
| problems.append("Control Video must be enabled.") |
| if "I" in video_prompt_type: |
| problems.append("Reference Images must be disabled.") |
| image_prompt_types_allowed = str(model_def.get("image_prompt_types_allowed", "") or "") |
| if "V" not in image_prompt_types_allowed: |
| problems.append("The selected model must support using a source video for continuation.") |
| audio_prompt_type = str(settings.get("audio_prompt_type") or "") |
| audio_features = [] |
| if "A" in audio_prompt_type: |
| audio_features.append("Audio Source") |
| if "B" in audio_prompt_type: |
| audio_features.append("Audio Source #2") |
| if "X" in audio_prompt_type: |
| audio_features.append("two-speaker auto separation") |
| if len(audio_features) > 0: |
| problems.append("Disable these audio features: " + ", ".join(audio_features) + ".") |
| return problems |
|
|
|
|
| def format_user_process_validation_error(process_definition: dict | None, problems: list[str]) -> str: |
| name = str((process_definition or {}).get("name") or "selected settings").strip() |
| if len(problems) == 0: |
| return "" |
| return f'Cannot add "{name}" as a full-video process:\n- ' + "\n- ".join(problems) |
|
|
