Update app.py

app.py

@@ -9,11 +9,52 @@ from datetime import datetime
 import json
 import concurrent.futures
 import time
+import platform
 
 # 储存目录设置
 STORAGE_DIR = os.path.expanduser("~/video_storage")
 STORAGE_CONFIG = os.path.join(STORAGE_DIR, "storage_config.json")
 
+# 检测系统硬件加速支持
+def detect_hardware_acceleration():
+    """检测可用的硬件加速"""
+    hardware_accel = {
+        'nvidia': False,
+        'amd': False,
+        'intel': False,
+        'encoder': 'libx264'  # 默认使用软件编码
+    }
+    
+    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硬件加速支持")
+        
+        # 检查AMD GPU
+        if 'h264_amf' in result.stdout:
+            hardware_accel['amd'] = True
+            if hardware_accel['encoder'] == 'libx264':
+                hardware_accel['encoder'] = 'h264_amf'
+            print("✅ 检测到AMD GPU硬件加速支持")
+        
+        # 检查Intel Quick Sync
+        if 'h264_qsv' in result.stdout:
+            hardware_accel['intel'] = True
+            if hardware_accel['encoder'] == 'libx264':
+                hardware_accel['encoder'] = 'h264_qsv'
+            print("✅ 检测到Intel Quick Sync硬件加速支持")
+            
+    except Exception as e:
+        print(f"⚠️ 硬件加速检测失败: {e}")
+    
+    return hardware_accel
+
+# 全局硬件加速信息
+HARDWARE_ACCEL = detect_hardware_acceleration()
+
 def init_storage():
     """初始化储存空间"""
     os.makedirs(STORAGE_DIR, exist_ok=True)
@@ -41,7 +82,6 @@ def save_to_storage(file_path, metadata=None):
             count += 1
         
         shutil.copy2(file_path, target_path)
-        # update_storage_config()  # 移除这行，在批量操作后统一更新
         return target_path
     except Exception as e:
         print(f"❌ 储存文件失败: {e}")
@@ -112,12 +152,10 @@ def download_all_storage():
             
             # 添加清单
             manifest = f"""# 储存空间视频清单
-
 ## 📊 下载信息
 - 下载时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
 - 视频总数: {len(video_files)} 个
 - 总大小: {config['total_size_mb']}MB
-
 ## 📁 文件列表
 """
             for video in video_files:
@@ -126,7 +164,6 @@ def download_all_storage():
             manifest += """
 ## 💡 使用说明
 这些是您的储存空间中保存的所有混剪视频，可直接使用或进一步编辑。
-
 ---
 FFmpeg 储存管理系统
 """
@@ -166,12 +203,10 @@ def download_selected_storage(selected_files):
             
             # 添加清单
             manifest = f"""# 选中视频下载清单
-
 ## 📊 下载信息
 - 下载时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
 - 选中文件: {len(valid_files)} 个
 - 总大小: {round(total_size, 1)}MB
-
 ## 📁 文件列表
 """
             for video in valid_files:
@@ -213,27 +248,133 @@ def clear_storage():
     except Exception as e:
         return f"❌ 清空失败: {str(e)}"
 
