Update app.py

app.py

@@ -1,222 +1,493 @@
 import gradio as gr
 import os
 import tempfile
-import subprocess
-import shutil
-from pathlib import Path
-import sys
+import random
+from datetime import datetime
+from moviepy.editor import VideoFileClip, concatenate_videoclips, AudioFileClip, vfx
+from openai import OpenAI
 
-def setup_environment():
-    """設置環境"""
+# 配置 OpenAI API
+client = OpenAI(
+    api_key=os.getenv("sk-ZYyxdh3NexC863OUdicSlKrunXEPlRlsLBMwc1vXMGEVwZdb")
+)
+
+def analyze_video_content(video_path, video_count):
+    """使用 OpenAI 分析视频内容"""
     try:
-        # 確保系統包已安裝
-        subprocess.run(["which", "ffmpeg"], check=True, capture_output=True)
-        print("✅ FFmpeg已安裝")
-    except subprocess.CalledProcessError:
-        print("⚠️ FFmpeg未找到，但Docker鏡像應該已包含")
+        prompt = f"""作为专业视频内容分析师，请为用户上传的 {video_count} 个视频文件生成详细分析报告。
 
-def clone_repo():
-    """克隆項目倉庫"""
-    repo_url = "https://github.com/SamurAIGPT/AI-Youtube-Shorts-Generator.git"
-    repo_dir = "./AI-Youtube-Shorts-Generator"
-    
-    if os.path.exists(repo_dir):
-        shutil.rmtree(repo_dir)
-    
+分析要点：
+1. 📹 视频内容推测（基于用户上传行为）
+2. 🎵 推荐的背景音乐风格和节奏
+3. ✂️ 最佳剪辑策略和转场效果
+4. 🎯 目标受众和传播渠道建议
+5. 📈 内容优化和提升建议
+
+请用中文回答，格式清晰，每个要点用emoji标识。"""
+
+        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)}\n💡 请检查 OpenAI API Key 是否正确设置"
+
+def generate_creative_titles(video_count, content_theme="创意混剪"):
+    """使用 OpenAI 生成创意标题"""
     try:
-        subprocess.run(["git", "clone", repo_url, repo_dir], 
-                      check=True, capture_output=True, text=True)
-        return repo_dir, "✅ 項目克隆成功"
-    except subprocess.CalledProcessError as e:
-        return None, f"❌ 克隆失敗: {e.stderr}"
-
-def install_project_deps(repo_dir):
-    """安裝項目依賴"""
+        prompt = f"""请为 {video_count} 个{content_theme}视频生成吸引眼球的标题。
+
+要求：
+- 🔥 热门关键词和流行元素
+- 📱 适合抖音、小红书等平台
+- 💯 每个标题15-25字最佳
+- 🎭 风格多样化（搞笑、治愈、励志等）
+- 🚀 包含动作词汇增加点击欲望
+
+请直接给出 {video_count} 个标题，每行一个，前面加上序号。"""
+
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {
+                    "role": "system",
+                    "content": "你是一位专业的新媒体内容创作者，精通各种平台的标题优化技巧。"
+                },
+                {
+                    "role": "user",
+                    "content": prompt
+                }
+            ],
+            max_tokens=400,
+            temperature=0.9  # 提高创意度
+        )
+        
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"❌ 标题生成失败: {str(e)}"
+
+def generate_project_summary(video_count, processing_time):
+    """生成项目处理总结"""
     try:
-        # 進入項目目錄安裝依賴
-        original_dir = os.getcwd()
-        os.chdir(repo_dir)
-        
-        # 檢查是否有requirements.txt
-        if os.path.exists("requirements.txt"):
-            result = subprocess.run([
-                sys.executable, "-m", "pip", "install", "-r", "requirements.txt"
-            ], capture_output=True, text=True, timeout=180)
-            
-            if result.returncode != 0:
-                os.chdir(original_dir)
-                return False, f"依賴安裝失敗: {result.stderr}"
+        prompt = f"""请为刚完成的视频混剪项目生成专业总结报告：
+
+项目信息：
+- 处理视频数量: {video_count} 个
+- 处理用时: {processing_time}
+- 使用技术: MoviePy + OpenAI GPT
+
+请包含：
+1. 🎬 项目执行概况
+2. ⚡ 技术亮点和特色
+3. 📊 质量评估和优势
+4. 🔮 后续优化方向
+5. 💼 商业应用价值
+
+用专业但易懂的语言，中文回答。"""
+
+        response = client.chat.completions.create(
+            model="gpt-4o",  # 使用更强的模型做总结
+            messages=[
+                {
+                    "role": "system",
+                    "content": "你是一位技术项目管理专家，擅长撰写项目总结和技术方案评估报告。"
+                },
+                {
+                    "role": "user",
+                    "content": prompt
+                }
+            ],
+            max_tokens=700,
+            temperature=0.6
+        )
         
