init

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import time
 import uuid
+import re
 from util import (
     create_task_v3, 
     get_task_result,
@@ -9,10 +10,95 @@ from util import (
 
 IP_Dict = {}
 
-def generate_trump_voice_with_realtime_updates(text, word_num, request: gr.Request):
+# 支持的语言列表
+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_trump_voice_with_realtime_updates(text, language_display, request: gr.Request):
     """
     Trump 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:
@@ -53,9 +139,10 @@ def generate_trump_voice_with_realtime_updates(text, word_num, request: gr.Reque
     
     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)
+        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:
@@ -183,9 +270,15 @@ with gr.Blocks(title="Donald Trump AI Voice", theme=gr.themes.Soft()) as demo:
             )
         
         with gr.Column(scale=1):
-            word_num_slider = gr.Slider(
-                20, 60, value=60, step=1,
-                label="⏱️ Duration Limit"
+            # 创建语言选择器的选项
+            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(
@@ -229,7 +322,7 @@ with gr.Blocks(title="Donald Trump AI Voice", theme=gr.themes.Soft()) as demo:
                 <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;">🌺 Garden Interview</h4>
                     <video id="video-garden" style="width: 100%; height: 180px; border-radius: 10px; object-fit: cover;" controls preload="metadata">
-                        <source src="https://www.trumpaivoice.net/SelfitCamera/2025-09-09/62_1757427854_wDCsM8tb/62_1757427854_wDCsM8tb-r.mp4" type="video/mp4">
+                        <source src="https://www.trumpaivoice.net/SelfitAssert/Heygem/Trump/trump_showcases/musk0710.mp4" type="video/mp4">
                         Your browser does not support the video tag.
                     </video>
                 </div>
@@ -237,7 +330,7 @@ with gr.Blocks(title="Donald Trump AI Voice", theme=gr.themes.Soft()) as demo:
                 <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;">🏛️ Presidential Office</h4>
                     <video id="video-office" style="width: 100%; height: 180px; border-radius: 10px; object-fit: cover;" controls preload="metadata">
-                        <source src="https://www.trumpaivoice.net/SelfitCamera/2025-09-09/64_1757428412_qzDSespW/merged_mv.mp4" type="video/mp4">
+                        <source src="https://www.trumpaivoice.net/SelfitAssert/Heygem/Trump/trump_showcases/62_1757764805_9klRXL24-r.mp4" type="video/mp4">
                         Your browser does not support the video tag.
                     </video>
                 </div>
@@ -245,7 +338,7 @@ with gr.Blocks(title="Donald Trump AI Voice", theme=gr.themes.Soft()) as demo:
                 <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;">🇺🇸 Flag Background</h4>
                     <video id="video-flag" style="width: 100%; height: 180px; border-radius: 10px; object-fit: cover;" controls preload="metadata">
-                        <source src="https://www.trumpaivoice.net/SelfitCamera/2025-09-09/62_1757428209_ly0hK4XR/62_1757428209_ly0hK4XR-r.mp4" type="video/mp4">
+                        <source src="https://www.trumpaivoice.net/SelfitAssert/Heygem/Trump/trump_showcases/puting0710.mp4" type="video/mp4">
                         Your browser does not support the video tag.
                     </video>
                 </div>
@@ -449,10 +542,17 @@ with gr.Blocks(title="Donald Trump AI Voice", theme=gr.themes.Soft()) as demo:
     # 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_trump_voice_with_realtime_updates,
-        inputs=[text_input, word_num_slider],
+        inputs=[text_input, language_dropdown],
         outputs=[status_output, audio_output, action_links, task_url_state]
     )
 

util.py

@@ -48,10 +48,10 @@ class VideoTemplate:
     UsFlag = "us-flag"
     WhiteHouse = "white-house"
 
-def create_task_v3(task_type, text, word_num, is_rewrite):
+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_v2"
+    url = f"{BackendUrl}/trump_process_ctx_api_v3"
     headers = {
         "Content-Type": "application/json"
     }
@@ -62,6 +62,7 @@ def create_task_v3(task_type, text, word_num, is_rewrite):
         "text": text,
         "word_num": word_num,
         "is_rewrite": False,
+        "language": language,
         "watermark": True,
         "type": "voice",
         "cost_credits": 2,
@@ -114,5 +115,5 @@ if __name__ == "__main__":
     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)
+    task_result = create_task_v3(task_type, text, word_num, is_rewrite)
     print(f"task_result: {task_result}")