init

app.py

@@ -0,0 +1,552 @@
+import gradio as gr
+import time
+import uuid
+import re
+from util import (
+    create_task_v3, 
+    get_task_result,
+)
+
+
+IP_Dict = {}
+
+# 支持的语言列表
+SUPPORTED_LANGUAGES = [
+    {"flag": "🇺🇸", "name": "English", "code": "en"},
+    {"flag": "🇨🇳", "name": "中文", "code": "zh"},
+    {"flag": "🇪🇸", "name": "Español", "code": "es"},
+    {"flag": "🇫🇷", "name": "Français", "code": "fr"},
+    {"flag": "🇩🇪", "name": "Deutsch", "code": "de"},
+    {"flag": "🇮🇹", "name": "Italiano", "code": "it"},
+    {"flag": "🇯🇵", "name": "日本語", "code": "ja"},
+    {"flag": "🇰🇷", "name": "한국어", "code": "ko"},
+    {"flag": "🇷🇺", "name": "Русский", "code": "ru"}
+]
+
+def detect_language(text):
+    """
+    语言检测函数 - 只检测容易区分的语言
+    """
+    if len(text) < 5:
+        return None
+    
+    # 语言特征字符
+    language_features = {
+        "zh": re.compile(r'[\u4e00-\u9fff\u3400-\u4dbf\uff00-\uffef\u3000-\u303f]'),
+        "ja": re.compile(r'[\u3040-\u309f\u30a0-\u30ff]'),
+        "ko": re.compile(r'[\uac00-\ud7af]'),
+        "ru": re.compile(r'[\u0400-\u04ff]')
+    }
+    
+    scores = {lang: 0 for lang in language_features.keys()}
+    
+    # 字符特征检测 - 只检测明显的非拉丁字符
+    for char in text:
+        for lang, pattern in language_features.items():
+            if pattern.search(char):
+                scores[lang] += 1
+    
+    # 找到得分最高的语言
+    max_score = 0
+    detected_lang = None
+    
+    for lang, score in scores.items():
+        if score > max_score and score > 0:
+            max_score = score
+            detected_lang = lang
+    
+    # 检查是否满足阈值 - 如果有明显的非拉丁字符就切换
+    if detected_lang and max_score >= 2:
+        return detected_lang
+    
+    # 没有检测到其他语言，默认保持英语
+    return None
+
+def get_language_display_text(language_code):
+    """获取语言的显示文本（国旗 + 名称）"""
+    for lang in SUPPORTED_LANGUAGES:
+        if lang["code"] == language_code:
+            return f"{lang['flag']} {lang['name']}"
+    return "🇺🇸 English"
+
+def get_language_code_from_display(display_text):
+    """从显示文本中提取语言代码"""
+    for lang in SUPPORTED_LANGUAGES:
+        if f"{lang['flag']} {lang['name']}" == display_text:
+            return lang['code']
+    return "en"
+
+def auto_detect_language(text, current_language_display):
+    """
+    自动检测语言并返回更新后的语言选择
+    """
+    if not text or len(text.strip()) < 5:
+        return current_language_display
+    
+    current_code = get_language_code_from_display(current_language_display)
+    detected_lang = detect_language(text)
+    
+    if detected_lang and detected_lang != current_code:
+        print(f"Language auto-detected: {current_code} -> {detected_lang}")
+        return get_language_display_text(detected_lang)
+    
+    return current_language_display
+
+def generate_obama_voice_with_realtime_updates(text, language_display, request: gr.Request):
+    """
+    Obama AI voice generation function with real-time status updates
+    """
+    # 从显示文本中提取语言代码
+    language = get_language_code_from_display(language_display)
+    
+    client_ip = request.client.host
+    x_forwarded_for = dict(request.headers).get('x-forwarded-for')
+    if x_forwarded_for:
+        client_ip = x_forwarded_for   
+    if client_ip not in IP_Dict:
+        IP_Dict[client_ip] = 0
+    IP_Dict[client_ip] += 1
+    print(f"client_ip: {client_ip}, count: {IP_Dict[client_ip]}")
+    if IP_Dict[client_ip] >= 6:
+        msg = "You have reached the maximum number of requests"
+        # Create "Get More Tries" button HTML
+        get_more_tries_html = f"""
+        <div style='display: flex; justify-content: center; gap: 30px; margin: 10px 0 25px 0; padding: 0px;'>
+            <a href='https://trumpaivoice.net/#generator' target='_blank' style='
+                display: inline-flex; 
+                align-items: center;
+                justify-content: center;
+                padding: 16px 32px; 
+                background: linear-gradient(135deg, #11998e 0%, #38ef7d 100%);
+                color: white; 
+                text-decoration: none; 
+                border-radius: 12px; 
+                font-weight: 600;
+                font-size: 16px;
+                text-align: center;
+                min-width: 160px;
+                box-shadow: 0 4px 15px rgba(17, 153, 142, 0.4);
+                transition: all 0.3s ease;
+                border: none;
+            '>🚀 Get More Tries for Free</a>
+        </div>
+        """
+        yield msg, None, gr.update(value=get_more_tries_html, visible=True), ""
+        return msg, None, gr.update(value=get_more_tries_html, visible=True), ""
+
+    if not text or len(text.strip()) < 3:
+        return "Text too short, please enter at least 3 characters", None, gr.update(visible=False), ""
+    
+    try:
+        task_type = "voice"
+        word_num = len(text.strip().split())  # 自动计算单词数
+        
+        # Create task
+        task_result = create_task_v3(task_type, text.strip(), word_num, is_rewrite=False, language=language)
+        if not task_result:
+            return "Failed to create task", None, gr.update(visible=False), ""
+        else:
+            yield "Task created successfully", None, gr.update(visible=False), ""
+
+        max_polls = 300
+        poll_interval = 1
+        task_url = f"https://trumpaivoice.net/task/{task_result['uuid']}"
+        
+        for i in range(max_polls):
+            time.sleep(poll_interval)
+            task = get_task_result(task_result['uuid'])
+            # print(task, i, "get_task_result")
+            if task.get('data', {}):
+                status = task.get('data').get('status', '')
+                text_final = task.get('data').get('text_final', '')
+                if status in ['completed',]:
+                    voice_url = task.get('data').get('voice_url', '')
+                    print(voice_url, "===>voice_url")
+                    
+                    # 下载音频文件到本地以避免SSRF保护问题
+                    local_audio_path = voice_url
+                    
+                    # Create action buttons HTML
+                    action_buttons_html = f"""
+                    <div style='display: flex; justify-content: center; gap: 30px; margin: 25px 0; padding: 20px;'>
+                        <a href='https://trumpaivoice.net/#generator' target='_blank' style='
+                            display: inline-flex; 
+                            align-items: center;
+                            justify-content: center;
+                            padding: 16px 32px; 
+                            background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                            color: white; 
+                            text-decoration: none; 
+                            border-radius: 12px; 
+                            font-weight: 600;
+                            font-size: 16px;
+                            text-align: center;
+                            min-width: 160px;
+                            box-shadow: 0 4px 15px rgba(102, 126, 234, 0.4);
+                            transition: all 0.3s ease;
+                            border: none;
+                        '>🎬 Generate 4K Video</a>
+                        <a href='{task_url}' target='_blank' style='
+                            display: inline-flex; 
+                            align-items: center;
+                            justify-content: center;
+                            padding: 16px 32px; 
+                            background: linear-gradient(135deg, #11998e 0%, #38ef7d 100%);
+                            color: white; 
+                            text-decoration: none; 
+                            border-radius: 12px; 
+                            font-weight: 600;
+                            font-size: 16px;
+                            text-align: center;
+                            min-width: 160px;
+                            box-shadow: 0 4px 15px rgba(17, 153, 142, 0.4);
+                            transition: all 0.3s ease;
+                            border: none;
+                        '>👀 Check Generate Details</a>
+                    </div>
+                    """
+                    yield f"✅ success!!!", local_audio_path, gr.update(value=action_buttons_html, visible=True), task_url
+                    return "✅ Generation successful!", local_audio_path, gr.update(value=action_buttons_html, visible=True), task_url
+                elif status in ['failed', 'voice_error', 'no_credits']:
+                    yield "❌ Generation failed!", None, gr.update(visible=False), ""
+                    return "❌ Generation failed!", None, gr.update(visible=False), ""
+                else:
+                    yield f"query {i} times, on processing, go to task page {task_url} to check status", None, gr.update(visible=False), task_url
+        return "❌ Generation failed!", None, gr.update(visible=False), ""
+    except Exception as e:
+        error_msg = f"Generation failed: {str(e)}"
+        yield error_msg, None, gr.update(visible=False), ""
+        return error_msg, None, gr.update(visible=False), ""
+
+# Create Gradio Interface
+with gr.Blocks(title="Barack Obama AI Voice", theme=gr.themes.Soft()) as demo:
+    
+    # Main title - at the top
+    gr.HTML("""
+    <div style="text-align: center; margin: 5px auto 0px auto; max-width: 800px;">
+        <h1 style="color: #2c3e50; margin: 0; font-size: 3.5em; font-weight: 800; letter-spacing: 3px; text-shadow: 2px 2px 4px rgba(0,0,0,0.1);">
+            🎤 Obama AI Voice
+        </h1>
+    </div>
+    """, padding=False)
+    
+    # # Showcase link banner - second
+    # gr.HTML("""
+    # <div style="text-align: center; margin: 0px auto 40px auto; max-width: 600px;">
+    #     <div style="background: linear-gradient(135deg, #ff6b6b 0%, #feca57 50%, #48dbfb 100%); padding: 15px 5px; border-radius: 15px; box-shadow: 0 8px 25px rgba(255, 107, 107, 0.3);">
+    #         <h3 style="color: white; margin: 0; font-size: 18px;">
+    #             🎬 <a href="https://trumpaivoice.net/showcase" target="_blank" style="color: white; text-decoration: none; font-weight: bold;">Check out Obama AI videos created by others →</a>
+    #         </h3>
+    #     </div>
+    # </div>
+    # """, padding=False)
+
+    # Powered by link - small text
+    gr.HTML("""
+    <div style="text-align: center; margin: 0px auto -5px auto;">
+        <p style="margin: 0; font-size: 16px; color: #999; font-weight: 400;">
+            powered by <a href="https://trumpaivoice.net/" target="_blank" style="color: #667eea; text-decoration: none;">trumpaivoice.net</a>
+        </p>
+    </div>
+    """, padding=False)
+    
+    # Simple description text - third
+    # gr.HTML("""
+    # <div style="text-align: center; margin: 15px auto 30px auto; max-width: 500px;">
+    #     <p style="color: #666; margin: 0; font-size: 1em; font-weight: 500; line-height: 1.4;">
+    #         🔥 Try the most advanced Obama AI Voice and Video generator for FREE at 
+    #         <a href="https://trumpaivoice.net/" target="_blank" style="color: #667eea; text-decoration: none; font-weight: bold;">donaldtrumpaivoice.com</a>!
+    #     </p>
+    # </div>
+    # """)
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            text_input = gr.Textbox(
+                label="📝 Input Text",
+                lines=4,
+                placeholder="Enter what you want Obama to say...",
+                value="Hello everyone, this is a demonstration of the Obama AI Voice system with real-time status monitoring."
+            )
+        
+        with gr.Column(scale=1):
+            # 创建语言选择器的选项
+            language_choices = [f"{lang['flag']} {lang['name']}" for lang in SUPPORTED_LANGUAGES]
+            language_values = [lang['code'] for lang in SUPPORTED_LANGUAGES]
+            
+            language_dropdown = gr.Dropdown(
+                choices=language_choices,
+                value="🇺🇸 English",  # 默认值
+                label="🌍 Language",
+                interactive=True
+            )
+
+            submit_btn = gr.Button(
+                "🚀 Generate Obama AI Voice",
+                variant="primary",
+                size="lg"
+            )
+    
+    with gr.Row():
+        status_output = gr.Textbox(
+            label="📊 Status",
+            interactive=False,
+            placeholder="Waiting for generation..."
+        )
+    
+    # Action buttons that will show after task completion
+    with gr.Row():
+        action_links = gr.HTML(visible=False)
+    
+    with gr.Row():
+        audio_output = gr.Audio(
+            label="🎵 Obama AI Voice",
+            interactive=False
+        )
+    
+    
+    # Video Showcase Section
+    gr.HTML("""
+    <div style="width: 100%; margin: 40px 0; padding: 0 20px;">
+        <div style="text-align: center; background: linear-gradient(135deg, #e3f2fd 0%, #f3e5f5 50%, #fce4ec 100%); padding: 30px; border-radius: 20px; box-shadow: 0 10px 30px rgba(0,0,0,0.1); margin-bottom: 30px;">
+            <h2 style="color: #2c3e50; margin: 0 0 15px 0; font-size: 2.2em; font-weight: 700;">
+                🎬 Video Showcase
+            </h2>
+            <p style="color: #34495e; font-size: 1.2em; line-height: 1.6; margin: 0 0 25px 0;">
+                Watch amazing AI-generated Obama videos with realistic animations and perfect lip-sync!
+            </p>
+            
+            <!-- Video Gallery -->
+            <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(280px, 1fr)); gap: 20px; margin: 30px 0;">
+                
+                <div style="background: rgba(255,255,255,0.8); padding: 20px; border-radius: 15px; box-shadow: 0 4px 15px rgba(0,0,0,0.1);">
+                    <h4 style="color: #2c3e50; margin: 0 0 15px 0; font-size: 1.1em;">🌺 White House Garden</h4>
+                    <video id="video-garden" style="width: 100%; height: 300px; border-radius: 10px; object-fit: cover;" controls preload="metadata">
+                        <source src="https://www.trumpaivoice.net/SelfitAssert/Heygem/Obama/show/62_1758155530_8uRFAtYB-r.mp4" type="video/mp4">
+                        Your browser does not support the video tag.
+                    </video>
+                </div>
+                
+                <div style="background: rgba(255,255,255,0.8); padding: 20px; border-radius: 15px; box-shadow: 0 4px 15px rgba(0,0,0,0.1);">
+                    <h4 style="color: #2c3e50; margin: 0 0 15px 0; font-size: 1.1em;">🏛️ Oval Office Address</h4>
+                    <video id="video-office" style="width: 100%; height: 300px; border-radius: 10px; object-fit: cover;" controls preload="metadata">
+                        <source src="https://www.trumpaivoice.net/SelfitAssert/Heygem/Obama/show/62_1758155476_KdNzaG9p-r.mp4" type="video/mp4">
+                        Your browser does not support the video tag.
+                    </video>
+                </div>
+
+            </div>
+            
+            <!-- Action Buttons -->
+            <div style="display: flex; justify-content: center; gap: 20px; flex-wrap: wrap; margin: 25px 0 0 0;">
+                <a href="https://trumpaivoice.net/obama-ai-voice#generator" target="_blank" style="
+                    display: inline-flex; 
+                    align-items: center;
+                    justify-content: center;
+                    padding: 18px 35px; 
+                    background: linear-gradient(135deg, #ff6b6b 0%, #feca57 100%);
+                    color: white; 
+                    text-decoration: none; 
+                    border-radius: 15px; 
+                    font-weight: 700;
+                    font-size: 16px;
+                    text-align: center;
+                    min-width: 220px;
+                    box-shadow: 0 8px 25px rgba(255, 107, 107, 0.4);
+                    transition: all 0.3s ease;
+                    border: none;
+                    transform: scale(1);
+                " onmouseover="this.style.transform='scale(1.05)'" onmouseout="this.style.transform='scale(1)'">
+                    🎬 Generate 4K Obama Video
+                </a>
+                
+            </div>
+            
+            <p style="color: #555; font-size: 1em; margin: 15px 0 0 0;">
+                Create high-quality Obama AI videos and original songs with custom text and scenarios!
+            </p>
+        </div>
+    </div>
+    
+    <script>
+        // Ensure only one video plays at a time
+        document.addEventListener('DOMContentLoaded', function() {
+            const videos = ['video-garden', 'video-office', 'video-flag'];
+            
+            videos.forEach(function(videoId) {
+                const video = document.getElementById(videoId);
+                if (video) {
+                    video.addEventListener('play', function() {
+                        // Pause all other videos when this one starts playing
+                        videos.forEach(function(otherId) {
+                            if (otherId !== videoId) {
+                                const otherVideo = document.getElementById(otherId);
+                                if (otherVideo && !otherVideo.paused) {
+                                    otherVideo.pause();
+                                }
+                            }
+                        });
+                    });
+                }
+            });
+        });
+    </script>
+    """, padding=False)
+
+    # Comprehensive introduction section
+    gr.HTML("""
+    <div style="width: 100%; margin: 30px 0; padding: 0 20px;">
+        
+        <!-- Hero Description -->
+        <div style="text-align: center; margin: 25px auto; background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%); padding: 30px; border-radius: 20px; box-shadow: 0 10px 30px rgba(0,0,0,0.1);">
+            <h2 style="color: #2c3e50; margin: 0 0 15px 0; font-size: 1.8em; font-weight: 700;">
+                🇺🇸 Experience the Power of AI-Generated Obama Voice
+            </h2>
+                         <p style="color: #555; font-size: 1.1em; line-height: 1.6; margin: 0 0 20px 0; width: 100%; padding: 0 20px;">
+                 Transform any text into authentic Barack Obama speech with our cutting-edge AI voice synthesis technology. 
+                 Whether you're creating content for entertainment, education, or social media, our advanced neural network 
+                 captures Obama's distinctive speaking style, intonation, and rhetorical patterns with remarkable accuracy.
+             </p>
+        </div>
+
+        <!-- Features Grid -->
+        <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(280px, 1fr)); gap: 20px; margin: 40px 0;">
+            
+            <div style="background: white; padding: 25px; border-radius: 15px; box-shadow: 0 5px 20px rgba(0,0,0,0.08); border-left: 5px solid #e74c3c;">
+                <h3 style="color: #e74c3c; margin: 0 0 12px 0; font-size: 1.3em; font-weight: 600;">
+                    🎯 Ultra-Realistic Voice
+                </h3>
+                <p style="color: #666; margin: 0; line-height: 1.5; font-size: 0.95em;">
+                    Our AI model is trained on thousands of hours of Obama speeches, capturing his unique vocal characteristics, 
+                    pronunciation patterns, and speaking rhythm to deliver incredibly lifelike results.
+                </p>
+            </div>
+
+            <div style="background: white; padding: 25px; border-radius: 15px; box-shadow: 0 5px 20px rgba(0,0,0,0.08); border-left: 5px solid #3498db;">
+                <h3 style="color: #3498db; margin: 0 0 12px 0; font-size: 1.3em; font-weight: 600;">
+                    ⚡ Lightning Fast Generation
+                </h3>
+                <p style="color: #666; margin: 0; line-height: 1.5; font-size: 0.95em;">
+                    Generate high-quality Obama AI voice clips in seconds, not minutes. Our optimized infrastructure 
+                    ensures rapid processing while maintaining exceptional audio quality.
+                </p>
+            </div>
+
+            <div style="background: white; padding: 25px; border-radius: 15px; box-shadow: 0 5px 20px rgba(0,0,0,0.08); border-left: 5px solid #27ae60;">
+                <h3 style="color: #27ae60; margin: 0 0 12px 0; font-size: 1.3em; font-weight: 600;">
+                    🎨 Creative Content Creation
+                </h3>
+                <p style="color: #666; margin: 0; line-height: 1.5; font-size: 0.95em;">
+                    Perfect for memes, podcasts, educational content, entertainment videos, or any creative project 
+                    that needs an authentic Obama voice performance.
+                </p>
+            </div>
+
+        </div>
+
+        
+                 <!-- Celebrity Voices Section -->
+         <div style="background: linear-gradient(135deg, #ff6b6b 0%, #feca57 50%, #48dbfb 100%); color: white; padding: 40px; border-radius: 20px; margin: 40px 0; text-align: center;">
+             <h2 style="margin: 0 0 20px 0; font-size: 1.8em; font-weight: 700;">
+                 🎭 Try More Celebrity AI Voices
+             </h2>
+             <p style="margin: 0 0 25px 0; font-size: 1.1em; opacity: 0.95; line-height: 1.5;">
+                 Explore our premium collection of celebrity AI voices! Our high-quality service delivers 
+                 lightning-fast results with exceptional audio quality. Experience the best AI voice generation 
+                 with our reliable and responsive platform.
+             </p>
+             <div style="display: flex; justify-content: center; gap: 20px; flex-wrap: wrap;">
+                 <a href="https://trumpaivoice.net/explore" target="_blank" style="
+                     display: inline-flex; 
+                     align-items: center;
+                     justify-content: center;
+                     padding: 18px 35px; 
+                     background: rgba(255,255,255,0.9);
+                     color: #333; 
+                     text-decoration: none; 
+                     border-radius: 15px; 
+                     font-weight: 700;
+                     font-size: 16px;
+                     text-align: center;
+                     min-width: 200px;
+                     box-shadow: 0 6px 20px rgba(0,0,0,0.3);
+                     transition: all 0.3s ease;
+                     border: none;
+                 ">🌟 Explore Celebrity Voices</a>
+                 <a href="https://trumpaivoice.net/showcase" target="_blank" style="
+                     display: inline-flex; 
+                     align-items: center;
+                     justify-content: center;
+                     padding: 18px 35px; 
+                     background: rgba(255,255,255,0.2);
+                     color: white; 
+                     text-decoration: none; 
+                     border-radius: 15px; 
+                     font-weight: 700;
+                     font-size: 16px;
+                     text-align: center;
+                     min-width: 200px;
+                     box-shadow: 0 6px 20px rgba(0,0,0,0.2);
+                     transition: all 0.3s ease;
+                     border: 2px solid rgba(255,255,255,0.3);
+                 ">🎭 View Showcase</a>
+             </div>
+         </div>
+
+        <!-- Tips Section -->
+        <div style="background: linear-gradient(135deg, #ff9a9e 0%, #fecfef 50%, #fecfef 100%); padding: 25px; border-radius: 15px; margin: 40px 0;">
+            <h3 style="color: #8b5cf6; text-align: center; margin: 0 0 20px 0; font-size: 1.4em; font-weight: 700;">
+                💡 Pro Tips for Best Results
+            </h3>
+            <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(280px, 1fr)); gap: 15px;">
+                
+                <div style="background: rgba(255,255,255,0.8); padding: 15px; border-radius: 10px;">
+                    <strong style="color: #8b5cf6;">📖 Clear Text:</strong>
+                    <span style="color: #555;"> Use proper punctuation and avoid special characters for optimal results.</span>
+                </div>
+
+                <div style="background: rgba(255,255,255,0.8); padding: 15px; border-radius: 10px;">
+                    <strong style="color: #8b5cf6;">⏱️ Length Matters:</strong>
+                    <span style="color: #555;"> Shorter texts (20-60 words) typically produce the most natural-sounding results.</span>
+                </div>
+
+                <div style="background: rgba(255,255,255,0.8); padding: 15px; border-radius: 10px;">
+                    <strong style="color: #8b5cf6;">🎯 Obama Style:</strong>
+                    <span style="color: #555;"> Text written in Obama's speaking style will sound more authentic and natural.</span>
+                </div>
+
+            </div>
+        </div>
+
+    </div>
+    """, padding=False)
+    
+
+    # Powered by link - small text
+    gr.HTML("""
+    <div style="text-align: center; margin: 0px auto -5px auto;">
+        <p style="margin: 0; font-size: 16px; color: #999; font-weight: 400;">
+            Click <a href="https://trumpaivoice.net/showcase" target="_blank" style="color: #667eea; text-decoration: none;"> obama ai voices showcase </a> to see more videos
+        </p>
+    </div>
+    """, padding=False)
+
+    # Hidden state to store task_url
+    task_url_state = gr.State("")
+
+    # 自动语言检测事件
+    text_input.change(
+        auto_detect_language,
+        inputs=[text_input, language_dropdown],
+        outputs=[language_dropdown]
+    )
+
+    # Bind event
+    submit_btn.click(
+        generate_obama_voice_with_realtime_updates,
+        inputs=[text_input, language_dropdown],
+        outputs=[status_output, audio_output, action_links, task_url_state]
+    )
+
+if __name__ == "__main__":
+    demo.launch() 