-        os.chdir(original_dir)
-        return True, "✅ 項目依賴安裝成功"
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"❌ 总结生成失败: {str(e)}"
+
+def get_ai_processing_suggestions(file_count, audio_duration):
+    """获取处理建议"""
+    try:
+        prompt = f"""基于用户上传情况给出处理建议：
+- 视频文件数: {file_count}
+- 音频时长: {audio_duration:.1f}秒
+
+请提供：
+1. 最佳切片时长建议
+2. 推荐输出视频数量
+3. 效果和滤镜建议
+4. 注意事项提醒
+
+简洁回答，3-4条建议即可。"""
+
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "user", "content": prompt}
+            ],
+            max_tokens=300,
+            temperature=0.5
+        )
         
+        return response.choices[0].message.content
     except Exception as e:
-        if 'original_dir' in locals():
-            os.chdir(original_dir)
-        return False, f"安裝過程出錯: {str(e)}"
+        return "无法获取AI建议，将使用默认参数处理"
 
-def process_video(youtube_url, openai_api_key, progress=gr.Progress()):
-    """處理YouTube視頻"""
-    
-    if not youtube_url.strip():
-        return None, "❌ 請輸入YouTube URL"
-    
-    if not openai_api_key.strip():
-        return None, "❌ 請輸入OpenAI API Key"
+def process_videos_with_openai(video_files, audio_file, clip_duration=2, num_output_videos=3):
+    """集成 OpenAI GPT 的视频处理主函数"""
     
-    # 驗證URL格式
-    if not any(domain in youtube_url for domain in ['youtube.com', 'youtu.be']):
-        return None, "❌ 請輸入有效的YouTube URL"
+    if not video_files or not audio_file:
+        return "❌ 请上传视频文件和音频文件", [], "", "", ""
     
-    progress(0.1, desc="初始化環境...")
-    setup_environment()
+    start_time = datetime.now()
     
-    progress(0.2, desc="克隆項目代碼...")
-    repo_dir, clone_msg = clone_repo()
-    if not repo_dir:
-        return None, clone_msg
+    # 获取 AI 处理建议
+    audio_duration = 0
+    try:
+        temp_audio = AudioFileClip(audio_file.name)
+        audio_duration = temp_audio.duration
+        temp_audio.close()
+    except:
+        audio_duration = 30  # 默认值
     
-    progress(0.3, desc="安裝項目依賴...")
-    deps_ok, deps_msg = install_project_deps(repo_dir)
-    if not deps_ok:
-        return None, deps_msg
+    ai_suggestions = get_ai_processing_suggestions(len(video_files), audio_duration)
     
-    # 設置環境變量
-    env = os.environ.copy()
-    env["OPENAI_API_KEY"] = openai_api_key
+    # 开始视频分析
+    status_updates = "🤖 OpenAI GPT 正在分析视频内容...\n"
+    video_analysis = analyze_video_content(video_files[0].name if video_files else "", len(video_files))
     
     try:
-        progress(0.5, desc="處理視頻中...")
-        
-        original_dir = os.getcwd()
-        os.chdir(repo_dir)
-        
-        # 創建輸入文件
-        with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.txt') as f:
-            f.write(f"{youtube_url}\n")
-            input_file = f.name
-        
-        # 執行主程序
-        with open(input_file, 'r') as f:
-            result = subprocess.run(
-                [sys.executable, "main.py"],
-                stdin=f,
-                capture_output=True,
-                text=True,
-                env=env,
-                timeout=600  # 10分鐘超時
-            )
+        # 创建输出目录
+        temp_dir = tempfile.mkdtemp()
+        output_dir = os.path.join(temp_dir, f"openai_output_{datetime.now().strftime('%m%d_%H%M')}")
+        os.makedirs(output_dir, exist_ok=True)
         
-        # 清理輸入文件
-        os.unlink(input_file)
+        status_updates += "🎬 开始处理视频文件...\n"
         
-        progress(0.8, desc="查找生成的視頻...")
+        # 处理视频逻辑
+        video_clips = []
+        total_source_duration = 0
         
