Update app.py

app.py

@@ -1,15 +1,10 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-
-import gradio as gr
 import os
+import gradio as gr
 import tempfile
-import json
 import random
 from datetime import datetime
-from pathlib import Path
 
-# AI 和视频处理导入
+# AI 导入
 try:
     from openai import OpenAI
     client = OpenAI(api_key=os.getenv("OPENAI_API_KEY")) if os.getenv("OPENAI_API_KEY") else None
@@ -18,15 +13,14 @@ except ImportError:
     OPENAI_AVAILABLE = False
     client = None
 
-# 视频处理导入
+# 视频处理导入（MoviePy）
 try:
     from moviepy.editor import VideoFileClip, concatenate_videoclips, AudioFileClip, vfx
     MOVIEPY_AVAILABLE = True
 except ImportError:
     MOVIEPY_AVAILABLE = False
-    print("⚠️ MoviePy 不可用，将使用模拟模式")
 
-# YouTube 上传导入
+# YouTube API 导入
 try:
     from googleapiclient.discovery import build
     from googleapiclient.http import MediaFileUpload
@@ -37,248 +31,146 @@ try:
 except ImportError:
     YOUTUBE_AVAILABLE = False
 
-# 全局配置
 MAX_VIDEO_SIZE = 100 * 1024 * 1024  # 100MB
 MAX_AUDIO_SIZE = 50 * 1024 * 1024   # 50MB
 MAX_TOTAL_SIZE = 500 * 1024 * 1024  # 500MB
 
