init

.gitattributes

@@ -1,35 +1,35 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,14 +1,14 @@
----
-title: Trump Ai Voice
-emoji: 📉
-colorFrom: blue
-colorTo: purple
-sdk: gradio
-sdk_version: 5.37.0
-app_file: app.py
-pinned: false
-license: mit
-short_description: Trump Ai Voice Generator
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: Trump Ai Voice
+emoji: 📉
+colorFrom: blue
+colorTo: purple
+sdk: gradio
+sdk_version: 5.37.0
+app_file: app.py
+pinned: false
+license: mit
+short_description: Trump Ai Voice Generator
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,237 +1,237 @@
-import gradio as gr
-import time
-import uuid
-from util import (
-    create_task_v3, 
-    get_task_result,
-)
-
-
-IP_Dict = {}
-
-def generate_trump_voice_with_realtime_updates(text, word_num, request: gr.Request):
-    """
-    Trump AI voice generation function with real-time status updates
-    """
-    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] > 2:
-        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://donaldtrumpaivoice.com/#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, "No task information", gr.update(visible=False), ""
-    
-    try:
-        task_type = "voice"
-        
-        # Create task
-        task_result = create_task_v3(task_type, text.strip(), word_num, is_rewrite=False)
-        if not task_result:
-            return "Failed to create task", None, "Task creation failed", gr.update(visible=False), ""
-        else:
-            yield "Task created successfully", None, "Task creation successful", gr.update(visible=False), ""
-
-        max_polls = 300
-        poll_interval = 1
-        task_url = f"https://donaldtrumpaivoice.com/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', '')
-                    # 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://donaldtrumpaivoice.com/#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 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!!!", voice_url, text_final, gr.update(value=action_buttons_html, visible=True), task_url
-                    return "✅ Generation successful!", voice_url, "success", gr.update(value=action_buttons_html, visible=True), task_url
-                elif status in ['failed', 'voice_error', 'no_credits']:
-                    yield "❌ Generation failed!", None, None, gr.update(visible=False), ""
-                    return "❌ Generation failed!", None, None, gr.update(visible=False), ""
-                else:
-                    yield f"query {i} times, on processing, go to task page {task_url} to check status", None, text_final, gr.update(visible=False), task_url
-        return "❌ Generation failed!", None, None, gr.update(visible=False), ""
-    except Exception as e:
-        error_msg = f"Generation failed: {str(e)}"
-        yield error_msg, None, f"❌ Error message: {error_msg}", gr.update(visible=False), ""
-        return error_msg, None, f"❌ Error message: {error_msg}", gr.update(visible=False), ""
-
-# Create Gradio Interface
-with gr.Blocks(title="Donald Trump 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);">
-            🎤 Trump 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://donaldtrumpaivoice.com/showcase" target="_blank" style="color: white; text-decoration: none; font-weight: bold;">Check out Trump 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://donaldtrumpaivoice.com/" target="_blank" style="color: #667eea; text-decoration: none;">donaldtrumpaivoice.com</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 Trump AI Voice and Video generator for FREE at 
-    #         <a href="https://donaldtrumpaivoice.com/" 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 Trump to say...",
-                value="Hello everyone, this is a demonstration of the Trump AI Voice system with real-time status monitoring."
-            )
-        
-        with gr.Column(scale=1):
-            word_num_slider = gr.Slider(
-                10, 30, value=20, step=1,
-                label="⏱️ Duration (Word Count)"
-            )
-
-            submit_btn = gr.Button(
-                "🚀 Generate Trump 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="🎵 Trump AI Voice",
-            interactive=False
-        )
-    
-    with gr.Row():
-        task_info = gr.Textbox(
-            label="📋 AI Rewritten Text with Latest News",
-            interactive=False,
-            lines=12,
-            placeholder="AI rewritten text with the latest news will be shown here..."
-        )
-    
-    # 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://donaldtrumpaivoice.com/showcase" target="_blank" style="color: #667eea; text-decoration: none;"> trump ai voices showcase </a> to see more videos
-        </p>
-    </div>
-    """, padding=False)
-
-    # Hidden state to store task_url
-    task_url_state = gr.State("")
-
-    # Bind event
-    submit_btn.click(
-        generate_trump_voice_with_realtime_updates,
-        inputs=[text_input, word_num_slider],
-        outputs=[status_output, audio_output, task_info, action_links, task_url_state]
-    )
-
-if __name__ == "__main__":
+import gradio as gr
+import time
+import uuid
+from util import (
+    create_task_v3, 
+    get_task_result,
+)
+
+
+IP_Dict = {}
+
+def generate_trump_voice_with_realtime_updates(text, word_num, request: gr.Request):
+    """
+    Trump AI voice generation function with real-time status updates
+    """
+    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] > 3:
+        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://donaldtrumpaivoice.com/#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, "No task information", gr.update(visible=False), ""
+    
+    try:
+        task_type = "voice"
+        
+        # Create task
+        task_result = create_task_v3(task_type, text.strip(), word_num, is_rewrite=False)
+        if not task_result:
+            return "Failed to create task", None, "Task creation failed", gr.update(visible=False), ""
+        else:
+            yield "Task created successfully", None, "Task creation successful", gr.update(visible=False), ""
+
+        max_polls = 300
+        poll_interval = 1
+        task_url = f"https://donaldtrumpaivoice.com/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', '')
+                    # 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://donaldtrumpaivoice.com/#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 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!!!", voice_url, text_final, gr.update(value=action_buttons_html, visible=True), task_url
+                    return "✅ Generation successful!", voice_url, "success", gr.update(value=action_buttons_html, visible=True), task_url
+                elif status in ['failed', 'voice_error', 'no_credits']:
+                    yield "❌ Generation failed!", None, None, gr.update(visible=False), ""
+                    return "❌ Generation failed!", None, None, gr.update(visible=False), ""
+                else:
+                    yield f"query {i} times, on processing, go to task page {task_url} to check status", None, text_final, gr.update(visible=False), task_url
+        return "❌ Generation failed!", None, None, gr.update(visible=False), ""
+    except Exception as e:
+        error_msg = f"Generation failed: {str(e)}"
+        yield error_msg, None, f"❌ Error message: {error_msg}", gr.update(visible=False), ""
+        return error_msg, None, f"❌ Error message: {error_msg}", gr.update(visible=False), ""
+
+# Create Gradio Interface
+with gr.Blocks(title="Donald Trump 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);">
+            🎤 Trump 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://donaldtrumpaivoice.com/showcase" target="_blank" style="color: white; text-decoration: none; font-weight: bold;">Check out Trump 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://donaldtrumpaivoice.com/" target="_blank" style="color: #667eea; text-decoration: none;">donaldtrumpaivoice.com</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 Trump AI Voice and Video generator for FREE at 
+    #         <a href="https://donaldtrumpaivoice.com/" 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 Trump to say...",
+                value="Hello everyone, this is a demonstration of the Trump AI Voice system with real-time status monitoring."
+            )
+        
+        with gr.Column(scale=1):
+            word_num_slider = gr.Slider(
+                10, 50, value=30, step=1,
+                label="⏱️ Duration (Word Count)"
+            )
+
+            submit_btn = gr.Button(
+                "🚀 Generate Trump 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="🎵 Trump AI Voice",
+            interactive=False
+        )
+    
+    with gr.Row():
+        task_info = gr.Textbox(
+            label="📋 AI Rewritten Text with Latest News",
+            interactive=False,
+            lines=12,
+            placeholder="AI rewritten text with the latest news will be shown here..."
+        )
+    
+    # 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://donaldtrumpaivoice.com/showcase" target="_blank" style="color: #667eea; text-decoration: none;"> trump ai voices showcase </a> to see more videos
+        </p>
+    </div>
+    """, padding=False)
+
+    # Hidden state to store task_url
+    task_url_state = gr.State("")
+
+    # Bind event
+    submit_btn.click(
+        generate_trump_voice_with_realtime_updates,
+        inputs=[text_input, word_num_slider],
+        outputs=[status_output, audio_output, task_info, action_links, task_url_state]
+    )
+
+if __name__ == "__main__":
     demo.launch() 

requirements.txt

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

util.py

@@ -1,116 +1,116 @@
-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]
-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):
-    import json
-    is_rewrite = False
-    url = f"{BackendUrl}/trump_process_ctx_api"
-    headers = {
-        "Content-Type": "application/json"
-    }
-    data = {
-        "template": VideoTemplate.UsFlag,
-        "text": text,
-        "word_num": word_num,
-        "is_rewrite": is_rewrite,
-        "watermark": True,
-        "type": task_type,
-        "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_v2(task_type, text, word_num, is_rewrite)
-    print(f"task_result: {task_result}")
+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]
+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):
+    import json
+    is_rewrite = False
+    url = f"{BackendUrl}/trump_process_ctx_api_v2"
+    headers = {
+        "Content-Type": "application/json"
+    }
+    print(url)
+    data = {
+        "video_template": VideoTemplate.UsFlag,
+        "speaker_template": "trump",
+        "text": text,
+        "word_num": word_num,
+        "is_rewrite": False,
+        "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_v2(task_type, text, word_num, is_rewrite)
+    print(f"task_result: {task_result}")