-        # 查找輸出視頻
-        video_files = []
-        for pattern in ['*.mp4', '*.avi', '*.mov']:
-            video_files.extend(Path('.').rglob(pattern))
+        # 加载并切割所有视频
+        for i, video_file in enumerate(video_files):
+            status_updates += f"📹 处理第 {i+1}/{len(video_files)} 个视频...\n"
+            
+            clip = VideoFileClip(video_file.name)
+            total_source_duration += clip.duration
+            
+            # 按时长切割视频
+            for start_time_clip in range(0, int(clip.duration), clip_duration):
+                end_time = min(start_time_clip + clip_duration, clip.duration)
+                if end_time - start_time_clip >= 0.5:  # 至少0.5秒的片段
+                    sub_clip = clip.subclip(start_time_clip, end_time).without_audio()
+                    video_clips.append(sub_clip)
+        
+        # 加载音频
+        audio = AudioFileClip(audio_file.name)
+        target_duration = audio.duration
         
-        os.chdir(original_dir)
+        # 智能打乱视频片段
+        random.shuffle(video_clips)
         
-        progress(1.0, desc="處理完成!")
+        status_updates += f"🎵 音频时长: {target_duration:.1f}秒\n"
+        status_updates += f"📊 生成 {len(video_clips)} 个视频片段\n"
+        status_updates += f"🔄 准备输出 {num_output_videos} 个混剪视频...\n"
         
