Update app.py

app.py

@@ -15,6 +15,30 @@ def ffmpeg_cut_video(input_path, start_time, duration, output_path):
     process = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     return process.returncode == 0
 
+def ffmpeg_resize_video(input_path, output_path, target_ratio):
+    """调整视频比例"""
+    if target_ratio == '9:16':
+        # 竖屏：高度保持，宽度=高度*9/16，居中裁剪
+        scale_filter = "scale=-2:1920,crop=1080:1920"
+    elif target_ratio == '16:9':
+        # 横屏：宽度保持，高度=宽度*9/16，居中裁剪  
+        scale_filter = "scale=1920:-2,crop=1920:1080"
+    else:
+        # 默认保持原比例
+        scale_filter = "scale=-2:-2"
+
+    command = [
+        'ffmpeg',
+        '-i', input_path,
+        '-vf', scale_filter,
+        '-c:a', 'copy',
+        '-y',
+        output_path
+    ]
+    
+    process = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    return process.returncode == 0
+
 def concat_videos_ffmpeg(file_list, output_path):
     list_file = tempfile.NamedTemporaryFile(delete=False, mode='w', suffix='.txt')
     try:
@@ -60,7 +84,7 @@ def extract_clips_from_videos(video_files, clip_duration):
         shutil.rmtree(temp_dir, ignore_errors=True)
         raise
 