+def get_optimal_threads():
+    """获取最优线程数"""
+    try:
+        cpu_count = os.cpu_count()
+        if cpu_count:
+            # 对于视频处理，使用CPU核心数的1.5倍通常效果较好
+            return max(1, min(cpu_count * 2, 16))  # 最多16个线程
+        return 4
+    except:
+        return 4
+
+def build_ffmpeg_command(input_path, start_time, duration, output_path, operation='cut'):
+    """构建优化的FFmpeg命令"""
+    threads = get_optimal_threads()
+    encoder = HARDWARE_ACCEL['encoder']
+    
+    base_command = ['ffmpeg', '-threads', str(threads)]
+    
+    # 硬件加速解码
+    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.extend(['-i', input_path])
+    
+    if operation == 'cut':
+        base_command.extend(['-ss', str(start_time), '-t', str(duration)])
+        
+        # 编码参数
+        if encoder == 'h264_nvenc':
+            # NVIDIA GPU编码参数
+            base_command.extend([
+                '-c:v', 'h264_nvenc',
+                '-preset', 'p1',  # 最快预设
+                '-tune', 'ull',   # 超低延迟
+                '-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
+            ])
+        else:
+            # 软件编码参数 - 使用ultrafast预设
+            base_command.extend([
+                '-c:v', 'libx264',
+                '-preset', 'ultrafast',  # 最快预设
+                '-crf', '28',            # 质量参数
+                '-tune', 'fastdecode',   # 优化解码速度
+                '-x264opts', 'no-scenecut',  # 禁用场景检测以加快速度
+                '-c:a', 'aac',
+                '-b:a', '128k',
+                '-movflags', '+faststart',
+                '-avoid_negative_ts', 'make_zero',
+                '-y', output_path
+            ])
+    
+    elif operation == 'resize':
+        # 调整尺寸的编码参数
+        if encoder == 'h264_nvenc':
+            base_command.extend([
+                '-c:v', 'h264_nvenc',
+                '-preset', 'p1',
+                '-tune', 'ull',
+                '-cq', '28',
+                '-c:a', 'copy',
+                '-movflags', '+faststart',
+                '-y', output_path
+            ])
+        else:
+            base_command.extend([
+                '-c:v', 'libx264',
+                '-preset', 'ultrafast',
+                '-crf', '28',
+                '-tune', 'fastdecode',
+                '-x264opts', 'no-scenecut',
+                '-c:a', 'copy',
+                '-movflags', '+faststart',
+                '-y', output_path
+            ])
+    
+    return base_command
+
 def ffmpeg_cut_video(input_path, start_time, duration, output_path):
-    """[优化] 快速的视频切割，使用 superfast 预设"""
-    command = [
-        'ffmpeg', '-i', input_path, '-ss', str(start_time), '-t', str(duration),
-        '-c:v', 'libx264', '-preset', 'superfast', '-crf', '28',
-        '-c:a', 'aac', '-b:a', '128k', '-avoid_negative_ts', 'make_zero', '-y', output_path
-    ]
+    """[优化] 快速的视频切割，使用硬件加速和ultrafast预设"""
+    command = build_ffmpeg_command(input_path, start_time, duration, output_path, 'cut')
     result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     return result.returncode == 0 and os.path.exists(output_path)
 
 def ffmpeg_resize_video(input_path, output_path, target_ratio):
-    """[优化] 快速的比例调整，使用 superfast 预设"""
+    """[优化] 快速的比例调整，使用硬件加速和ultrafast预设"""
     if target_ratio == '9:16':
         filter_complex = "scale=1080:1920:force_original_aspect_ratio=decrease,pad=1080:1920:(ow-iw)/2:(oh-ih)/2:black"
     else:
         filter_complex = "scale=1920:1080:force_original_aspect_ratio=decrease,pad=1920:1080:(ow-iw)/2:(oh-ih)/2:black"
     
-    command = [
-        'ffmpeg', '-i', input_path, '-vf', filter_complex,
-        '-c:v', 'libx264', '-preset', 'superfast', '-crf', '28', '-c:a', 'copy', '-y', output_path
-    ]
+    command = build_ffmpeg_command(input_path, 0, 0, output_path, 'resize')
+    # 在resize操作中插入滤镜
+    command.insert(-5, '-vf')  # 在-y之前插入
+    command.insert(-4, filter_complex)
+    
     result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     return result.returncode == 0 and os.path.exists(output_path)
 
@@ -253,7 +394,9 @@ def concat_videos(file_list, output_path):
             list_file.write(f"file '{abs_path}'\n")
         list_file.close()
         
-        command = ['ffmpeg', '-f', 'concat', '-safe', '0', '-i', list_file.name, '-c', 'copy', '-y', output_path]
+        # 使用多线程进行合并
+        threads = get_optimal_threads()
+        command = ['ffmpeg', '-threads', str(threads), '-f', 'concat', '-safe', '0', '-i', list_file.name, '-c', 'copy', '-y', output_path]
         result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         return result.returncode == 0 and os.path.exists(output_path)
     finally:
