import sys
from pathlib import Path
from transformers import AutoConfig, AutoModel
from qwen_tts.core.models.configuration_qwen3_tts import Qwen3TTSConfig
from qwen_tts.core.models.modeling_qwen3_tts import Qwen3TTSForConditionalGeneration

try:
    AutoConfig.register("qwen3_tts", Qwen3TTSConfig)
    AutoModel.register(Qwen3TTSConfig, Qwen3TTSForConditionalGeneration)
    print("Pre-registered architecture successfully!")
except Exception as reg_err:
    print("Pre-registration warning:", reg_err)

import qwen_tts
Model = getattr(qwen_tts, 'Qwen3TTSModel', None)
if Model is None:
    print("Qwen3TTSModel not found!")
    sys.exit(1)

MODEL_ROOT = '/opt/ai-avatar-demo/models/qwen3-tts'
print("Loading model from pretrained...")
model = Model.from_pretrained(MODEL_ROOT)
print("Model loaded successfully!")

print("Supported Speakers:", model.get_supported_speakers())
print("Supported Languages:", model.get_supported_languages())