Update app.py

app.py

@@ -522,8 +522,8 @@ def concat_videos(file_list, output_path):
         except:
             pass
 
-def process_single_video(video_file, clip_duration, temp_dir):
-    """[优化] 处理单个视频文件，返回其所有切片路径的列表"""
+def process_single_video(video_file, clip_duration, temp_dir, max_clips=50):
+    """[优化] 处理单个视频文件，返回其所有切片路径的列表，限制最大片段数"""
     video_path = video_file.name
     clips = []
     try:
@@ -532,13 +532,19 @@ def process_single_video(video_file, clip_duration, temp_dir):
         result = subprocess.run(cmd, capture_output=True, text=True, timeout=10)
         total_duration = float(result.stdout.strip())
         print(f"视频总时长: {total_duration} 秒")
+        
+        # 计算需要多少个片段
+        max_possible_clips = int(total_duration / clip_duration) + 1
+        actual_max_clips = min(max_possible_clips, max_clips)
+        print(f"将生成最多 {actual_max_clips} 个片段")
+        
     except Exception as e:
         print(f"处理视频 {video_path} 时获取时长失败: {e}")
         return clips
 
     start = 0.0
     count = 0
-    while start < total_duration:
+    while start < total_duration and count < actual_max_clips:
         duration = min(clip_duration, total_duration - start)
         # 修改文件名，包含原始文件名和计数，避免不同视频的切片重名
         clip_path = os.path.join(temp_dir, f"clip_{os.path.splitext(os.path.basename(video_path))[0]}_{count}.mp4")
@@ -554,6 +560,7 @@ def process_single_video(video_file, clip_duration, temp_dir):
         start += clip_duration
         count += 1
 
+    print(f"实际生成 {len(clips)} 个片段")
     return clips
 
 def process_videos_with_storage(video_files, clip_duration, num_output_videos, target_ratio):
@@ -575,8 +582,15 @@ def process_videos_with_storage(video_files, clip_duration, num_output_videos, t
         # 使用线程池进行并行处理（避免GIL限制）
         print(f"🚀 启动并行切片（{min(4, os.cpu_count() or 1)} 线程）...")
         with concurrent.futures.ThreadPoolExecutor(max_workers=min(4, os.cpu_count() or 1)) as executor:
+            # 计算每个视频最多需要多少个片段
+            # 为了生成num_output_videos个混剪视频，每个混剪视频需要3-5个片段
+            # 所以总共需要大约 num_output_videos * 4 个片段
+            # 如果有多个视频，平均分配
+            total_needed_clips = num_output_videos * 4  # 每个混剪视频平均4个片段
+            clips_per_video = max(10, total_needed_clips // len(video_files))  # 每个视频至少10个片段
+            
             future_to_video = {
-                executor.submit(process_single_video, vf, clip_duration, temp_dir): vf 
+                executor.submit(process_single_video, vf, clip_duration, temp_dir, clips_per_video): vf 
                 for vf in video_files
             }
             
@@ -597,6 +611,12 @@ def process_videos_with_storage(video_files, clip_duration, num_output_videos, t
         if not all_clips:
             return "❌ 切割失败，请检查视频文件", None, "", ""
 
+        # 如果片段太多，随机选择需要的数量
+        if len(all_clips) > total_needed_clips:
+            print(f"片段数量({len(all_clips)})超过需求({total_needed_clips})，随机选择...")
+            all_clips = random.sample(all_clips, total_needed_clips)
+            print(f"选择后剩余 {len(all_clips)} 个片段")
+
         random.shuffle(all_clips)
         clips_per_video = max(1, len(all_clips) // num_output_videos)
         output_files = []
@@ -808,7 +828,7 @@ def main():
                         
                         with gr.Row():
                             clip_duration = gr.Number(value=3, label="切片时长(秒)", minimum=1, maximum=3600)
-                            num_output = gr.Number(value=3, label="生成数量", minimum=1, maximum=100)
+                            num_output = gr.Number(value=5, label="生成数量", minimum=1, maximum=100)
                         
                         ratio_selection = gr.Radio(
                             choices=["9:16", "16:9"], 
@@ -952,6 +972,11 @@ def main():
         - **快速预设**: 使用ultrafast预设提升速度
         - **并行处理**: 多个视频同时处理
         
+        **🎯 智能切片:**
+        - **按需生成**: 根据要生成的视频数量智能计算需要的片段数
+        - **避免冗余**: 不会生成超过需要的片段，节省处理时间
+        - **随机选择**: 从生成的片段中随机选择，确保混剪多样性
+        
         **⚠️ 注意事项:**
         - 下载文件为ZIP格式，需要解压使用
         - 一键下载包含储存空间中所有视频文件