-        if video_files:
-            # 復制視頻到當前目錄以便訪問
-            output_video = str(video_files[0])
-            local_video = f"./output_{os.path.basename(output_video)}"
-            shutil.copy2(os.path.join(repo_dir, output_video), local_video)
+        # 生成输出视频
+        output_files = []
+        clips_per_video = max(1, len(video_clips) // num_output_videos)
+        
+        for i in range(num_output_videos):
+            start_idx = i * clips_per_video
+            end_idx = start_idx + clips_per_video
+            
+            # 最后一个视频包含剩余片段
+            if i == num_output_videos - 1:
+                end_idx = len(video_clips)
             
-            return local_video, f"✅ 成功生成視頻! 找到 {len(video_files)} 個文件。"
-        else:
-            error_msg = f"處理完成但未找到視頻文件。\n\n標準輸出:\n{result.stdout}\n\n錯誤輸出:\n{result.stderr}"
-            return None, error_msg
+            selected_clips = video_clips[start_idx:end_idx]
             
-    except subprocess.TimeoutExpired:
-        os.chdir(original_dir)
-        return None, "❌ 處理超時 (10分鐘)。請嘗試較短的視頻。"
+            if selected_clips:
+                # 拼接视频片段
+                final_video = concatenate_videoclips(selected_clips)
+                
+                # 调整视频长度匹配音频
+                if final_video.duration > target_duration:
+                    final_video = final_video.subclip(0, target_duration)
+                elif final_video.duration < target_duration:
+                    # 循环播放直到匹配音频长度
+                    loops_needed = int(target_duration / final_video.duration) + 1
+                    looped_video = concatenate_videoclips([final_video] * loops_needed)
+                    final_video = looped_video.subclip(0, target_duration)
+                
+                # 添加音频和效果
+                final_video = final_video.set_audio(audio)
+                
+                # 添加镜像效果（防止版权检测）
+                final_video = final_video.fx(vfx.mirror_x)
+                
+                # 保存视频
+                output_filename = f"openai_mixcut_{i+1:02d}.mp4"
+                output_path = os.path.join(output_dir, output_filename)
+                
+                final_video.write_videofile(
+                    output_path,
+                    codec='libx264',
+                    audio_codec='aac',
+                    temp_audiofile='temp-audio.m4a',
+                    remove_temp=True,
+                    verbose=False,
+                    logger=None
+                )
+                
+                output_files.append(output_path)
+                
+                # 清理内存
+                final_video.close()
+        
+        # 清理资源
+        for clip in video_clips:
+            clip.close()
+        audio.close()
+        
+        # 计算处理时间
+        end_time = datetime.now()
+        processing_time = (end_time - start_time).total_seconds()
+        
+        # 生成 AI 标题和总结
+        status_updates += "🎯 生成创意标题...\n"
+        ai_titles = generate_creative_titles(num_output_videos)
+        
+        status_updates += "📋 生成项目总结...\n"
+        project_summary = generate_project_summary(num_output_videos, f"{processing_time:.1f}秒")
+        
+        # 最终状态
+        final_status = f"""✅ 处理完成！OpenAI GPT 增强版
+        
+📊 处理统计：
+• 输入视频: {len(video_files)} 个
+• 生成片段: {len(video_clips)} 个  
+• 输出视频: {len(output_files)} 个
+• 处理时长: {processing_time:.1f} 秒
+• 音频匹配: {target_duration:.1f} 秒
+
+🤖 AI 增强功能已完成：
+• ✓ 内容智能分析
+• ✓ 创意标题生成  
+• ✓ 专业项目总结
+• ✓ 处理建议优化
+
+⏰ 完成时间: {end_time.strftime('%H:%M:%S')}"""
+        
+        return final_status, output_files, video_analysis, ai_titles, project_summary
+        
     except Exception as e:
-        if 'original_dir' in locals():
-            os.chdir(original_dir)
-        return None, f"❌ 處理失敗: {str(e)}"
+        error_msg = f"""❌ 处理出现错误: {str(e)}
 
-# 創建Gradio界面
-with gr.Blocks(title="AI YouTube Shorts Generator", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("""
-    # 🎬 AI YouTube Shorts Generator
-    
-    自動將長視頻轉換為吸引人的YouTube Shorts！
+🔧 可能的解决方案：
+1. 检查视频文件格式是否支持
+2. 确认音频文件完整性
+3. 减少输出视频数量
+4. 检查系统资源使用情况"""
+        
+        return error_msg, [], ai_suggestions, "", ""
+
+# 创建增强版 Gradio 界面
+with gr.Blocks(
+    title="pyTovideo2 + OpenAI GPT", 
+    theme=gr.themes.Soft(),
+    css="""
+    .gradio-container {
+        font-family: 'Arial', sans-serif;
+    }
+    .main-header {
+        text-align: center; 
+        margin-bottom: 25px;
+        padding: 20px;
+        background: linear-gradient(90deg, #667eea 0%, #764ba2 100%);
+        border-radius: 10px;
+        color: white;
+    }
+    """
+) as demo:
     
-    ## ✨ 功能特點
-    - 🎯 **智能內容分析**: 使用GPT-4識別視頻亮點
-    - 🗣️ **說話者檢測**: 自動檢測不同說話者
-    - ✂️ **垂直裁剪**: 轉換為Shorts格式 (9:16)
-    - 🎵 **語音轉錄**: 基於Whisper的高質量轉錄
+    # 主标题区域
+    gr.HTML("""
+    <div class="main-header">
+        <h1 style="margin-bottom: 10px;">🎬 pyTovideo2 + OpenAI GPT</h1>
+        <p style="margin: 0; opacity: 0.9;">AI 驱动的智能视频混剪工具 | GPT-4 内容分析与创意生成</p>
+    </div>
     """)
     
+    # 主要操作区域
     with gr.Row():
+        # 左侧：文件上传和参数设置
         with gr.Column(scale=1):
-            youtube_url = gr.Textbox(
-                label="📺 YouTube視頻URL", 
-                placeholder="https://www.youtube.com/watch?v=dQw4w9WgXcQ",
-                info="輸入要處理的YouTube視頻鏈接"
+            gr.Markdown("### 📤 素材上传")
+            
+            video_input = gr.File(
+                label="🎥 选择视频文件 (支持批量上传)",
+                file_count="multiple",
+                file_types=[".mp4", ".mov", ".avi", ".mkv"],
+                height=130
             )
             
-            openai_key = gr.Textbox(
-                label="🔑 OpenAI API Key", 
-                type="password",
-                placeholder="sk-proj-...",
-                info="需要GPT-4/GPT-4o訪問權限"
+            audio_input = gr.File(
+                label="🎵 选择背景音频",
+                file_types=[".mp3", ".wav", ".m4a", ".aac"],
+                height=90
             )
             
+            gr.Markdown("### ⚙️ 处理参数")
+            
             with gr.Row():
-                process_btn = gr.Button("🚀 開始生成", variant="primary", size="lg")
-                clear_btn = gr.Button("🗑️ 清除", variant="secondary")
+                clip_duration = gr.Number(
+                    label="⏱️ 切片时长(秒)",
+                    value=2,
+                    minimum=1,
+                    maximum=8,
+                    step=0.5,
+                    info="每个视频片段的长度"
+                )
+                
+                num_videos = gr.Number(
+                    label="📊 输出数量",
+                    value=3,
+                    minimum=1,
+                    maximum=6,
+                    step=1,
+                    info="生成混剪视频的数量"
+                )
             
+            # 处理按钮
+            process_btn = gr.Button(
+                "🚀 开始 AI 智能处理",
+                variant="primary",
+                size="lg",
+                scale=1
+            )
+            
+            # AI 模型信息
+            gr.Markdown("""
+            <div style="background: #f0f8ff; padding: 10px; border-radius: 8px; margin-top: 15px;">
+            <b>🤖 AI 模型信息:</b><br>
+            • 内容分析: GPT-4o-mini<br>
+            • 标题生成: GPT-4o-mini<br>  
+            • 项目总结: GPT-4o<br>
+            • 成本优化: 智能模型选择
+            </div>
+            """)
+        
+        # 右侧：状态显示和文件下载
         with gr.Column(scale=1):
-            output_video = gr.Video(label="🎥 生成的Shorts")
-            output_status = gr.Textbox(
-                label="📊 處理狀態", 
-                lines=8, 
-                max_lines=10,
-                show_copy_button=True
+            gr.Markdown("### 📊 处理状态")
+            
+            status_output = gr.Textbox(
+                label="🔥 实时处理状态",
+                lines=8,
+                max_lines=12,
+                interactive=False,
+                show_copy_button=True,
+                placeholder="等待开始处理..."
+            )
+            
+            gr.Markdown("### 📁 输出文件")
+            
+            video_output = gr.File(
+                label="⬇️ 下载混剪视频",
+                file_count="multiple",
+                height=120,
+                interactive=False
             )
     
-    gr.Markdown("""
-    ## 📋 使用說明
-    
-    1. **輸入URL**: 粘貼YouTube視頻鏈接
-    2. **API Key**: 提供OpenAI API密鑰 (需GPT-4權限)
-    3. **開始處理**: 點擊生成按鈕，等待5-10分鐘
-    4. **下載視頻**: 處理完成後下載生成的Shorts
-    
-    ## ⚠️ 注意事項
+    # AI 分析结果展示区域
+    gr.Markdown("---")
+    gr.HTML("<h2 style='text-align: center; color: #4a90e2;'>🧠 OpenAI GPT 智能分析结果</h2>")
     
-    - **處理時間**: 取決於視頻長度，通常5-10分鐘
-    - **視頻要求**: 確保YouTube視頻公開可訪問
-    - **API用量**: 會消耗OpenAI API額度用於內容分析
-    - **格式支持**: 輸出為MP4格式的豎向視頻 (9:16)
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("### 📈 内容分析报告")
+            analysis_output = gr.Textbox(
+                label="🎯 GPT-4 视频内容分析",
+                lines=6,
+                interactive=False,
+                show_copy_button=True,
+                placeholder="AI 分析结果将在这里显示..."
+            )
+        
+        with gr.Column():
+            gr.Markdown("### 🏷️ 创意标题推荐")
+            titles_output = gr.Textbox(
+                label="💡 GPT 生成创意标题",
+                lines=6,
+                interactive=False,
+                show_copy_button=True,
+                placeholder="创意标题将在这里显示..."
+            )
     
-    ---
-    💡 **提示**: 較短的源視頻 (10-30分鐘) 處理效果更佳
-    """)
+    # 项目总结区域
+    gr.Markdown("### 📋 AI 项目分析总结")
+    summary_output = gr.Textbox(
+        label="📊 GPT-4 专业项目总结",
+        lines=5,
+        interactive=False,
+        show_copy_button=True,
+        placeholder="详细的项目分析总结将在这里显示..."
+    )
     
-    # 事件綁定
+    # 绑定处理函数
     process_btn.click(
-        fn=process_video,
-        inputs=[youtube_url, openai_key],
-        outputs=[output_video, output_status],
+        fn=process_videos_with_openai,
+        inputs=[video_input, audio_input, clip_duration, num_videos],
+        outputs=[status_output, video_output, analysis_output, titles_output, summary_output],
         show_progress=True
     )
     
-    clear_btn.click(
-        lambda: ("", "", None, ""),
-        outputs=[youtube_url, openai_key, output_video, output_status]
-    )
+    # 使用指南
+    gr.Markdown("""
+    ---
+    ### 📖 使用指南
+    
+    **🎯 支持格式:**
+    - 视频: MP4, MOV, AVI, MKV
+    - 音频: MP3, WAV, M4A, AAC
+    
+    **💾 建议规格:**
+    - 单个视频 < 200MB
+    - 音频文件 < 50MB  
+    - 总处理时间 < 5分钟
+    
+    **🤖 AI 功能特色:**
+    - ✨ GPT-4 智能内容分析
+    - 🎨 创意标题自动生成
+    - 📊 专业项目评估报告
+    - 💡 个性化优化建议
+    
+    **⚡ 处理流程:**
+    1. 上传视频和音频文件
+    2. 调整处理参数
+    3. 点击开始处理
+    4. 等待 AI 分析完成
+    5. 下载结果文件
+    """)
 
+# 启动应用
 if __name__ == "__main__":
     demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        show_error=True
+        share=False,
+        show_error=True,
+        show_tips=True
     )