-def generate_mixed_videos(clips, num_output_videos):
+def generate_mixed_videos(clips, num_output_videos, target_ratio):
     temp_dir = tempfile.mkdtemp()
     random.shuffle(clips)
     clips_per_video = max(1, len(clips) // num_output_videos)
@@ -72,25 +96,31 @@ def generate_mixed_videos(clips, num_output_videos):
             end_idx = len(clips) if i == num_output_videos - 1 else (start_idx + clips_per_video)
             selected_clips = clips[start_idx:end_idx]
             
-            # 为每个视频生成更有意义的文件名
-            timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
-            output_path = os.path.join(temp_dir, f"混剪视频_{i+1}_{timestamp}.mp4")
-            
-            ok = concat_videos_ffmpeg(selected_clips, output_path)
+            # 先合并片段
+            temp_video_path = os.path.join(temp_dir, f"temp_mixed_{i+1}.mp4")
+            ok = concat_videos_ffmpeg(selected_clips, temp_video_path)
             if not ok:
                 raise RuntimeError("视频合并失败")
+            
+            # 调整视频比例
+            timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
+            output_path = os.path.join(temp_dir, f"混剪视频_{target_ratio.replace(':', 'x')}_{i+1}_{timestamp}.mp4")
+            ok = ffmpeg_resize_video(temp_video_path, output_path, target_ratio)
+            if not ok:
+                raise RuntimeError("视频比例调整失败")
+            
             output_files.append(output_path)
+            
         return output_files, temp_dir
     except Exception:
         shutil.rmtree(temp_dir, ignore_errors=True)
         raise
 
-def create_video_package(output_files, original_file_names):
+def create_video_package(output_files, original_file_names, target_ratio):
     """将生成的视频文件打包成zip，并创建说明文件"""
     
-    # 创建临时目录用于打包
     package_dir = tempfile.mkdtemp()
-    zip_path = os.path.join(package_dir, f"混剪视频包_{datetime.now().strftime('%Y%m%d_%H%M%S')}.zip")
+    zip_path = os.path.join(package_dir, f"混剪视频包_{target_ratio.replace(':', 'x')}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.zip")
     
     try:
         with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
@@ -105,8 +135,26 @@ def create_video_package(output_files, original_file_names):
 ## 📊 生成信息
 - 生成时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
 - 视频数量: {len(output_files)} 个
+- 视频比例: {target_ratio}
 - 源文件: {', '.join(original_file_names)}
 
+## 📱 比例说明
+"""
+            
+            if target_ratio == '9:16':
+                readme_content += """- 🔥 9:16 竖屏格式
+- 📱 适合平台：抖音、快手、小红书、Instagram Stories
+- 🎯 最佳用途：短视频、Vlog、才艺展示
+- 📐 分辨率：1080x1920 (推荐)
+"""
+            else:
+                readme_content += """- 🖥️ 16:9 横屏格式  
+- 📺 适合平台：YouTube、B站、腾讯视频、爱奇艺
+- 🎯 最佳用途：长视频、教程、游戏、影视解说
+- 📐 分辨率：1920x1080 (推荐)
+"""
+            
+            readme_content += f"""
 ## 📁 文件列表
 """
             
@@ -115,18 +163,39 @@ def create_video_package(output_files, original_file_names):
                 readme_content += f"- 混剪视频_{i}.mp4 ({file_size:.1f}MB)\n"
             
             readme_content += f"""
-## 🎬 使用建议
-- 适合发布到抖音、快手、小红书等短视频平台
-- 建议添加背景音乐以提升观看体验
-- 可以进一步编辑添加字幕、特效等
+## 🎬 制作技术
+- 切片算法: FFmpeg 自动分割
+- 混剪逻辑: 智能随机重组
+- 比例调整: {target_ratio} 专业适配
+- 视频编码: H.264/AAC 高质���
 
-## ⚡ 技术信息
-- 制作工具: FFmpeg 自动剪辑
-- 处理方式: 智能切片 + 随机重组
-- 视频编码: H.264/AAC
+## 📱 发布建议
+"""
+            
+            if target_ratio == '9:16':
+                readme_content += """- 添加动感背景音乐
+- 考虑添加字幕提升观看体验
+- 开头3秒要有吸引力
+- 时长控制在15-60秒最佳
+- 发布时间：晚上7-9点黄金时段
+"""
+            else:
+                readme_content += """- 制作精美封面图
+- 添加详细标题和描述
+- 考虑制作系列内容
+- 时长可以3-15分钟
+- 注意SEO关键词优化
+"""
+            
+            readme_content += """
+## 💡 后续处理
+- 可用剪映、PR等软件进一步编辑
+- 建议添加个人水印
+- 支持再次调整分辨率
+- 可添加转场特效和滤镜
 
 ---
-感谢使用 FFmpeg 自动混剪工具！
+🎬 感谢使用 FFmpeg 自动混剪工具！
 """
             
             # 将说明文件添加到zip
@@ -138,7 +207,7 @@ def create_video_package(output_files, original_file_names):
         shutil.rmtree(package_dir, ignore_errors=True)
         raise e
 
-def process_and_package(video_files, clip_duration, num_output_videos):
+def process_and_package_with_ratio(video_files, clip_duration, num_output_videos, target_ratio):
     if not video_files or len(video_files) == 0:
         return "❌ 请上传至少一个视频文件", None, ""
     
@@ -148,27 +217,37 @@ def process_and_package(video_files, clip_duration, num_output_videos):
         
         # 1. 视频处理
         clips, clips_temp_dir = extract_clips_from_videos(video_files, clip_duration)
-        output_files, output_temp_dir = generate_mixed_videos(clips, num_output_videos)
+        output_files, output_temp_dir = generate_mixed_videos(clips, num_output_videos, target_ratio)
         
         # 清理切片临时目录
         shutil.rmtree(clips_temp_dir, ignore_errors=True)
         
         # 2. 打包所有视频
-        zip_file_path = create_video_package(output_files, original_names)
+        zip_file_path = create_video_package(output_files, original_names, target_ratio)
         
         # 计算总大小
         total_size = os.path.getsize(zip_file_path) / (1024 * 1024)  # MB
         
+        # 平台信息
+        platform_info = ""
+        if target_ratio == '9:16':
+            platform_info = "📱 抖音、快手、小红书等短视频平台"
+        else:
+            platform_info = "🖥️ YouTube、B站等长视频平台"
+        
         status_msg = f"""✅ 处理完成！
 
 📊 **生成统计:**
 • 混剪视频数量: {len(output_files)} 个
+• 视频比例: {target_ratio} 
+• 适合平台: {platform_info}
 • 打包文件大小: {total_size:.1f}MB
 • 处理时间: {datetime.now().strftime('%H:%M:%S')}
 
 📦 **打包内容:**
-• {len(output_files)} 个混剪视频文件
-• 1 个使用说明文档
+• {len(output_files)} 个 {target_ratio} 比例混剪视频
+• 1 个详细使用说明文档
+• 平台发布建议和技术参数
 • 所有文件已压缩打包
 
 💾 **下载说明:**
@@ -176,7 +255,7 @@ def process_and_package(video_files, clip_duration, num_output_videos):
 """
         
         # 生成详细信息
-        details = f"""🎬 **视频详情:**
+        details = f"""🎬 **视频详情 ({target_ratio}):**
 
 """
         for i, video_file in enumerate(output_files, 1):
@@ -185,24 +264,43 @@ def process_and_package(video_files, clip_duration, num_output_videos):
         
         details += f"""
 🔧 **技术参数:**
+• 视频比例: {target_ratio}
 • 切片时长: {clip_duration}秒
-• 随机混剪算法: 智能重组
+• 随机混剪: 智能重组算法
 • 视频编码: H.264
 • 音频编码: AAC
+"""
 
-📱 **发布建议:**
-• 适合短视频平台发布
-• 建议添加背景音乐
-• 可进一步编辑添加特效
-• 注意平台尺寸要求
+        if target_ratio == '9:16':
+            details += """
+📱 **竖屏优势:**
+• 符合移动端观看习惯
+• 抖音算法友好
+• 全屏沉浸式体验
+• 适合碎片化观看
 
-💡 **后续处理:**
-• 解压后即可使用
-• 支持主流播放器播放
-• 可用视频编辑软件进一步处理
+🎯 **发布建议:**
+• 开头3秒要抓眼球
+• 添加热门背景音乐
+• 考虑添加字幕
+• 时长15-60秒最佳
+"""
+        else:
+            details += """
+🖥️ **横屏优势:**
+• 传统影视观看体验
+• 适合详细内容展示
+• YouTube推荐算法友好
+• 便于添加复杂字幕
+
+🎯 **发布建议:**
+• 制作吸引人的封面
+• 详细标题和描述
+• 3-15分钟时长合适
+• SEO关键词优化
 """
         
-        # 清理视频临时目录（但保留zip文件）
+        # 清理视频临时目录
         shutil.rmtree(output_temp_dir, ignore_errors=True)
         
         return status_msg, zip_file_path, details
@@ -212,14 +310,14 @@ def process_and_package(video_files, clip_duration, num_output_videos):
 
 def main():
     with gr.Blocks(
-        title="FFmpeg自动剪辑+打包下载", 
+        title="FFmpeg自动剪辑+比例调整+打包下载", 
         theme=gr.themes.Soft()
     ) as demo:
         
         gr.HTML("""
-        <div style="text-align: center; padding: 20px; background: linear-gradient(90deg, #667eea 0%, #764ba2 100%); border-radius: 10px; color: white; margin-bottom: 20px;">
-            <h1>🎬 FFmpeg 自动剪辑 + 打包下载</h1>
-            <p style="margin: 10px 0 0 0;">长视频自动切片 → 智能混剪 → 一键打包下载</p>
+        <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>
         </div>
         """)
         
@@ -246,6 +344,14 @@ def main():
                         maximum=8
                     )
                 
