from dashscope.common.constants import DASHSCOPE_API_KEY_ENV


ENV = 'test'
# ENV = 'local'


DEBUG = True
### 数据库
#dev
DB_USER = 'root'
DB_PASSWORD = 'Hikoon123!'
DB_HOST = 'rm-bp18h64ad9ak4d7h5do.mysql.rds.aliyuncs.com'
DB_DATABASE = 'music_partner'

#Redis
REDIS_HOST = '172.23.209.46'
REDIS_PORT = 6379
REDIS_PSW = '1bvvpAmKXFhDDJXb'
REDIS_DB = 0
#新抖key
NEW_RANK_KEY = 'vh1gbvynpyegg6gebhgepgvc6'

BACK_BASE_URL = 'https://ai-test.hikoon.com/api/partner'

EMAIL_HOST = 'smtp.exmail.qq.com'
EMAIL_PORT = 465
EMAIL_HOST_USER = 'bigmusic@hikoon.com'
EMAIL_HOST_PASSWORD = 'Music!123'
#邮件接收人列表
EMAIL_RECEIVERS = ['1774507011@qq.com','yangsheng@hikoon.com']


#标签字典
TAG_DICT = {
 "viral_song": "网络热歌",
  "sad_songs": "伤感老歌",
  "folk_songs": "民谣",
  "catchy_pop": "口水歌",
  "kids_songs": "洗脑儿歌",
  "tk_songs": "抖音热歌",
  "net_songs": "网络歌曲",

  "dj_remix": "DJ嗨曲",
  "Cheesy_EDM": "土嗨/慢摇",
  "car_music": "车载音乐",
  "shout_rap": "喊麦",
  "heavy_metal": "重金属/土摇DJ嗨曲",

  "mandarin_pop": "华语流行",
  "mainstream_pop": "主流Pop",
  "sweet_songs": "甜歌/校园",
  "hip_rock": "嘻哈说唱R&B摇滚",
  "child_songs": "主流儿歌",

  "international_pop": "国外流行",
  "jp_pop": "日韩流行",
  "west_pop": "欧美流行",
  "el_edm": "电音EDM",

  "chinese_style": "国风",
  "opera_vocal": "戏腔/古韵",
  "guochao_EDM": "国潮电子",
  "gufeng_music": "传统器乐古风",

  "soundtrack_instrumental": "影视/纯音",
  "ys_ost": "影视OST",
  "pur_music": "纯音乐",
  "no_lyric": "无词BGM",

  "other_music": "其他",
  "jazz_blue": "爵士/蓝调",
  "voice_book": "有声书",
  "lab_music": "实验音乐",
  "healing": "治愈",
  "melancholy": "伤感",
  "lonely": "孤独",
  "sweet": "甜蜜",
  "inspiring": "励志",
  "missing": "思念",
  "nostalgic": "怀旧",
  "angry": "愤怒",
  "relaxing": "放松",
  "catchy": "魔性洗脑",
  "heroic": "悲壮",
  "calm": "平静",
  "festive": "喜庆",
  "romantic": "浪漫",
  "majestic": "雄壮",
  "bewitching": "蛊惑",
  "cathartic": "宣泄",
  "solemn": "庄重",
  "passionate": "激情",
  "heavy": "沉重",
  "happy": "快乐",
  "tense": "紧张",
  "horror": "恐怖",
  "touching": "感动",
  "spoof": "恶搞",
  "funny": "搞笑",
  "expectation": "期待",
  "remembrance": "怀念",
  "mysterious": "悬疑",
  "blessing": "祝福",
  "zen": "佛系",
  "soothing": "舒缓",
  "melodious": "悠扬",
  "warm": "温暖",
  "depressed": "忧郁",
  "elderly": "老年",
  "middle_aged": "中年",
  "young_adult": "青年",
  "teenager": "少年",
  "life_scene": "生活场景",
  "sports": "运动",
  "driving": "开车",
  "travel": "旅行",
  "sleep": "睡前",
  "study": "学习",
  "cafe": "咖啡厅",
  "bar": "酒吧",
  "douyin":"抖音",
  "restaurant": "餐厅",
  "car_scene": "汽车",
  "dance": "跳舞",
  "work": "工作",
  "nightclub": "夜店",
  "leisure": "休闲",
  "live_house": "live house",
  "square_dance": "广场舞",
  "wedding": "婚礼",
  "dating": "约会",
  "festival_scene": "节日场景",
  "summer": "夏天",
  "winter": "冬天",
  "autumn": "秋天",
  "spring_festival": "春节",
  "christmas": "圣诞",
  "valentine": "情人节",
  "time_scene": "时间场景",
  "morning": "清晨",
  "afternoon": "午后",
  "evening": "夜晚",
  "midnight": "深夜",
  "regional_scene": "地域场景",
  "campus": "校园",
  "city": "城市",
  "grassland": "草原",
  "tibet": "西藏",
  "xinjiang": "新疆",
  "transition_style": "转场类",
  "card_point_switch": "卡点切换画面类",
  "reverse_suspense": "反转悬念类",
  "emotion_contrast": "情绪对比类",
  "mashup_collection": "混剪合集类",
  "emotional_resonance": "情感共鸣向剪辑",
  "scene_adaptation": "场景适配剪辑",
  "highlight_slice": "高光切片剪辑",
  "live_performance": "现场表演类",
  "singer_live": "歌手现场演唱",
  "talent_cover": "达人翻唱表演",
  "audience_interaction": "观众互动表演",
  "card_point_speed": "卡点、变速类",
  "multi_scene_fragment": "多场景碎片化卡点",
  "tech_effect_speed": "技术流特效变速",
  "lyric_concrete": "歌词具象化卡点",
  "loop_speed_brainwash": "循环变速洗脑",
  "ugc_co_creation": "UGC共创类",
  "jianying_template": "剪映模板",
  "ai_singing": "AI唱歌",
  "emotional_quotes": "情感语录类",
  "late_night_emo": "深夜emo类",
  "morning_inspiration": "清晨励志类",
  "memory_destiny": "回忆杀/宿命感类",
  "dynamic_lyrics_visual": "动态歌词可视化",
  "basic_lyrics_effect": "基础歌词动效",
  "creative_visual_enhance": "创意视觉强化",
  "adaptation": "改编",
  "special_effects_interaction": "特效互动类",
  "gesture_magic_effect": "手势魔法特效互动",
  "lip_sync_challenge": "对口型挑战",
  "douyin_effect_show": "抖音特效变装秀",
  # 听感演绎流
    "singing_montage": "演唱混剪",
    "live_singing": "现场演唱",

    # 视觉冲击流
    "change_transition": "变装转场",
    "hand_dance": "手势舞",
    "addictive_dance": "魔性舞蹈",
    "landscape_account": "风景号",

    # 氛围素材流
    "cute_pets": "萌宠",
    "movie_anime_edit": "影视剧/动漫混剪",
    "chinese_classical": "古风",
    "mood_post": "图文心情",

    # 情感共鸣流
    "animated_lyrics": "动态歌词",
    "storytelling": "故事演绎",
    "beauty_snaps": "颜值随拍"
}