@@ -293,22 +436,24 @@ def process_single_video(video_file, clip_duration, temp_dir):
     return clips
 
 def process_videos_with_storage(video_files, clip_duration, num_output_videos, target_ratio):
-    """带储存功能的视频处理 [优化版：并行切割 + 批量更新 + 进度日志]"""
+    """带储存功能的视频处理 [优化版：硬件加速 + 并行切割 + 批量更新 + 进度日志]"""
     if not video_files:
         return "❌ 请上传视频文件", None, "", ""
     
-    start_time = time.time()  # ✅ 记录开始时间
+    start_time = time.time()
     temp_dir = tempfile.mkdtemp()
     
     try:
         print(f"🔍 开始处理 {len(video_files)} 个视频文件...")
+        print(f"⚡ 硬件加速: {HARDWARE_ACCEL['encoder']}")
+        print(f"🔧 线程数: {get_optimal_threads()}")
         print(f"⏱️ 切片时长: {clip_duration} 秒 | 生成数量: {num_output_videos} 个 | 比例: {target_ratio}")
 
         all_clips = []
 
-        # --- [优化] 使用线程池并行处理每个视频文件 ---
-        print(f"🚀 启动并行切片（{min(4, os.cpu_count() or 1)} 线程）...")
-        with concurrent.futures.ThreadPoolExecutor(max_workers=min(4, os.cpu_count() or 1)) as executor:
+        # 使用进程池进行并行处理（避免GIL限制）
+        print(f"🚀 启动并行切片（{min(8, os.cpu_count() or 1)} 进程）...")
+        with concurrent.futures.ProcessPoolExecutor(max_workers=min(8, os.cpu_count() or 1)) as executor:
             future_to_video = {
                 executor.submit(process_single_video, vf, clip_duration, temp_dir): vf 
                 for vf in video_files
@@ -326,7 +471,6 @@ def process_videos_with_storage(video_files, clip_duration, num_output_videos, t
                     video_file = future_to_video[future]
                     print(f"❌ 切片失败: {os.path.basename(video_file.name)} - {exc}")
 
-        # --- 并行处理结束 ---
         print(f"📊 所有视频切片完成，共生成 {len(all_clips)} 个片段")
 
         if not all_clips:
@@ -368,7 +512,6 @@ def process_videos_with_storage(video_files, clip_duration, num_output_videos, t
             else:
                 print(f"❌ 比例调整失败: 第 {i+1} 个视频")
 
-        # --- [优化] 所有文件保存后，统一更新一次配置 ---
         if stored_files:
             update_storage_config()
             print(f"✅ 已统一更新储存配置，共保存 {len(stored_files)} 个视频文件")
@@ -376,7 +519,6 @@ def process_videos_with_storage(video_files, clip_duration, num_output_videos, t
         if not output_files:
             return "❌ 生成混剪视频失败", None, "", ""
 
-        # --- 打包下载 ---
         package_dir = tempfile.mkdtemp()
         timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
         zip_path = os.path.join(package_dir, f"混剪视频包_{target_ratio.replace(':', 'x')}_{timestamp}.zip")
@@ -389,13 +531,12 @@ def process_videos_with_storage(video_files, clip_duration, num_output_videos, t
                 print(f"    ➤ 已添加: {arcname}")
             
             readme = f"""# 混剪视频包
-
 ## 📊 生成信息
 - 生成时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
 - 视频数量: {len(output_files)} 个
 - 视频比例: {target_ratio}
 - 切片时长: {clip_duration} 秒
-
+- 硬件加速: {HARDWARE_ACCEL['encoder']}
 ## 📁 文件列表
 """
             for i, vf in enumerate(output_files, 1):
@@ -406,7 +547,6 @@ def process_videos_with_storage(video_files, clip_duration, num_output_videos, t
 ## 💾 储存信息
 所有视频已自动保存到本地储存空间:
 {', '.join(stored_files)}
-
 视频已按 {target_ratio} 比例优化，可直接发布。
 """
             zipf.writestr("README.txt", readme.encode('utf-8'))
@@ -414,31 +554,30 @@ def process_videos_with_storage(video_files, clip_duration, num_output_videos, t
         total_size = os.path.getsize(zip_path) / (1024 * 1024)
         platform_info = "📱 短视频平台" if target_ratio == '9:16' else "🖥️ 长视频平台"
         
-        end_time = time.time()  # ✅ 计算耗时
+        end_time = time.time()
         elapsed = end_time - start_time
         print(f"🎉 混剪完成！总耗时: {elapsed:.1f} 秒")
         print(f"📥 下载包大小: {total_size:.1f}MB")
         print(f"📁 已保存文件: {len(stored_files)} 个")
 
         success_msg = f"""✅ 混剪完成并已储存！
-
 📊 **生成统计:**
 • 🎬 混剪视频: {len(output_files)} 个
 • 📐 视频比例: {target_ratio}
 • 🎯 适合平台: {platform_info}
 • 📦 下载包大小: {total_size:.1f}MB
-
+• ⚡ 硬件加速: {HARDWARE_ACCEL['encoder']}
+• 🔧 处理线程: {get_optimal_threads()}
+• ⏱️ 处理耗时: {elapsed:.1f} 秒
 💾 **自动储存:**
 • 📁 储存位置: ~/video_storage/
 • 🔥 已保存文件: {len(stored_files)} 个
 • ✅ 永久保存，不会丢失
-
 ⬇️ **立即下载:**
 点击下方按钮下载打包文件
 """
         
         details = f"""🎬 **视频详情:**
-
 """
         for i, vf in enumerate(output_files, 1):
             size_mb = os.path.getsize(vf) / (1024 * 1024)
@@ -452,11 +591,9 @@ def process_videos_with_storage(video_files, clip_duration, num_output_videos, t
         
         config, video_list = get_storage_info()
         storage_info = f"""📊 **储存空间状态:**
-
 💾 总计: {config['total_videos']} 个视频
 📦 总大小: {config['total_size_mb']}MB
 📁 位置: ~/video_storage/
-
 📋 **最新文件:**
 """
         
@@ -478,12 +615,11 @@ def refresh_storage_display():
     config, video_list = get_storage_info()
     
     storage_display = f"""💾 **储存空间概览**
-
 📊 **统计信息:**
 • 总视频数量: {config['total_videos']} 个
 • 总占用空间: {config['total_size_mb']}MB
 • 储存位置: ~/video_storage/
-
+• 硬件加速: {HARDWARE_ACCEL['encoder']}
 📁 **文件列表:**
 """
     
@@ -529,10 +665,11 @@ init_storage()
 def main():
     with gr.Blocks(title="FFmpeg混剪+储存+下载管理", theme=gr.themes.Soft()) as demo:
         
-        gr.HTML("""
+        gr.HTML(f"""
         <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>
         </div>
         """)
         
@@ -652,7 +789,7 @@ def main():
             outputs=[storage_display, file_selector, file_selector_download]
         )
         
-        gr.Markdown("""
+        gr.Markdown(f"""
         ---
         ### 📖 功能说明
         
@@ -661,6 +798,8 @@ def main():
         - 📐 支持9:16/16:9比例调整
         - 📦 打包下载所有生成视频
         - 💾 **自动储存到本地目录**
+        - 🚀 **硬件加速: {HARDWARE_ACCEL['encoder']}**
+        - 🔧 **多线程处理: {get_optimal_threads()} 线程**
         
         **💾 储存管理功能:**
         - 📁 所有生成视频自动保存到 `~/video_storage/`
@@ -685,6 +824,12 @@ def main():
         - **分享协作**: 打包分享给团队成员
         - **存档管理**: 定期下载备份到云盘
         
+        **⚡ 性能优化:**
+        - **硬件加速**: 自动检测并使用GPU加速
+        - **多线程处理**: 充分利用多核CPU
+        - **快速预设**: 使用ultrafast预设提升速度
+        - **并行处理**: 多个视频同时处理
+        
         **⚠️ 注意事项:**
         - 下载文件为ZIP格式，需要解压使用
         - 一键下载包含储存空间中所有视频文件