push.sh

@@ -0,0 +1,13 @@
+
+# 设置仓库级别用户名
+git config user.name "selfitcamera"
+git config user.email "ethan.blake@heybeauty.ai"
+
+# 验证
+git config user.name
+git config user.email
+
+
+git add .
+git commit -m "init"
+git push

requirements.txt

@@ -0,0 +1,4 @@
+gradio>=4.0.0
+requests>=2.31.0
+supabase>=2.0.0
+python-dotenv>=1.0.0 

util.py

@@ -0,0 +1,119 @@
+import requests
+import time
+import uuid
+import os
+from datetime import datetime
+from supabase import create_client, Client
+
+try:
+    OneKey = os.environ['OneKey'].strip()
+
+    OneKey = OneKey.split("#")
+    TrumpAiUrl = OneKey[0]
+    ApiKey = OneKey[1]
+    SUPABASE_URL = OneKey[2]
+    UserUuid = OneKey[3]
+    BackendUrl = OneKey[4]
+    BackendApiKey = OneKey[5]
+    SUPABASE_KEY = OneKey[6]
+    # GRADIO_ALLOWED_HOSTNAMES = OneKey[7]
+    # os.environ["GRADIO_ALLOWED_HOSTNAMES"] = GRADIO_ALLOWED_HOSTNAMES
+except Exception as e:
+    print(f"OneKey: {e}")
+    # exit(1)
+
+
+# 创建Supabase客户端
+supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
+
+# 任务状态枚举
+class TaskStatus:
+    Created = "created"
+    Processing = "processing"
+    TextRewrited = "text_rewrited"
+    TextFormated = "text_formated"
+    VoiceCompleted = "voice_completed"
+    VoiceError = "voice_error"
+    VideoCompleted = "video_completed"
+    VideoPublished = "video_published"
+    VideoError = "video_error"
+    Completed = "completed"
+    Failed = "failed"
+    Cancelled = "cancelled"
+    NoCredits = "no_credits"
+
+# 视频模板枚举
+class VideoTemplate:
+    Outdoor = "outdoor"
+    UsFlag = "us-flag"
+    WhiteHouse = "white-house"
+
+def create_task_v3(task_type, text, word_num, is_rewrite, language="en"):
+    import json
+    is_rewrite = False
+    url = f"{BackendUrl}/trump_process_ctx_api_v3"
+    headers = {
+        "Content-Type": "application/json"
+    }
+    print(url)
+    data = {
+        "video_template": "",
+        "speaker_template": "https://www.trumpaivoice.net/SelfitAssert/Heygem/Obama/obama_audio01.MP3",
+        "text": text,
+        "word_num": word_num,
+        "is_rewrite": False,
+        "language": language,
+        "watermark": True,
+        "type": "voice",
+        "cost_credits": 2,
+        "user_uuid": UserUuid,
+        "secret_key": "219ngu"
+    }
+    try:
+        resp = requests.post(url, headers=headers, data=json.dumps(data), timeout=60)
+        if not resp.ok:
+            print(f"调用trump_process_ctx_api失败: {resp.status_code} {resp.text}")
+            return None
+        try:
+            ctx_json = resp.json()
+        except Exception as e:
+            print(f"解析trump_process_ctx_api返回异常: {e}")
+            return None
+        if not ctx_json or ctx_json.get("code") != 0 or not ctx_json.get("data") or not ctx_json["data"].get("task_id"):
+            print(f"trump_process_ctx_api返回异常: {ctx_json}")
+            return None
+        return {
+            "task_id": ctx_json["data"]["task_id"],
+            "uuid": ctx_json["data"]["task_uuid"],
+            "status": "created",
+            "message": "任务创建成功，后台处理中"
+        }
+    except Exception as err:
+        print(f"create_task_v3异常: {err}")
+        return None
+
+def get_task_result(task_id):
+    # Poll for task status and result
+    url = f"{TrumpAiUrl}/api/task-status/uuid/{task_id}"
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {ApiKey}"
+    }
+    print(url)
+    try:
+        resp = requests.get(url, headers=headers, timeout=30)
+        resp.raise_for_status()
+        result = resp.json()
+        return result
+    except Exception as e:
+        return {}
+
+
+if __name__ == "__main__":
+
+    task_type = "voice"
+    text = "Hello, this is a test message for Trump AI Voice."
+    word_num = 10
+    is_rewrite = True
+    task_result = create_task_v3(task_type, text, word_num, is_rewrite)
+    print(f"task_result: {task_result}")