Update app.py

app.py

@@ -22,34 +22,64 @@ def detect_hardware_acceleration():
         'nvidia': False,
         'amd': False,
         'intel': False,
-        'encoder': 'libx264'  # 默认使用软件编码
+        'encoder': 'libx264',  # 默认使用软件编码
+        'working': False      # 标记硬件加速是否实际可用
     }
     
     try:
         # 检查NVIDIA GPU
         result = subprocess.run(['ffmpeg', '-encoders'], capture_output=True, text=True)
         if 'h264_nvenc' in result.stdout:
-            hardware_accel['nvidia'] = True
-            hardware_accel['encoder'] = 'h264_nvenc'
-            print("✅ 检测到NVIDIA GPU硬件加速支持")
+            # 进一步测试NVIDIA编码器是否实际可用
+            test_result = subprocess.run([
+                'ffmpeg', '-f', 'lavfi', '-i', 'testsrc=duration=1:size=32x32:rate=1', 
+                '-c:v', 'h264_nvenc', '-f', 'null', '-'
+            ], capture_output=True, text=True)
+            
+            if test_result.returncode == 0:
+                hardware_accel['nvidia'] = True
+                hardware_accel['encoder'] = 'h264_nvenc'
+                hardware_accel['working'] = True
+                print("✅ 检测到NVIDIA GPU硬件加速支持且工作正常")
+            else:
+                print("⚠️ 检测到NVIDIA GPU但无法使用，将回退到软件编码")
         
         # 检查AMD GPU
-        if 'h264_amf' in result.stdout:
-            hardware_accel['amd'] = True
-            if hardware_accel['encoder'] == 'libx264':
+        if not hardware_accel['working'] and 'h264_amf' in result.stdout:
+            test_result = subprocess.run([
+                'ffmpeg', '-f', 'lavfi', '-i', 'testsrc=duration=1:size=32x32:rate=1', 
+                '-c:v', 'h264_amf', '-f', 'null', '-'
+            ], capture_output=True, text=True)
+            
+            if test_result.returncode == 0:
+                hardware_accel['amd'] = True
                 hardware_accel['encoder'] = 'h264_amf'
-            print("✅ 检测到AMD GPU硬件加速支持")
+                hardware_accel['working'] = True
+                print("✅ 检测到AMD GPU硬件加速支持且工作正常")
+            else:
+                print("⚠️ 检测到AMD GPU但无法使用，将回退到软件编码")
         
         # 检查Intel Quick Sync
-        if 'h264_qsv' in result.stdout:
-            hardware_accel['intel'] = True
-            if hardware_accel['encoder'] == 'libx264':
+        if not hardware_accel['working'] and 'h264_qsv' in result.stdout:
+            test_result = subprocess.run([
+                'ffmpeg', '-f', 'lavfi', '-i', 'testsrc=duration=1:size=32x32:rate=1', 
+                '-c:v', 'h264_qsv', '-f', 'null', '-'
+            ], capture_output=True, text=True)
+            
+            if test_result.returncode == 0:
+                hardware_accel['intel'] = True
                 hardware_accel['encoder'] = 'h264_qsv'
-            print("✅ 检测到Intel Quick Sync硬件加速支持")
+                hardware_accel['working'] = True
+                print("✅ 检测到Intel Quick Sync硬件加速支持且工作正常")
+            else:
+                print("⚠️ 检测到Intel Quick Sync但无法使用，将回退到软件编码")
             
     except Exception as e:
         print(f"⚠️ 硬件加速检测失败: {e}")
     
+    if not hardware_accel['working']:
+        print("📌 将使用软件编码 (libx264)")
+    
     return hardware_accel
 
 # 全局硬件加速信息
