Update app.py

app.py

@@ -1,92 +1,118 @@
 import os
-import pickle
-from googleapiclient.discovery import build
-from googleapiclient.http import MediaFileUpload
-from google_auth_oauthlib.flow import InstalledAppFlow
-from google.auth.transport.requests import Request
-
-CLIENT_SECRETS_FILE = 'client_secrets.json'
-CREDENTIALS_FILE = 'youtube_token.pickle'
-SCOPES = ['https://www.googleapis.com/auth/youtube.upload']
-
-def authenticate_youtube():
-    creds = None
-    if os.path.exists(CREDENTIALS_FILE):
-        with open(CREDENTIALS_FILE, 'rb') as token:
-            creds = pickle.load(token)
-    if not creds or not creds.valid:
-        if creds and creds.expired and creds.refresh_token:
-            creds.refresh(Request())
-        else:
-            flow = InstalledAppFlow.from_client_secrets_file(
-                CLIENT_SECRETS_FILE, SCOPES)
-            # 不自动打开浏览器，手动复制链接访问授权
-            creds = flow.run_local_server(port=0, open_browser=False)
-            print("请复制以上链接，在浏览器中打开并完成授权，授权成功后程序将继续运行。")
-        with open(CREDENTIALS_FILE, 'wb') as token:
-            pickle.dump(creds, token)
-    youtube = build('youtube', 'v3', credentials=creds)
-    return youtube
-
-def upload_video(youtube, video_path, title, description, tags=None, privacy_status="private"):
-    if not os.path.exists(video_path):
-        print(f"错误：找不到视频文件 {video_path}")
-        return None
-    
-    body = {
-        'snippet': {
-            'title': title,
-            'description': description,
-            'tags': tags or [],
-            'categoryId': '22'  # People & Blogs
-        },
-        'status': {
-            'privacyStatus': privacy_status,
-            'selfDeclaredMadeForKids': False
-        }
-    }
-    
-    media = MediaFileUpload(video_path, chunksize=-1, resumable=True, mimetype='video/*')
-    request = youtube.videos().insert(
-        part=','.join(body.keys()),
-        body=body,
-        media_body=media
-    )
-    response = request.execute()
-    
-    print(f"上传成功！视频链接：https://www.youtube.com/watch?v={response['id']}")
-    return response['id']
+import gradio as gr
+import tempfile
+import random
+import subprocess
+from datetime import datetime
 
-def main():
-    youtube = authenticate_youtube()
-
-    # 这里填写你要上传的视频文件路径和信息
-    videos_to_upload = [
-        {
-            'path': 'your_video_1.mp4',
-            'title': '视频标题1',
-            'description': '这是视频描述1',
-            'tags': ['示例', '测试'],
-            'privacy_status': 'private'
-        },
-        {
-            'path': 'your_video_2.mp4',
-            'title': '视频标题2',
-            'description': '这是视频描述2',
-            'tags': ['样例', '上传'],
-            'privacy_status': 'unlisted'
-        }
+def ffmpeg_cut_video(input_path, start_time, duration, output_path):
+    """调用ffmpeg剪辑视频片段"""
+    command = [
+        'ffmpeg',
+        '-ss', str(start_time),
+        '-i', input_path,
+        '-t', str(duration),
+        '-c', '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):
+    """调用ffmpeg合并多个视频，使用文件列表"""
+    list_file = tempfile.NamedTemporaryFile(delete=False, mode='w', suffix='.txt')
+    try:
+        for f in file_list:
+            list_file.write(f"file '{f.replace('\'', '\'\\\'\'' )}'\n")
+        list_file.close()
+
+        command = [
+            'ffmpeg',
+            '-f', 'concat',
+            '-safe', '0',
+            '-i', list_file.name,
+            '-c', 'copy',
+            '-y',
+            output_path
+        ]
+        process = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        return process.returncode == 0
+    finally:
+        os.unlink(list_file.name)
+
+def process_videos(video_files, clip_duration=2, num_output_videos=3):
+    if not video_files or len(video_files) == 0:
+        return "❌ 请上传至少一个视频文件", []
+
+    temp_dir = tempfile.mkdtemp()
+    all_clips = []
+
+    try:
+        for idx, video_file in enumerate(video_files):
+            video_path = video_file.name
+            # 获取视频总时长
+            cmd_duration = [
+                'ffprobe', '-v', 'error', 
+                '-show_entries', 'format=duration',
+                '-of', 'default=noprint_wrappers=1:nokey=1',
+                video_path
+            ]
+            result = subprocess.run(cmd_duration, capture_output=True, text=True)
+            total_duration = float(result.stdout.strip())
+            # 按clip_duration分割视频
+            start = 0.0
+            count = 0
+            while start < total_duration:
+                duration = min(clip_duration, total_duration - start)
+                clip_path = os.path.join(temp_dir, f"clip_{idx}_{count}.mp4")
+                ok = ffmpeg_cut_video(video_path, start, duration, clip_path)
+                if not ok:
+                    return f"❌ 视频剪辑失败: {video_path}", []
+                all_clips.append(clip_path)
+                start += duration
+                count += 1
+
+        # 随机打乱clip顺序
+        random.shuffle(all_clips)
+
+        # 计算每个输出视频包含的clip数量
+        clips_per_video = max(1, len(all_clips) // num_output_videos)
 
-    for video in videos_to_upload:
-        upload_video(
-            youtube,
-            video['path'],
-            video['title'],
-            video['description'],
-            video['tags'],
-            video['privacy_status']
-        )
-
-if __name__ == '__main__':
+        output_files = []
+
+        for i in range(num_output_videos):
+            start_idx = i * clips_per_video
+            end_idx = len(all_clips) if i == num_output_videos -1 else (start_idx + clips_per_video)
+            selected_clips = all_clips[start_idx:end_idx]
+            output_path = os.path.join(temp_dir, f"mixed_video_{i +1}.mp4")
+            ok = concat_videos_ffmpeg(selected_clips, output_path)
+            if not ok:
+                return "❌ 视频合并失败", []
+            output_files.append(output_path)
+
+        return f"成功生成 {len(output_files)} 个混剪视频", output_files
+
+    except Exception as e:
+        return f"❌ 处理异常: {str(e)}", []
+
+def main():
+    with gr.Blocks() as demo:
+        gr.Markdown("## FFmpeg 自动剪辑混剪视频 (无需任何API)")
+        video_input = gr.File(file_types=[".mp4", ".mov", ".avi", ".mkv"], file_count="multiple", label="上传视频文件")
+        clip_duration = gr.Number(value=2, label="剪辑片段时长 (秒)", minimum=1, maximum=30)
+        num_output = gr.Number(value=3, label="输出混剪视频数量", minimum=1, maximum=10)
+        btn = gr.Button("开始剪辑")
+
+        status = gr.Textbox(label="剪辑状态", interactive=False, lines=4)
+        outputs = gr.File(file_count="multiple", label="下载混剪视频")
+
+        def process(video_files, clip_duration, num_output):
+            return process_videos(video_files, clip_duration, num_output)
+
+        btn.click(process, inputs=[video_input, clip_duration, num_output], outputs=[status, outputs])
+
+    demo.launch()
+    
+if __name__ == "__main__":
     main()