mirror of
https://github.com/zhouxiaoka/autoclip.git
synced 2026-05-07 06:17:12 +08:00
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
语音识别环境设置脚本
|
||
自动安装 bcut-asr 和相关依赖
|
||
"""
|
||
import sys
|
||
import os
|
||
from pathlib import Path
|
||
|
||
# 添加项目根目录到Python路径
|
||
project_root = Path(__file__).parent.parent
|
||
sys.path.insert(0, str(project_root))
|
||
|
||
from scripts.install_bcut_asr import install_dependencies, check_bcut_asr_installed, check_ffmpeg_installed
|
||
import logging
|
||
|
||
# 设置日志
|
||
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
||
logger = logging.getLogger(__name__)
|
||
|
||
def main():
|
||
"""主函数"""
|
||
logger.info("🎤 AutoClip 语音识别环境设置")
|
||
logger.info("=" * 50)
|
||
|
||
# 检查当前状态
|
||
logger.info("检查当前环境状态...")
|
||
bcut_available = check_bcut_asr_installed()
|
||
ffmpeg_available = check_ffmpeg_installed()
|
||
|
||
logger.info(f"bcut-asr 状态: {'✅ 已安装' if bcut_available else '❌ 未安装'}")
|
||
logger.info(f"ffmpeg 状态: {'✅ 已安装' if ffmpeg_available else '❌ 未安装'}")
|
||
|
||
if bcut_available and ffmpeg_available:
|
||
logger.info("🎉 所有依赖已安装，无需重复安装")
|
||
return True
|
||
|
||
# 安装依赖
|
||
logger.info("开始安装缺失的依赖...")
|
||
success = install_dependencies()
|
||
|
||
if success:
|
||
logger.info("🎉 语音识别环境设置完成！")
|
||
logger.info("现在可以使用以下功能:")
|
||
logger.info(" - bcut-asr 云端语音识别（快速）")
|
||
logger.info(" - whisper 本地语音识别（可靠）")
|
||
logger.info(" - 智能回退机制")
|
||
return True
|
||
else:
|
||
logger.error("❌ 环境设置失败")
|
||
logger.info("请检查:")
|
||
logger.info(" 1. 网络连接是否正常")
|
||
logger.info(" 2. 系统权限是否足够")
|
||
logger.info(" 3. 依赖工具是否可用（git, pip等）")
|
||
return False
|
||
|
||
if __name__ == "__main__":
|
||
success = main()
|
||
sys.exit(0 if success else 1)
|
||
|