Update app.py

app.py

@@ -4,16 +4,124 @@ import tempfile
 import random
 import subprocess
 import shutil
+import json
+import pickle
+from datetime import datetime
+
+# Google API 导入
+try:
+    from googleapiclient.discovery import build
+    from googleapiclient.http import MediaFileUpload
+    from google_auth_oauthlib.flow import InstalledAppFlow
+    from google.auth.transport.requests import Request
+    YOUTUBE_AVAILABLE = True
+except ImportError:
+    YOUTUBE_AVAILABLE = False
+
+# 内置Google Cloud凭据（用户无需设置）
+CLIENT_SECRETS = {
+    "installed": {
+        "client_id": "288005932157-k7dhq8c7rgmt8km0nm50ntuif9lse7j1.apps.googleusercontent.com",
+        "project_id": "corded-palisade-458105-p7",
+        "auth_uri": "https://accounts.google.com/o/oauth2/auth",
+        "token_uri": "https://oauth2.googleapis.com/token",
+        "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
+        "client_secret": "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",
+        "redirect_uris": ["http://localhost"]
+    }
+}
+
+SCOPES = ['https://www.googleapis.com/auth/youtube.upload']
+CREDENTIALS_FILE = 'youtube_token.pickle'
+
+class SimpleYouTubeUploader:
+    def __init__(self):
+        self.youtube = None
+        self.authenticated = False
+    
+    def authenticate(self):
+        if not YOUTUBE_AVAILABLE:
+            return False, "❌ YouTube API库未安装"
+        
+        try:
+            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:
+                    # 创建临时凭据文件
+                    temp_secrets = tempfile.NamedTemporaryFile(mode='w', suffix='.json', delete=False)
+                    json.dump(CLIENT_SECRETS, temp_secrets)
+                    temp_secrets.close()
+                    
+                    flow = InstalledAppFlow.from_client_secrets_file(temp_secrets.name, SCOPES)
+                    creds = flow.run_local_server(port=0, open_browser=False)
+                    
+                    # 清理临时文件
+                    os.unlink(temp_secrets.name)
+                
+                with open(CREDENTIALS_FILE, 'wb') as token:
+                    pickle.dump(creds, token)
+            
+            self.youtube = build('youtube', 'v3', credentials=creds)
+            self.authenticated = True
+            return True, "✅ YouTube认证成功"
+            
+        except Exception as e:
+            return False, f"❌ 认证失败: {str(e)}"
+    
+    def upload_video(self, video_path, title, description="", privacy="private"):
+        if not self.authenticated:
+            return {"success": False, "error": "请先完成认证"}
+        
+        if not os.path.exists(video_path):
+            return {"success": False, "error": f"文件不存在: {video_path}"}
+        
+        body = {
+            'snippet': {
+                'title': title,
+                'description': description,
+                'tags': ["混剪", "自动生成", "短视频", "创意"],
+                'categoryId': '22'
+            },
+            'status': {
+                'privacyStatus': privacy,
+                'selfDeclaredMadeForKids': False
+            }
+        }
+        
+        try:
+            media = MediaFileUpload(video_path, resumable=True, mimetype='video/*')
+            request = self.youtube.videos().insert(
+                part=','.join(body.keys()),
+                body=body,
+                media_body=media
+            )
+            
+            response = request.execute()
+            video_id = response['id']
+            url = f"https://www.youtube.com/watch?v={video_id}"
+            
+            return {
+                "success": True,
+                "video_id": video_id,
+                "url": url,
+                "title": title
+            }
+        except Exception as e:
+            return {"success": False, "error": str(e)}
+
+# 创建全局上传器实例
+youtube_uploader = SimpleYouTubeUploader()
 
 def ffmpeg_cut_video(input_path, start_time, duration, output_path):
     command = [
-        'ffmpeg',
-        '-ss', str(start_time),
-        '-i', input_path,
-        '-t', str(duration),
-        '-c', 'copy',
-        '-y',
-        output_path
+        '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
@@ -26,13 +134,8 @@ def concat_videos_ffmpeg(file_list, output_path):
         list_file.close()
 
         command = [
-            'ffmpeg',
-            '-f', 'concat',
-            '-safe', '0',
-            '-i', list_file.name,
-            '-c', 'copy',
-            '-y',
-            output_path
+            '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
@@ -46,13 +149,12 @@ def extract_clips_from_videos(video_files, clip_duration):
         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
+                '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())
+            
             start = 0.0
             count = 0
             while start < total_duration:
@@ -74,6 +176,7 @@ def generate_mixed_videos(clips, num_output_videos):
     random.shuffle(clips)
     clips_per_video = max(1, len(clips) // num_output_videos)
     output_files = []
+    
     try:
         for i in range(num_output_videos):
             start_idx = i * clips_per_video
@@ -89,34 +192,170 @@ def generate_mixed_videos(clips, num_output_videos):
         shutil.rmtree(temp_dir, ignore_errors=True)
         raise
 
-def process_auto_mixing(video_files, clip_duration, num_output_videos):
+def process_and_upload(video_files, clip_duration, num_output_videos, upload_to_youtube, privacy_setting):
     if not video_files or len(video_files) == 0:
-        return "❌ 请上传至少一个视频文件", []
+        return "❌ 请上传至少一个视频文件", [], ""
+    
     try:
+        # 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)
-        # 清理临时切割片段目录
         shutil.rmtree(clips_temp_dir, ignore_errors=True)
-        return f"成功生成 {len(output_files)} 个自动混剪视频", output_files
-    except Exception as e:
-        return f"❌ 处理失败: {str(e)}", []
+        
+        status_msg = f"✅ 成功生成 {len(output_files)} 个混剪视频"
+        upload_result = ""
+        
+        # 2. YouTube上传（如果勾选）
+        if upload_to_youtube and YOUTUBE_AVAILABLE:
+            auth_success, auth_msg = youtube_uploader.authenticate()
+            upload_result += auth_msg + "\n\n"
+            
+            if auth_success:
+                upload_result += "🚀 开始上传到YouTube...\n\n"
+                
+                for i, video_path in enumerate(output_files, 1):
+                    title = f"自动混剪视频 #{i} - {datetime.now().strftime('%Y%m%d_%H%M')}"
+                    description = f"""🎬 自动生成的混剪视频
 
-def main():
-    with gr.Blocks() as demo:
-        gr.Markdown("# FFmpeg 自动剪辑 + 长视频自动切片 + 自动混剪")
+📹 **制作信息：**
+• 切片时长：{clip_duration}秒
+• 生成时间：{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
+• 自动随机混剪
 
-        video_input = gr.File(label="上传视频文件", file_types=[".mp4", ".mov", ".avi"], file_count="multiple")
-        clip_duration = gr.Number(value=8, label="单个剪辑片段时长（秒）", minimum=1, maximum=60)
-        num_output = gr.Number(value=3, label="生成混剪视频数量", minimum=1, maximum=10)
-        btn = gr.Button("开始自动剪辑并混剪")
+🔧 **制作工具：**
+• FFmpeg 自动剪辑
+• Python 批量处理
+• 智能片段组合
 
-        status_box = gr.Textbox(label="处理状态", interactive=False, lines=4)
-        output_files = gr.File(label="下载混剪视频文件", file_count="multiple")
+👍 **如果喜欢请点赞支持！**
 
-        btn.click(process_auto_mixing, inputs=[video_input, clip_duration, num_output], outputs=[status_box, output_files])
+#混剪 #自动生成 #短视频 #创意 #FFmpeg"""
+                    
+                    result = youtube_uploader.upload_video(
+                        video_path=video_path,
+                        title=title,
+                        description=description,
+                        privacy=privacy_setting
+                    )
+                    
+                    if result["success"]:
+                        upload_result += f"✅ 视频 {i} 上传成功\n🔗 {result['url']}\n\n"
+                    else:
+                        upload_result += f"❌ 视频 {i} 上传失败: {result['error']}\n\n"
+            else:
+                upload_result += "❌ YouTube认证失败，无法上传"
+                
+        elif upload_to_youtube and not YOUTUBE_AVAILABLE:
+            upload_result = "⚠️ YouTube API库未安装，无法上传"
+        
+        return status_msg, output_files, upload_result
+        
+    except Exception as e:
+        return f"❌ 处理失败: {str(e)}", [], ""
 
+def main():
+    with gr.Blocks(
+        title="FFmpeg自动剪辑+YouTube一键上传", 
+        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 自动剪辑 + YouTube 一键上传</h1>
+            <p style="margin: 10px 0 0 0;">长视频自动切片 → 智能混剪 → 一键上传YouTube</p>
+        </div>
+        """)
+        
+        with gr.Row():
+            with gr.Column(scale=2):
+                video_input = gr.File(
+                    label="📤 上传视频文件", 
+                    file_types=[".mp4", ".mov", ".avi"], 
+                    file_count="multiple",
+                    height=100
+                )
+                
+                with gr.Row():
+                    clip_duration = gr.Number(
+                        value=3, 
+                        label="切片时长(秒)", 
+                        minimum=1, 
+                        maximum=30
+                    )
+                    num_output = gr.Number(
+                        value=3, 
+                        label="生成视频数", 
+                        minimum=1, 
+                        maximum=6
+                    )
+                
+                with gr.Row():
+                    upload_youtube = gr.Checkbox(
+                        label="🚀 一键上传到YouTube", 
+                        value=False
+                    )
+                    privacy_setting = gr.Dropdown(
+                        choices=["private", "unlisted", "public"], 
+                        value="private",
+                        label="YouTube隐私设置"
+                    )
+                
+                process_btn = gr.Button(
+                    "🎬 开始处理并上传", 
+                    variant="primary", 
+                    size="lg"
+                )
+            
+            with gr.Column(scale=1):
+                status_output = gr.Textbox(
+                    label="📊 处理状态", 
+                    lines=6, 
+                    interactive=False
+                )
+        
+        with gr.Row():
+            with gr.Column():
+                output_files = gr.File(
+                    label="📁 下载生成的视频", 
+                    file_count="multiple"
+                )
+            
+            with gr.Column():
+                upload_result = gr.Textbox(
+                    label="🚀 YouTube上传结果", 
+                    lines=10, 
+                    interactive=False,
+                    show_copy_button=True
+                )
+        
+        process_btn.click(
+            fn=process_and_upload,
+            inputs=[video_input, clip_duration, num_output, upload_youtube, privacy_setting],
+            outputs=[status_output, output_files, upload_result]
+        )
+        
+        gr.Markdown("""
+        ---
+        ### 📖 使用说明
+        
+        **🎬 视频处理：**
+        - 上传一个或多个视频文件
+        - 自动按指定时长切片
+        - 随机重组生成新的混剪视频
+        
+        **🚀 YouTube上传：**
+        - 勾选"一键上传到YouTube"
+        - **首次使用**：会弹出授权链接，需要在浏览器中登录Google账号授权
+        - **后续使用**：直接上传，无需重复授权
+        - 支持设置视频隐私级别
+        
+        **⚠️ 注意事项：**
+        - 需要环境安装FFmpeg
+        - YouTube上传需要Google账号授权（仅需授权一次）
+        - 建议先设为私密视频测试
+        """)
+    
     demo.launch()
 
 if __name__ == "__main__":
     main()
-