mirror of
https://github.com/linyqh/NarratoAI.git
synced 2025-12-11 10:32:49 +00:00
6715c29057
在video_processor.py中，添加了对Windows N卡硬件加速的支持，优化了帧提取过程，改进了提取成功率的统计和错误处理。同时，在generate_script_docu.py中，增强了对硬件加速失败的处理逻辑，提供了详细的错误信息和解决建议，提升了用户体验。
281 lines
9.9 KiB
Python
281 lines
9.9 KiB
Python
"""
|
||
FFmpeg 诊断和配置组件
|
||
为用户提供 FFmpeg 兼容性诊断和配置选项
|
||
"""
|
||
|
||
import streamlit as st
|
||
import platform
|
||
from typing import Dict, Any
|
||
from loguru import logger
|
||
|
||
try:
|
||
from app.utils import ffmpeg_utils
|
||
from app.config.ffmpeg_config import FFmpegConfigManager
|
||
except ImportError as e:
|
||
logger.error(f"导入模块失败: {e}")
|
||
ffmpeg_utils = None
|
||
FFmpegConfigManager = None
|
||
|
||
|
||
def show_ffmpeg_diagnostics():
|
||
"""显示 FFmpeg 诊断信息"""
|
||
st.subheader("🔧 FFmpeg 兼容性诊断")
|
||
|
||
if ffmpeg_utils is None or FFmpegConfigManager is None:
|
||
st.error("❌ 无法加载 FFmpeg 工具模块")
|
||
return
|
||
|
||
# 基础信息
|
||
col1, col2 = st.columns(2)
|
||
|
||
with col1:
|
||
st.write("**系统信息**")
|
||
st.write(f"- 操作系统: {platform.system()} {platform.release()}")
|
||
st.write(f"- 架构: {platform.machine()}")
|
||
st.write(f"- Python: {platform.python_version()}")
|
||
|
||
with col2:
|
||
st.write("**FFmpeg 状态**")
|
||
|
||
# 检查 FFmpeg 安装
|
||
if ffmpeg_utils.check_ffmpeg_installation():
|
||
st.success("✅ FFmpeg 已安装")
|
||
else:
|
||
st.error("❌ FFmpeg 未安装或不在 PATH 中")
|
||
st.info("请安装 FFmpeg 并确保其在系统 PATH 中")
|
||
return
|
||
|
||
# 硬件加速信息
|
||
st.write("**硬件加速检测**")
|
||
|
||
try:
|
||
hwaccel_info = ffmpeg_utils.get_ffmpeg_hwaccel_info()
|
||
|
||
if hwaccel_info.get("available", False):
|
||
st.success(f"✅ {hwaccel_info.get('message', '硬件加速可用')}")
|
||
|
||
# 显示详细信息
|
||
with st.expander("硬件加速详情"):
|
||
st.write(f"- 加速类型: {hwaccel_info.get('type', '未知')}")
|
||
st.write(f"- 编码器: {hwaccel_info.get('encoder', '未知')}")
|
||
st.write(f"- GPU 厂商: {hwaccel_info.get('gpu_vendor', '未知')}")
|
||
st.write(f"- 独立显卡: {'是' if hwaccel_info.get('is_dedicated_gpu', False) else '否'}")
|
||
|
||
if hwaccel_info.get("tested_methods"):
|
||
st.write(f"- 测试的方法: {', '.join(hwaccel_info['tested_methods'])}")
|
||
else:
|
||
st.warning(f"⚠️ {hwaccel_info.get('message', '硬件加速不可用')}")
|
||
|
||
except Exception as e:
|
||
st.error(f"❌ 硬件加速检测失败: {str(e)}")
|
||
|
||
# 配置文件推荐
|
||
st.write("**推荐配置**")
|
||
|
||
try:
|
||
recommended_profile = FFmpegConfigManager.get_recommended_profile()
|
||
profile = FFmpegConfigManager.get_profile(recommended_profile)
|
||
|
||
st.info(f"🎯 推荐配置: **{profile.description}**")
|
||
|
||
# 显示配置详情
|
||
with st.expander("配置详情"):
|
||
st.write(f"- 配置名称: {profile.name}")
|
||
st.write(f"- 硬件加速: {'启用' if profile.hwaccel_enabled else '禁用'}")
|
||
st.write(f"- 编码器: {profile.encoder}")
|
||
st.write(f"- 质量预设: {profile.quality_preset}")
|
||
st.write(f"- 兼容性等级: {profile.compatibility_level}/5")
|
||
|
||
except Exception as e:
|
||
st.error(f"❌ 配置推荐失败: {str(e)}")
|
||
|
||
# 兼容性报告
|
||
if st.button("🔍 生成详细兼容性报告"):
|
||
try:
|
||
report = FFmpegConfigManager.get_compatibility_report()
|
||
|
||
st.write("**详细兼容性报告**")
|
||
st.json(report)
|
||
|
||
# 显示建议
|
||
if report.get("suggestions"):
|
||
st.write("**优化建议**")
|
||
for suggestion in report["suggestions"]:
|
||
st.write(f"- {suggestion}")
|
||
|
||
except Exception as e:
|
||
st.error(f"❌ 生成报告失败: {str(e)}")
|
||
|
||
|
||
def show_ffmpeg_settings():
|
||
"""显示 FFmpeg 设置选项"""
|
||
st.subheader("⚙️ FFmpeg 设置")
|
||
|
||
if FFmpegConfigManager is None:
|
||
st.error("❌ 无法加载配置管理器")
|
||
return
|
||
|
||
# 配置文件选择
|
||
profiles = FFmpegConfigManager.list_profiles()
|
||
|
||
# 获取当前推荐配置
|
||
try:
|
||
recommended_profile = FFmpegConfigManager.get_recommended_profile()
|
||
except Exception:
|
||
recommended_profile = "universal_software"
|
||
|
||
# 配置文件选择器
|
||
selected_profile = st.selectbox(
|
||
"选择 FFmpeg 配置文件",
|
||
options=list(profiles.keys()),
|
||
index=list(profiles.keys()).index(recommended_profile) if recommended_profile in profiles else 0,
|
||
format_func=lambda x: f"{profiles[x]} {'(推荐)' if x == recommended_profile else ''}",
|
||
help="不同的配置文件针对不同的硬件和兼容性需求进行了优化"
|
||
)
|
||
|
||
# 显示选中配置的详情
|
||
if selected_profile:
|
||
profile = FFmpegConfigManager.get_profile(selected_profile)
|
||
|
||
st.write("**配置详情**")
|
||
col1, col2 = st.columns(2)
|
||
|
||
with col1:
|
||
st.write(f"- 硬件加速: {'✅ 启用' if profile.hwaccel_enabled else '❌ 禁用'}")
|
||
st.write(f"- 编码器: {profile.encoder}")
|
||
st.write(f"- 质量预设: {profile.quality_preset}")
|
||
|
||
with col2:
|
||
st.write(f"- 像素格式: {profile.pixel_format}")
|
||
st.write(f"- 兼容性等级: {profile.compatibility_level}/5")
|
||
if profile.additional_args:
|
||
st.write(f"- 额外参数: {' '.join(profile.additional_args)}")
|
||
|
||
# 高级设置
|
||
with st.expander("🔧 高级设置"):
|
||
st.write("**强制设置选项**")
|
||
|
||
col1, col2 = st.columns(2)
|
||
|
||
with col1:
|
||
if st.button("🚫 强制禁用硬件加速"):
|
||
try:
|
||
ffmpeg_utils.force_software_encoding()
|
||
st.success("✅ 已强制禁用硬件加速")
|
||
st.info("这将使用纯软件编码，兼容性最高但性能较低")
|
||
except Exception as e:
|
||
st.error(f"❌ 操作失败: {str(e)}")
|
||
|
||
with col2:
|
||
if st.button("🔄 重置硬件加速检测"):
|
||
try:
|
||
ffmpeg_utils.reset_hwaccel_detection()
|
||
st.success("✅ 已重置硬件加速检测")
|
||
st.info("下次使用时将重新检测硬件加速能力")
|
||
except Exception as e:
|
||
st.error(f"❌ 操作失败: {str(e)}")
|
||
|
||
# 测试按钮
|
||
st.write("**测试功能**")
|
||
|
||
if st.button("🧪 测试 FFmpeg 兼容性"):
|
||
with st.spinner("正在测试 FFmpeg 兼容性..."):
|
||
try:
|
||
# 这里可以调用测试脚本
|
||
st.info("请在终端运行 `python test_video_extraction.py <video_path>` 进行完整测试")
|
||
|
||
# 简单的兼容性测试
|
||
if ffmpeg_utils and ffmpeg_utils.check_ffmpeg_installation():
|
||
hwaccel_info = ffmpeg_utils.get_ffmpeg_hwaccel_info()
|
||
if hwaccel_info.get("available"):
|
||
st.success("✅ 基础兼容性测试通过")
|
||
else:
|
||
st.warning("⚠️ 硬件加速不可用，但软件编码应该可以工作")
|
||
else:
|
||
st.error("❌ FFmpeg 不可用")
|
||
|
||
except Exception as e:
|
||
st.error(f"❌ 测试失败: {str(e)}")
|
||
|
||
|
||
def show_troubleshooting_guide():
|
||
"""显示故障排除指南"""
|
||
st.subheader("🆘 故障排除指南")
|
||
|
||
# 常见问题
|
||
st.write("**常见问题及解决方案**")
|
||
|
||
with st.expander("❌ 关键帧提取失败 - 滤镜链错误"):
|
||
st.write("""
|
||
**问题描述**: 出现 "Impossible to convert between the formats" 错误
|
||
|
||
**解决方案**:
|
||
1. 在设置中选择 "兼容性配置" 或 "Windows NVIDIA 优化配置"
|
||
2. 点击 "强制禁用硬件加速" 按钮
|
||
3. 重新尝试关键帧提取
|
||
4. 如果仍然失败，请更新显卡驱动程序
|
||
""")
|
||
|
||
with st.expander("⚠️ 硬件加速不可用"):
|
||
st.write("""
|
||
**可能原因**:
|
||
- 显卡驱动程序过旧
|
||
- FFmpeg 版本不支持当前硬件
|
||
- 系统缺少必要的运行库
|
||
|
||
**解决方案**:
|
||
1. 更新显卡驱动程序到最新版本
|
||
2. 对于 NVIDIA 用户，安装 CUDA 工具包
|
||
3. 对于 AMD 用户，安装 AMD Media SDK
|
||
4. 使用软件编码作为备用方案
|
||
""")
|
||
|
||
with st.expander("🐌 处理速度很慢"):
|
||
st.write("""
|
||
**优化建议**:
|
||
1. 启用硬件加速（如果可用）
|
||
2. 选择 "高性能配置"
|
||
3. 降低视频质量设置
|
||
4. 增加关键帧提取间隔
|
||
5. 关闭其他占用 GPU 的程序
|
||
""")
|
||
|
||
with st.expander("📁 文件权限问题"):
|
||
st.write("""
|
||
**解决方案**:
|
||
1. 确保对输出目录有写入权限
|
||
2. 以管理员身份运行程序（Windows）
|
||
3. 检查磁盘空间是否充足
|
||
4. 避免使用包含特殊字符的文件路径
|
||
""")
|
||
|
||
# 联系支持
|
||
st.write("**需要更多帮助？**")
|
||
st.info("""
|
||
如果上述解决方案都无法解决您的问题，请：
|
||
1. 运行 `python test_video_extraction.py` 生成详细的测试报告
|
||
2. 记录具体的错误信息和系统环境
|
||
3. 联系技术支持并提供相关信息
|
||
""")
|
||
|
||
|
||
def render_ffmpeg_diagnostics_page():
|
||
"""渲染 FFmpeg 诊断页面"""
|
||
st.title("🔧 FFmpeg 诊断与配置")
|
||
|
||
# 选项卡
|
||
tab1, tab2, tab3 = st.tabs(["🔍 诊断信息", "⚙️ 配置设置", "🆘 故障排除"])
|
||
|
||
with tab1:
|
||
show_ffmpeg_diagnostics()
|
||
|
||
with tab2:
|
||
show_ffmpeg_settings()
|
||
|
||
with tab3:
|
||
show_troubleshooting_guide()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
render_ffmpeg_diagnostics_page()
|