+                # 新增比例选择
+                ratio_selection = gr.Radio(
+                    choices=["9:16", "16:9"], 
+                    value="9:16",
+                    label="📐 选择视频比例",
+                    info="9:16适合抖音快手，16:9适合YouTube B站"
+                )
+                
                 process_btn = gr.Button(
                     "🎬 开始自动剪辑并打包", 
                     variant="primary", 
@@ -255,7 +361,7 @@ def main():
             with gr.Column(scale=1):
                 status_output = gr.Textbox(
                     label="📊 处理状态", 
-                    lines=10, 
+                    lines=12, 
                     interactive=False,
                     show_copy_button=True
                 )
@@ -270,14 +376,14 @@ def main():
             with gr.Column(scale=1):
                 details_output = gr.Textbox(
                     label="📝 详细信息", 
-                    lines=15, 
+                    lines=18, 
                     interactive=False,
                     show_copy_button=True
                 )
         
         process_btn.click(
-            fn=process_and_package,
-            inputs=[video_input, clip_duration, num_output],
+            fn=process_and_package_with_ratio,
+            inputs=[video_input, clip_duration, num_output, ratio_selection],
             outputs=[status_output, download_file, details_output]
         )
         
@@ -287,24 +393,34 @@ def main():
         
         **🎬 视频处理流程：**
         1. **上传视频** → 支持 MP4、MOV、AVI、MKV 格式
-        2. **自动切片** → 按指定时长将视频切成小段
-        3. **智能混剪** → 随机重组片段生成新视频
-        4. **自动打包** → 所有视频 + 说明文档打包成ZIP
-        5. **一键下载** → 保存到本地硬盘
+        2. **设置参数** → 切片时长、输出数量、视频比例
+        3. **自动切片** → 按指定时长将视频切成小段
+        4. **智能混剪** → 随机重组片段生成新视频
+        5. **比例调整** → 自动适配选择的目标比例
+        6. **自动打包** → 所有视频 + 说明文档打包成ZIP
+        7. **一键下载** → 保存到本地硬盘
+        
+        **📐 比例选择指南：**
+        
+        **🔥 9:16 竖屏 (推荐短视频)**
+        - 📱 **适合平台：** 抖音、快手、小红书、Instagram Stories
+        - 🎯 **内容类型：** 短视频、Vlog、才艺展示、产品介绍
+        - ⏱️ **最佳时长：** 15-60秒
+        - 🎵 **建议配乐：** 动感节拍强的背景音乐
+        
+        **🖥️ 16:9 横屏 (推荐长视频)**  
+        - 📺 **适合平台：** YouTube、B站、腾讯视频、爱奇艺
+        - 🎯 **内容类型：** 教程、游戏、影视解说、直播回放
+        - ⏱️ **最佳时长：** 3-15分钟
+        - 🖼️ **建议优化：** 精美封面、详细描述
         
         **💾 打包内容：**
-        - ✅ 所有生成的混剪视频文件
+        - ✅ 所有生成的混剪视频文件（指定比例）
         - ✅ 详细的使用说明文档  
-        - ✅ 技术参数和建议信息
+        - ✅ 平台发布建议和优化技巧
+        - ✅ 技术参数和后续处理指南
         - ✅ 压缩打包，节省存储空间
         
-        **🚀 优势特点：**
-        - 🔥 **完全本地化** → 无需任何在线服务
-        - ⚡ **批量处理** → 支持多视频同时处理
-        - 🎯 **智能算法** → 随机重组避免重复
-        - 📱 **适配平台** → 生成适合短视频平台的内容
-        - 💾 **打包下载** → 一次下载全部文件
-        
         **⚠️ 系统要求：**
         - 需要安装 FFmpeg 和 FFprobe
         - 建议使用稳定的网络连接