-class YouTubeUploader:
-    """YouTube 上传工具类"""
-    
-    def __init__(self):
-        self.SCOPES = ['https://www.googleapis.com/auth/youtube.upload']
-        self.youtube = None
-        self.authenticated = False
-    
-    def setup_credentials(self, client_secrets_content):
-        """设置 Google Cloud 凭据"""
-        try:
-            # 使用你已有的凭据
-            credentials_data = {
-                "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"]
-                }
-            }
-            
-            # 保存到临时文件
-            temp_file = tempfile.NamedTemporaryFile(mode='w', suffix='.json', delete=False)
-            json.dump(credentials_data, temp_file)
-            temp_file.close()
-            
-            return temp_file.name
-        except Exception as e:
-            return None
-    
-    def generate_upload_script(self, video_titles):
-        """生成 YouTube 上传脚本"""
-        
-        script_template = f'''#!/usr/bin/env python3
-"""
-pyTovideo2 YouTube 自动上传脚本
-使用说明：
-1. pip install google-api-python-client google-auth-oauthlib
-2. 将此脚本与你的视频文件放在同一目录
-3. 运行：python youtube_upload_script.py
-"""
-
-import os
-import pickle
-import json
-from googleapiclient.discovery import build
-from googleapiclient.http import MediaFileUpload
-from google_auth_oauthlib.flow import InstalledAppFlow
-from google.auth.transport.requests import Request
-
-# 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"]
-    }}
-}}
-
+# YouTube 上传类
 class YouTubeUploader:
     def __init__(self):
         self.SCOPES = ['https://www.googleapis.com/auth/youtube.upload']
         self.youtube = None
-        
-        # 保存凭据文件
-        with open('client_secrets.json', 'w') as f:
-            json.dump(CLIENT_SECRETS, f)
+        self.credentials_file = 'youtube_token.pickle'
+        self.client_secrets_file = 'client_secrets.json'
     
     def authenticate(self):
-        """认证 YouTube API"""
-        credentials = None
-        
-        if os.path.exists('youtube_token.pickle'):
-            with open('youtube_token.pickle', 'rb') as token:
-                credentials = pickle.load(token)
-        
-        if not credentials or not credentials.valid:
-            if credentials and credentials.expired and credentials.refresh_token:
-                credentials.refresh(Request())
+        creds = None
+        if os.path.exists(self.credentials_file):
+            with open(self.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.json', self.SCOPES)
-                credentials = flow.run_local_server(port=0)
-            
-            with open('youtube_token.pickle', 'wb') as token:
-                pickle.dump(credentials, token)
-        
-        self.youtube = build('youtube', 'v3', credentials=credentials)
+                flow = InstalledAppFlow.from_client_secrets_file(self.client_secrets_file, self.SCOPES)
+                creds = flow.run_local_server(port=0)
+            with open(self.credentials_file, 'wb') as token:
+                pickle.dump(creds, token)
+        self.youtube = build('youtube', 'v3', credentials=creds)
         return True
     
     def upload_video(self, video_path, title, description="", tags="", privacy="private"):
-        """上传视频到 YouTube"""
-        body = {{
-            'snippet': {{
+        if not os.path.exists(video_path):
+            return {"success": False, "error": "视频文件不存在"}
+
+        body = {
+            'snippet': {
                 'title': title,
                 'description': description,
                 'tags': tags.split(',') if tags else [],
-                'categoryId': '22'
-            }},
-            'status': {{
-                'privacyStatus': privacy
-            }}
-        }}
-        
-        media = MediaFileUpload(video_path, resumable=True)
-        request = self.youtube.videos().insert(
-            part=','.join(body.keys()),
-            body=body,
-            media_body=media
-        )
-        
+                'categoryId': '22'  # People & Blogs
+            },
+            'status': {
+                'privacyStatus': privacy,
+                'selfDeclaredMadeForKids': False
+            }
+        }
+        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()
-        return f"https://www.youtube.com/watch?v={{response['id']}}"
-
-def main():
-    """主程序"""
-    print("🎬 pyTovideo2 YouTube 上传工具")
-    print("=" * 40)
-    
-    uploader = YouTubeUploader()
-    
-    if not uploader.authenticate():
-        print("❌ 认证失败")
-        return
-    
-    print("✅ YouTube 认证成功！")
-    
-    # 要上传的视频列表
-    videos_info = ['''
-        
-        for i, title in enumerate(video_titles, 1):
-            script_template += f'''
-        {{
-            "file": "混剪视频_{i:02d}.mp4",
-            "title": "{title}",
-            "description": """🎬 使用 pyTovideo2 制作的精彩混剪视频
-
-✨ 制作特色：
-• 智能视频切片处理
-• 音乐节拍完美同步
-• 创意片段随机组合  
-• 镜像翻转防重复检测
-
-💡 工具优势：
-• 批量处理高效率
-• 支持多种视频格式
-• 一键生成多个版本
-• 完美适配各大平台
-
-👍 如果喜欢请：
-• 点赞支持这个项目
-• 订阅获取更多内容
-• 分享给感兴趣的朋友
-• 评论交流制作心得
-
-🔗 项目开源地址: https://github.com/BassZou/pyToVideo2
-
-#短视频 #混剪 #Python #自动化 #pyTovideo2 #视频制作""",
-            "tags": "短视频,混剪,Python,自动化,pyTovideo2,视频制作,内容创作",
-            "privacy": "private"  # private, unlisted, public
-        }},'''
-        
-        script_template += f'''
-    ]
-    
-    # 批量上传
-    print(f"🚀 开始批量上传 {{len(videos_info)}} 个视频...")
-    
-    for i, video_info in enumerate(videos_info, 1):
-        if os.path.exists(video_info['file']):
-            print(f"\\n--- 上传第 {{i}}/{{len(videos_info)}} 个视频 ---")
-            print(f"📹 文件: {{video_info['file']}}")
-            
-            try:
-                url = uploader.upload_video(
-                    video_path=video_info['file'],
-                    title=video_info['title'],
-                    description=video_info['description'],
-                    tags=video_info['tags'],
-                    privacy=video_info['privacy']
-                )
-                print(f"✅ 上传成功: {{url}}")
-            except Exception as e:
-                print(f"❌ 上传失败: {{e}}")
-        else:
-            print(f"❌ 文件不存在: {{video_info['file']}}")
-    
-    print("\\n🎉 批量上传完成！")
-
-if __name__ == "__main__":
-    main()
-'''
-        
-        return script_template
+        video_id = response.get('id')
+        video_url = f"https://www.youtube.com/watch?v={video_id}"
+        return {"success": True, "video_id": video_id, "url": video_url}
 
+# AI 视频分析
 def generate_ai_analysis(video_count, file_names):
-    """使用 AI 生成视频分析"""
     if not OPENAI_AVAILABLE:
-        return f"""
-📊 **视频分析报告**（基于文件信息）
-
-📹 **内容预测**：
-根据上传的 {video_count} 个视频文件，预测内容类型：
-• 生活记录类：适合温馨背景音乐
-• 技能展示类：适合节奏感强的音乐
-• 风景旅行类��适合轻松舒缓音乐
-• 娱乐搞笑类：适合活泼欢快音乐
-
-🎵 **音乐建议**：
-• 节拍：120-140 BPM 最适合短视频
-• 风格：流行、电子、轻音乐
-• 时长：与最终视频长度匹配
-
-✂️ **剪辑策略**：
-• 快切：适合动感内容，2-3秒一切
-• 慢节奏：适合情感内容，4-5秒一切
-• 混合节奏：根据音乐变化调整
-
-🎯 **平台优化**：
-• 抖音：竖屏 9:16，15-60秒
-• 小红书：方形 1:1，15-90秒  
-• YouTube：横屏 16:9，3-10分钟
-        """
-    
+        return "⚠️ 未配置 OpenAI，无法生成 AI 分析。"
     try:
-        file_info = ", ".join([f"'{name}'" for name in file_names[:3]])  # 只取前3个文件名
-        
-        prompt = f"""作为专业视频分析师，基于用户上传的 {video_count} 个视频文件（文件名包括：{file_info}等），请提供详细分析：
+        file_info = ", ".join([f"'{name}'" for name in file_names[:3]])
+        prompt = f"""作为专业视频分析师，基于用户上传的 {video_count} 个视频文件（文件名包括：{file_info} 等），请提供详细分析：
+1. 内容类型推测和特点
+2. 最佳背景音乐风格和节拍
+3. 推荐剪辑策略和转场效果
+4. 目标受众分析
+
+请用中文回答，格式清晰。"""
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": "你是专业的视频内容分析专家。"},
+                {"role": "user", "content": prompt}
+            ],
+            max_tokens=600,
+            temperature=0.7
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"❌ AI 分析失败：{str(e)}"
+
+# 视频文件大小检查
+def check_filesize(video_files, audio_file):
+    total_size = 0
+    for v in video_files:
+        size = os.path.getsize(v.name)
+        if size > MAX_VIDEO_SIZE:
+            return False, f"单个视频文件 {os.path.basename(v.name)} 过大 {size/(1024*1024):.1f}MB，最大限制为 100MB"
+        total_size += size
+    if audio_file:
+        audio_size = os.path.getsize(audio_file.name)
+        if audio_size > MAX_AUDIO_SIZE:
+            return False, f"音频文件 {os.path.basename(audio_file.name)} 过大 {audio_size/(1024*1024):.1f}MB，最大限制为 50MB"
+        total_size += audio_size
+    if total_size > MAX_TOTAL_SIZE:
+        return False, f"上传文件总大小过大 {total_size/(1024*1024):.1f}MB，最大限制为 500MB"
+    return True, ""
+
+# 视频处理示范（简化，未完整实现）
+def process_videos(video_files, audio_file, clip_duration=2, num_output=3):
+    ok, msg = check_filesize(video_files, audio_file)
+    if not ok:
+        return msg, [], ""
+    # 这里可以使用 MoviePy 处理视频，当前仅模拟
+    out_files = []  # 真实项目中应为生成的视频路径列表
+    return "✅ 视频上传成功，处理已完成（演示模式，不生成文件）", out_files, ""
+
+# Gradio 主界面
+def main_app(video_files, audio_file, clip_duration, num_output, upload_yt, yt_privacy):
+    status, outfiles, _ = process_videos(video_files, audio_file, clip_duration, num_output)
+    file_names = [os.path.basename(f.name) for f in (video_files or [])]
+    ai_analysis = generate_ai_analysis(len(video_files or []), file_names) if OPENAI_AVAILABLE else "未启用 AI 分析功能"
+    
+    yt_upload_res = ""
+    if upload_yt:
+        if not YOUTUBE_AVAILABLE:
+            yt_upload_res = "⚠️ 未安装 YouTube API 库，上传功能不可用"
+        elif not os.path.exists("client_secrets.json"):
+            yt_upload_res = "⚠️ client_secrets.json 文件缺失，无法上传"
+        else:
+            uploader = YouTubeUploader()
+            auth_ok = uploader.authenticate()
+            if not auth_ok:
+                yt_upload_res = "❌ YouTube 认证失败"
+            else:
+                # 这里示范上传已生成文件，示例为空，需替换为真实文件路径
+                yt_upload_res = "上传功能暂未实现，请在本地运行上传脚本"
+
+    return status, ai_analysis, yt_upload_res
+
+with gr.Blocks() as demo:
+    gr.Markdown("## pyTovideo2 视频处理 + YouTube 上传示范")
+    video_files = gr.File(file_types=[".mp4", ".mov", ".avi"], label="上传视频文件(s)", file_count="multiple")
+    audio_file = gr.File(file_types=[".mp3", ".wav", ".m4a"], label="上传音频文件", file_count="single")
+    clip_duration = gr.Number(value=2, label="切片时长 (秒)", minimum=1, maximum=10)
+    num_output = gr.Number(value=3, label="输出视频数量", minimum=1, maximum=10)
+    upload_yt = gr.Checkbox(label="处理后自动上传至 YouTube")
+    yt_privacy = gr.Dropdown(choices=["public", "private", "unlisted"], value="private", label="YouTube 视频隐私状态")
+    btn = gr.Button("开始处理")
+    status_out = gr.Textbox(label="处理状态", lines=4)
+    ai_out = gr.Textbox(label="AI 内容分析", lines=12)
+    yt_out = gr.Textbox(label="YouTube 上传结果", lines=8)
+    
+    btn.click(main_app, inputs=[video_files, audio_file, clip_duration, num_output, upload_yt, yt_privacy],
+              outputs=[status_out, ai_out, yt_out])
 
-1. 📹 内容类型推测和特点分析
-2. 🎵 最佳背景音乐风格和节拍建议
-3. ✂️ 推荐的剪辑策略和转场效果
-4. 🎯 目标受
+demo.launch()