238 lines
9.8 KiB
Python
238 lines
9.8 KiB
Python
# Copyright (C) 2025 AIDC-AI
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
"""
|
|
Style configuration components for web UI (middle column)
|
|
"""
|
|
|
|
import os
|
|
from pathlib import Path
|
|
|
|
import streamlit as st
|
|
from loguru import logger
|
|
|
|
from web.i18n import tr, get_language
|
|
from web.utils.async_helpers import run_async
|
|
from pixelle_video.config import config_manager
|
|
|
|
|
|
def render_style_config(pixelle_video):
|
|
"""Render style configuration section (middle column)"""
|
|
# TTS Section (moved from left column)
|
|
# ====================================================================
|
|
with st.container(border=True):
|
|
st.markdown(f"**{tr('section.tts')}**")
|
|
|
|
with st.expander(tr("help.feature_description"), expanded=False):
|
|
st.markdown(f"**{tr('help.what')}**")
|
|
st.markdown(tr("digital_tts.what"))
|
|
|
|
# Get TTS config
|
|
comfyui_config = config_manager.get_comfyui_config()
|
|
tts_config = comfyui_config["tts"]
|
|
|
|
# 只使用local模式，移除inference_mode选择
|
|
tts_mode = "local"
|
|
|
|
# ================================================================
|
|
# Local Mode UI (简化版本)
|
|
# ================================================================
|
|
# Import voice configuration
|
|
from pixelle_video.tts_voices import EDGE_TTS_VOICES, get_voice_display_name
|
|
|
|
# Get saved voice from config
|
|
local_config = tts_config.get("local", {})
|
|
saved_voice = local_config.get("voice", "zh-CN-YunjianNeural")
|
|
saved_speed = local_config.get("speed", 1.2)
|
|
|
|
# Build voice options with i18n
|
|
voice_options = []
|
|
voice_ids = []
|
|
default_voice_index = 0
|
|
|
|
for idx, voice_config in enumerate(EDGE_TTS_VOICES):
|
|
voice_id = voice_config["id"]
|
|
display_name = get_voice_display_name(voice_id, tr, get_language())
|
|
voice_options.append(display_name)
|
|
voice_ids.append(voice_id)
|
|
|
|
# Set default index if matches saved voice
|
|
if voice_id == saved_voice:
|
|
default_voice_index = idx
|
|
|
|
# Two-column layout: Voice | Speed
|
|
voice_col, speed_col = st.columns([1, 1])
|
|
|
|
with voice_col:
|
|
# Voice selector
|
|
selected_voice_display = st.selectbox(
|
|
tr("tts.voice_selector"),
|
|
voice_options,
|
|
index=default_voice_index,
|
|
key="digital_tts_local_voice"
|
|
)
|
|
|
|
# Get actual voice ID
|
|
selected_voice_index = voice_options.index(selected_voice_display)
|
|
selected_voice = voice_ids[selected_voice_index]
|
|
|
|
with speed_col:
|
|
# Speed slider
|
|
tts_speed = st.slider(
|
|
tr("tts.speed"),
|
|
min_value=0.5,
|
|
max_value=2.0,
|
|
value=saved_speed,
|
|
step=0.1,
|
|
format="%.1fx",
|
|
key="digital_tts_local_speed"
|
|
)
|
|
st.caption(tr("tts.speed_label", speed=f"{tts_speed:.1f}"))
|
|
|
|
# ================================================================
|
|
# TTS Preview (只支持local模式)
|
|
# ================================================================
|
|
with st.expander(tr("tts.preview_title"), expanded=False):
|
|
# Preview text input
|
|
preview_text = st.text_input(
|
|
tr("tts.preview_text"),
|
|
value="大家好，这是一段测试语音。",
|
|
placeholder=tr("tts.preview_text_placeholder"),
|
|
key="digital_tts_preview_text"
|
|
)
|
|
|
|
# Preview button
|
|
if st.button(tr("tts.preview_button"), key="digital_preview_tts", use_container_width=True):
|
|
with st.spinner(tr("tts.previewing")):
|
|
try:
|
|
# Build TTS params for local mode
|
|
tts_params = {
|
|
"text": preview_text,
|
|
"inference_mode": "local",
|
|
"voice": selected_voice,
|
|
"speed": tts_speed
|
|
}
|
|
|
|
audio_path = run_async(pixelle_video.tts(**tts_params))
|
|
|
|
# Play the audio
|
|
if audio_path:
|
|
st.success(tr("tts.preview_success"))
|
|
if audio_path.startswith('http'):
|
|
st.audio(audio_path)
|
|
else:
|
|
st.audio(audio_path, format="audio/mp3")
|
|
|
|
# Show file path
|
|
st.caption(f"📁 {audio_path}")
|
|
else:
|
|
st.error("Failed to generate preview audio")
|
|
except Exception as e:
|
|
st.error(tr("tts.preview_failed", error=str(e)))
|
|
logger.exception(e)
|
|
|
|
# ====================================================================
|
|
# Storyboard Template Section
|
|
# ====================================================================
|
|
|
|
def get_template_preview_path(template_path: str, language: str = "zh_CN") -> str:
|
|
"""
|
|
Get the preview image path for a template based on language.
|
|
|
|
Args:
|
|
template_path: Template path like "1080x1920/image_default.html"
|
|
language: Language code, either "zh_CN" or "en"
|
|
|
|
Returns:
|
|
Path to preview image in docs/images/
|
|
"""
|
|
# Extract size and template name from path
|
|
# e.g., "1080x1920/image_default.html" -> size="1080x1920", name="image_default"
|
|
path_parts = template_path.split('/')
|
|
if len(path_parts) >= 2:
|
|
size = path_parts[0] # e.g., "1080x1920"
|
|
template_file = path_parts[1] # e.g., "image_default.html"
|
|
template_name = template_file.replace('.html', '') # e.g., "image_default"
|
|
|
|
# Build preview image path
|
|
# Format: docs/images/{size}/{template_name}.jpg or {template_name}_en.jpg
|
|
# Chinese uses Chinese preview, all other languages use English preview for better i18n
|
|
suffix = "" if language == "zh_CN" else "_en"
|
|
|
|
# Try different image extensions
|
|
for ext in ['.jpg', '.png']:
|
|
preview_path = f"docs/images/{size}/{template_name}{suffix}{ext}"
|
|
if os.path.exists(preview_path):
|
|
return preview_path
|
|
|
|
# Fallback: try without language suffix (for templates with only one version)
|
|
for ext in ['.jpg', '.png']:
|
|
preview_path = f"docs/images/{size}/{template_name}{ext}"
|
|
if os.path.exists(preview_path):
|
|
return preview_path
|
|
|
|
# If no preview found, return empty string
|
|
return ""
|
|
|
|
# ====================================================================
|
|
# Workflow Selection Section (Auto-configured for Digital Human)
|
|
# ====================================================================
|
|
with st.container(border=True):
|
|
st.markdown(f"**{tr('digital_human.section.workflow')}**")
|
|
|
|
with st.expander(tr("help.feature_description"), expanded=False):
|
|
st.markdown(f"**{tr('help.what')}**")
|
|
st.markdown(tr("digital_human.workflow.what"))
|
|
|
|
# Get available workflows
|
|
all_workflows = pixelle_video.media.list_workflows()
|
|
|
|
# Auto-select digital human workflows (check file existence directly)
|
|
|
|
# First workflow: digital_image (generates collage and copy)
|
|
first_workflow = "runninghub/digital_image.json"
|
|
first_workflow_path = "workflows/runninghub/digital_image.json"
|
|
first_workflow_found = os.path.exists(first_workflow_path)
|
|
|
|
# Second workflow: digital_combination (generates final video)
|
|
second_workflow = "runninghub/digital_combination.json"
|
|
second_workflow_path = "workflows/runninghub/digital_combination.json"
|
|
second_workflow_found = os.path.exists(second_workflow_path)
|
|
|
|
# First workflow status
|
|
if first_workflow_found:
|
|
st.success(f"✅ {tr('digital_human.workflow.first_step')}")
|
|
else:
|
|
st.error(f"❌ {tr('digital_human.workflow.first_step')} {tr('digital_human.workflow.not_found')}")
|
|
|
|
# Second workflow status
|
|
if second_workflow_found:
|
|
st.success(f"✅ {tr('digital_human.workflow.second_step')}")
|
|
else:
|
|
st.error(f"❌ {tr('digital_human.workflow.second_step')} {tr('digital_human.workflow.not_found')}")
|
|
|
|
# Show workflow info
|
|
if first_workflow_found and second_workflow_found:
|
|
st.info(f"🎯 {tr('digital_human.workflow.ready')}")
|
|
else:
|
|
st.warning(f"⚠️ {tr('digital_human.workflow.missing')}")
|
|
|
|
# Return all style configuration parameters (简化版本，只支持local TTS)
|
|
return {
|
|
"tts_inference_mode": "local",
|
|
"tts_voice": selected_voice,
|
|
"tts_speed": tts_speed,
|
|
"first_workflow": first_workflow,
|
|
"second_workflow": second_workflow
|
|
}
|
|
|