import os
import sys

if getattr(sys, 'frozen', False):
    # 打包后，exe 目录就是顶级目录
    BASE_DIR = os.path.abspath(os.path.dirname(sys.executable))
else:
    # 开发环境，基于当前文件往上两级
    BASE_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))

VIDEO_DIR_DRAFT = os.path.join(BASE_DIR, "data/video_draft")
VIDEO_DIR_FINAL = os.path.join(BASE_DIR, "data/video_final")
COOKIE_DIR = os.path.join(BASE_DIR, "data/cookies")
DOCX_DIR_DOWNLOADED = os.path.join(BASE_DIR, "data/docx_downloaded")
DOCX_DIR_SELECTED = os.path.join(BASE_DIR, "data/docx_selected")
DOCX_DIR_FINAL = os.path.join(BASE_DIR, "data/docx_final")


FACE_SWAPPER_MODEL_PATH = os.path.join(BASE_DIR,"models","inswapper_128.onnx")

FACE_ENHANCER_MODEL_PATH = os.path.join(BASE_DIR,"models","GFPGANv1.4.pth") 


LOG_DIR = os.path.join(BASE_DIR, "logs")
os.makedirs(COOKIE_DIR, exist_ok=True)
os.makedirs(LOG_DIR, exist_ok=True)

DATABASE_PATH = os.path.join(BASE_DIR,"data/auto_media_publisher.sqlite3")


DEBUG = True
MAX_VIDEO_FILES = 14

HIDE_UPLOAD_PAGE = False   # 上传视频的时候是否隐藏页面，以观察错误/调试

DEFAULT_USER_AGENT= 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/127.0.4324.150 Safari/537.36'

PLAYWRIGHT_STEALTH_JS_PATH = os.path.join(BASE_DIR,"auto_media_publisher/utils/stealth.min.js")

DAJIALA_JSON_FILE_NAME = "data/dajiala_top15_data.json"

DAJIALA_URL_PREFIX = "https://www.dajiala.com/fbmain/main/v1/today_hot?industry=26&home=1&page=1&num=6&topType="
# 定义类别字典
DAJIALA_CATEGORY_DICT = {
    1: "综合", 2: "国际", 3: "体育", 4: "娱乐", 5: "社会", 6: "财经",
    7: "时事", 8: "科技", 9: "情感", 10: "汽车", 11: "教育", 12: "时尚",
    13: "游戏", 14: "军事", 15: "旅游", 16: "美食", 17: "文化", 18: "健康",
    19: "养生", 20: "搞笑", 21: "家居", 22: "动漫", 23: "宠物", 24: "母婴",
    25: "育儿", 26: "星座", 27: "运势", 28: "历史", 29: "音乐"
}

DAJIALA_MAX_CONCURRENT_PAGE = 10

WEIXIN_GZH_WATERMARK_HEIGHT = 40

from .conf_llm import *


# task 数据库 related
TASK_CONFIG_DIR  = os.path.join(BASE_DIR, "config/users")
TASK_THREAD_INTERVAL_SECONDS = 10 * 60  # 10分钟
MEDIA_CONCURRENT_COUNT = 2

PLATFORM_LOGIN_CONCURRENT_COUNT = 5
PLATFORM_UPLOAD_CONCURRENT_COUNT = 5



PUBLIC_IP_CITY_URL= ["http://ip-api.com/json" ,"https://ipinfo.io/json" ]   # ip/city
PUBLIC_IPV6_CITY_URL = [ "https://api.ip.sb/geoip" , "https://ifconfig.co/json" ]  # ip/city (ipV6)