backend/server.py

from __future__ import annotations

import os
import uvicorn

from .app import app
from .config import load_config


if __name__ == "__main__":
    cfg = load_config()
    # HF 环境固定使用端口 7860
    port = 7860
    host = "0.0.0.0"
    
    # 检测是否在 HF Space 环境
    space_id = os.environ.get("SPACE_ID", "")
    space_host = os.environ.get("SPACE_HOST", "")
    
    print(f"[New NAI HF] 服务运行于 http://{host}:{port}")
    
    if space_id:
        # HF Space 环境
        if space_host:
            public_url = f"https://{space_host}"
        else:
            # 根据 SPACE_ID 构造 URL (格式: username-spacename)
            public_url = f"https://{space_id.replace('/', '-')}.hf.space"
        print(f"[New NAI HF] HF Space ID: {space_id}")
        print(f"[New NAI HF] 公共访问地址: {public_url}")
    else:
        print(f"[New NAI HF] 本地模式 - 请使用 http://localhost:{port} 访问")
    
    # 使用 0.0.0.0 监听以便外部访问
    uvicorn.run(app, host=host, port=port, reload=False)