@@ -268,15 +298,7 @@ def build_ffmpeg_command(input_path, start_time, duration, output_path, operatio
     input_path = os.path.abspath(input_path)
     output_path = os.path.abspath(output_path)
     
-    base_command = ['ffmpeg']
-    
-    # 硬件加速解码 - 简化处理，避免兼容性问题
-    # if HARDWARE_ACCEL['nvidia']:
-    #     base_command.extend(['-hwaccel', 'cuda'])
-    # elif HARDWARE_ACCEL['amd']:
-    #     base_command.extend(['-hwaccel', 'amf'])
-    # elif HARDWARE_ACCEL['intel']:
-    #     base_command.extend(['-hwaccel', 'qsv'])
+    base_command = ['ffmpeg', '-threads', str(threads)]
     
     base_command.extend(['-i', input_path])
     
@@ -284,41 +306,40 @@ def build_ffmpeg_command(input_path, start_time, duration, output_path, operatio
         base_command.extend(['-ss', str(start_time), '-t', str(duration)])
         
         # 编码参数
-        if encoder == 'h264_nvenc':
-            # NVIDIA GPU编码参数 - 简化版本
-            base_command.extend([
-                '-c:v', 'h264_nvenc',
-                '-preset', 'fast',  # 使用fast而不是p1，更兼容
-                '-cq', '28',        # 质量参数
-                '-c:a', 'aac',
-                '-b:a', '128k',
-                '-movflags', '+faststart',
-                '-y', output_path
-            ])
-        elif encoder == 'h264_amf':
-            # AMD GPU编码参数 - 简化版本
-            base_command.extend([
-                '-c:v', 'h264_amf',
-                '-quality', 'speed',
-                '-rc', 'cqp',
-                '-cqp_i', '28',
-                '-cqp_p', '28',
-                '-c:a', 'aac',
-                '-b:a', '128k',
-                '-movflags', '+faststart',
-                '-y', output_path
-            ])
-        elif encoder == 'h264_qsv':
-            # Intel Quick Sync编码参数 - 简化版本
-            base_command.extend([
-                '-c:v', 'h264_qsv',
-                '-preset', 'veryfast',
-                '-global_quality', '28',
-                '-c:a', 'aac',
-                '-b:a', '128k',
-                '-movflags', '+faststart',
-                '-y', output_path
-            ])
+        if HARDWARE_ACCEL['working']:
+            # 硬件编码参数
+            if encoder == 'h264_nvenc':
+                base_command.extend([
+                    '-c:v', 'h264_nvenc',
+                    '-preset', 'fast',
+                    '-cq', '28',
+                    '-c:a', 'aac',
+                    '-b:a', '128k',
+                    '-movflags', '+faststart',
+                    '-y', output_path
+                ])
+            elif encoder == 'h264_amf':
+                base_command.extend([
+                    '-c:v', 'h264_amf',
+                    '-quality', 'speed',
+                    '-rc', 'cqp',
+                    '-cqp_i', '28',
+                    '-cqp_p', '28',
+                    '-c:a', 'aac',
+                    '-b:a', '128k',
+                    '-movflags', '+faststart',
+                    '-y', output_path
+                ])
+            elif encoder == 'h264_qsv':
+                base_command.extend([
+                    '-c:v', 'h264_qsv',
+                    '-preset', 'veryfast',
+                    '-global_quality', '28',
+                    '-c:a', 'aac',
+                    '-b:a', '128k',
+                    '-movflags', '+faststart',
+                    '-y', output_path
+                ])
         else:
             # 软件编码参数 - 使用ultrafast预设
             base_command.extend([
@@ -335,16 +356,39 @@ def build_ffmpeg_command(input_path, start_time, duration, output_path, operatio
     
     elif operation == 'resize':
         # 调整尺寸的编码参数
-        if encoder == 'h264_nvenc':
-            base_command.extend([
-                '-c:v', 'h264_nvenc',
-                '-preset', 'fast',
-                '-cq', '28',
-                '-c:a', 'copy',
-                '-movflags', '+faststart',
-                '-y', output_path
-            ])
+        if HARDWARE_ACCEL['working']:
+            # 硬件编码参数
+            if encoder == 'h264_nvenc':
+                base_command.extend([
+                    '-c:v', 'h264_nvenc',
+                    '-preset', 'fast',
+                    '-cq', '28',
+                    '-c:a', 'copy',
+                    '-movflags', '+faststart',
+                    '-y', output_path
+                ])
+            elif encoder == 'h264_amf':
+                base_command.extend([
+                    '-c:v', 'h264_amf',
+                    '-quality', 'speed',
+                    '-rc', 'cqp',
+                    '-cqp_i', '28',
+                    '-cqp_p', '28',
+                    '-c:a', 'copy',
+                    '-movflags', '+faststart',
+                    '-y', output_path
+                ])
+            elif encoder == 'h264_qsv':
+                base_command.extend([
+                    '-c:v', 'h264_qsv',
+                    '-preset', 'veryfast',
+                    '-global_quality', '28',
+                    '-c:a', 'copy',
+                    '-movflags', '+faststart',
+                    '-y', output_path
+                ])
         else:
+            # 软件编码参数
             base_command.extend([
                 '-c:v', 'libx264',
                 '-preset', 'ultrafast',
@@ -370,6 +414,22 @@ def ffmpeg_cut_video(input_path, start_time, duration, output_path):
         
         if result.returncode != 0:
             print(f"FFmpeg错误: {result.stderr}")  # 输出错误信息
+            
+            # 如果硬件加速失败，尝试使用软件编码
+            if HARDWARE_ACCEL['working'] and "Cannot load libcuda.so.1" in result.stderr:
+                print("⚠️ 硬件加速失败，回退到软件编码...")
+                HARDWARE_ACCEL['working'] = False
+                HARDWARE_ACCEL['encoder'] = 'libx264'
+                
+                # 重新构建命令，使用软件编码
+                command = build_ffmpeg_command(input_path, start_time, duration, output_path, 'cut')
+                print(f"重试FFmpeg命令(软件编码): {' '.join(command)}")
+                
+                result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+                if result.returncode != 0:
+                    print(f"FFmpeg软件编码错误: {result.stderr}")
+                    return False
+            
             return False
         
         return os.path.exists(output_path)
@@ -399,6 +459,24 @@ def ffmpeg_resize_video(input_path, output_path, target_ratio):
         
         if result.returncode != 0:
             print(f"FFmpeg调整错误: {result.stderr}")  # 输出错误信息
+            
+            # 如果硬件加速失败，尝试使用软件编码
+            if HARDWARE_ACCEL['working'] and "Cannot load libcuda.so.1" in result.stderr:
+                print("⚠️ 硬件加速失败，回退到软件编码...")
+                HARDWARE_ACCEL['working'] = False
+                HARDWARE_ACCEL['encoder'] = 'libx264'
+                
+                # 重新构建命令，使用软件编码
+                command = build_ffmpeg_command(input_path, 0, 0, output_path, 'resize')
+                command.insert(-5, '-vf')  # 在-y之前插入
+                command.insert(-4, filter_complex)
+                
+                print(f"重试FFmpeg调整命令(软件编码): {' '.join(command)}")
+                result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+                if result.returncode != 0:
+                    print(f"FFmpeg软件编码错误: {result.stderr}")
+                    return False
+            
             return False
         
         return os.path.exists(output_path)
@@ -488,7 +566,7 @@ def process_videos_with_storage(video_files, clip_duration, num_output_videos, t
     
     try:
         print(f"🔍 开始处理 {len(video_files)} 个视频文件...")
-        print(f"⚡ 硬件加速: {HARDWARE_ACCEL['encoder']}")
+        print(f"⚡ 硬件加速: {HARDWARE_ACCEL['encoder']} {'(工作正常)' if HARDWARE_ACCEL['working'] else '(不可用，使用软件编码)'}")
         print(f"🔧 线程数: {get_optimal_threads()}")
         print(f"⏱️ 切片时长: {clip_duration} 秒 | 生成数量: {num_output_videos} 个 | 比例: {target_ratio}")
 
@@ -579,7 +657,7 @@ def process_videos_with_storage(video_files, clip_duration, num_output_videos, t
 - 视频数量: {len(output_files)} 个
 - 视频比例: {target_ratio}
 - 切片时长: {clip_duration} 秒
-- 硬件加速: {HARDWARE_ACCEL['encoder']}
+- 硬件加速: {HARDWARE_ACCEL['encoder']} {'(工作正常)' if HARDWARE_ACCEL['working'] else '(不可用，使用软件编码)'}
 ## 📁 文件列表
 """
             for i, vf in enumerate(output_files, 1):
@@ -609,7 +687,7 @@ def process_videos_with_storage(video_files, clip_duration, num_output_videos, t
 • 📐 视频比例: {target_ratio}
 • 🎯 适合平台: {platform_info}
 • 📦 下载包大小: {total_size:.1f}MB
-• ⚡ 硬件加速: {HARDWARE_ACCEL['encoder']}
+• ⚡ 硬件加速: {HARDWARE_ACCEL['encoder']} {'(工作正常)' if HARDWARE_ACCEL['working'] else '(不可用，使用软件编码)'}
 • 🔧 处理线程: {get_optimal_threads()}
 • ⏱️ 处理耗时: {elapsed:.1f} 秒
 💾 **自动储存:**
@@ -662,7 +740,7 @@ def refresh_storage_display():
 • 总视频数量: {config['total_videos']} 个
 • 总占用空间: {config['total_size_mb']}MB
 • 储存位置: ~/video_storage/
-• 硬件加速: {HARDWARE_ACCEL['encoder']}
+• 硬件加速: {HARDWARE_ACCEL['encoder']} {'(工作正常)' if HARDWARE_ACCEL['working'] else '(不可用，使用软件编码)'}
 📁 **文件列表:**
 """
     
@@ -712,7 +790,7 @@ def main():
         <div style="text-align: center; padding: 20px; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); border-radius: 15px; color: white; margin-bottom: 20px;">
             <h1>🎬 FFmpeg 混剪工具 + 储存管理 + 一键下载</h1>
             <p style="margin: 10px 0 0 0;">长视频切片 → 智能混剪 → 比例调整 → 自动储存 → 一键下载</p>
-            <p style="margin: 5px 0 0 0; font-size: 14px; opacity: 0.9;">⚡ 硬件加速: {HARDWARE_ACCEL['encoder']} | 🔧 线程数: {get_optimal_threads()}</p>
+            <p style="margin: 5px 0 0 0; font-size: 14px; opacity: 0.9;">⚡ 硬件加速: {HARDWARE_ACCEL['encoder']} {'(工作正常)' if HARDWARE_ACCEL['working'] else '(不可用，使用软件编码)'}</p>
         </div>
         """)
         
@@ -841,7 +919,7 @@ def main():
         - 📐 支持9:16/16:9比例调整
         - 📦 打包下载所有生成视频
         - 💾 **自动储存到本地目录**
-        - 🚀 **硬件加速: {HARDWARE_ACCEL['encoder']}**
+        - 🚀 **硬件加速: {HARDWARE_ACCEL['encoder']} {'(工作正常)' if HARDWARE_ACCEL['working'] else '(不可用，使用软件编码)'}**
         - 🔧 **多线程处理: {get_optimal_threads()} 线程**
         
         **💾 储存管理功能:**
@@ -869,6 +947,7 @@ def main():
         
         **⚡ 性能优化:**
         - **硬件加速**: 自动检测并使用GPU加速
+        - **自动降级**: 硬件加速失败时自动回退到软件编码
         - **多线程处理**: 充分利用多核CPU
         - **快速预设**: 使用ultrafast预设提升速度
         - **并行处理**: 多个视频同时处理