app_hf.py

"""
Hugging Face Spaces 启动脚本
这个文件是为了兼容某些不支持完整 Dockerfile 的 Spaces 配置
"""

import os
import sys

# 确保环境变量
os.environ.setdefault('SERVER_HOST', '0.0.0.0')
os.environ.setdefault('SERVER_PORT', '8000')
os.environ.setdefault('DATA_DIR', '/data')
os.environ.setdefault('LOG_DIR', '/data/logs')

print("==> Grok2API Hugging Face Spaces 版本启动中...")
print(f"==> Python 版本: {sys.version}")
print(f"==> 数据目录: {os.getenv('DATA_DIR')}")
print(f"==> 端口: {os.getenv('SERVER_PORT')}")

# 导入主应用
from app.main import app

if __name__ == "__main__":
    import uvicorn
    
    host = os.getenv('SERVER_HOST', '0.0.0.0')
    port = int(os.getenv('SERVER_PORT', 8000))
    
    print(f"==> 启动服务器: http://{host}:{port}")
    
    uvicorn.run(
        app,
        host=host,
        port=port,
        log_level="info",
        access_log=True,
    )