# 模型相关配置
BASE_MODEL = "/data/qufeng/models--MIT--ast-finetuned-audioset-10-10-0.4593/snapshots/f826b80d28226b62986cc218e5cec390b1096902"
MOE_DIR = "/data/qufeng/moe_outputs"
BASELINE_CHECKPOINT = "/data/qufeng/best_epoch_base.pt"
LABEL_MAPPING = "/data/qufeng/label_mapping.txt"
DEVICE = "cuda"  # 可选: cuda/mps/cpu，为空时自动选择
ROUTER_CHECKPOINT = ""  # 为空时自动从 moe_dir/joint_train/joint/router_best.pt 推断
EXPERTS_DIR = ""  # 为空时自动从 moe_dir/experts_train/experts 推断

# 音频处理配置
CHUNK_SECONDS = 10.24  # 按多少秒切块推理
CROP_SECONDS = 204.8  # 若音频超过该时长，则仅截取中间这段再切块
MAX_CHUNKS = 10  # 每首歌最多使用多少个切片参与推理
CHUNK_BATCH_SIZE = 8  # 切块推理的 batch size
ROUTING_THRESHOLD = 0.6

API_CONFIG = {
    "api_key": "sk-d9b4d3581bde47d887354f9160a509a2",
    "base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
    "model": "qwen3-omni-flash",
    "audio_mode": "auto",
    "timeout": 15,
    "lyrics_timeout": 60,
    "lyrics_retries": 2,
    "max_retries": 5,
    "retry_delay": 5
}
# API_CONFIG_91 = {
#     "api_key": "sk-E90VNVMyhfk2zDBDoToCXoipzGofD2SobwBqaCzbG3junlob",
#     "base_url": "https://api.91aopusi.com/v1",
#     "model": "qwen3-omni-flash",
#     "audio_mode": "auto",
#     "timeout": 30,
#     "lyrics_timeout": 60,
#     "max_retries": 5,
#     "retry_delay": 5
# }

DASHSCOPE_API_KEY = 'sk-d9b4d3581bde47d887354f9160a509a2'

OSS_ACCESS_KEY_ID='LTAI4G7UvaW2e4UTCb3KCNjN'
OSS_ACCESS_KEY_SECRET='ow5hlVMmJAQY9o7nEAtMER6MFkPedm'
OSS_ENDPOINT='oss-cn-hangzhou.aliyuncs.com'
OSS_ENDPOINT_INTERNAL='oss-cn-hangzhou-internal.aliyuncs.com'
OSS_BUCKET_NAME='ai-sound-data-test'