main.py

"""
主程序入口
调用环境检查脚本并显示结果
"""

import sys
import os

# 添加env_library到Python路径
sys.path.append(os.path.join(os.path.dirname(__file__), 'env_library'))

def main():
    """主函数"""
    print("=" * 60)
    print("项目环境验证工具")
    print("=" * 60)
    print("此工具将检查所有必要的Python包是否已正确安装。")
    print("对于torch，只检查是否安装，不检查具体版本。")
    print("其他包必须与requirements.txt中的版本完全一致。")
    print("\n按回车键开始检查...")
    input()
    
    try:
        # 导入并运行检查脚本
        from check_env import main as check_main
        return check_main()
    except ImportError as e:
        print(f"错误: 无法导入检查脚本 - {e}")
        print("请确保env_library/check_env.py存在")
        return 1
    except Exception as e:
        print(f"未知错误: {e}")
        return 1


if __name__ == "__main__":
    sys.exit(main())