Update app.py

app.py

@@ -1,4 +1,4 @@
-import os
+import os 
 import gradio as gr
 import openai
 
@@ -54,18 +54,14 @@ def generate_natural_language_description_deepseek(tags, api_key, base_url=None)
     if not api_key:
         return "Error: DeepSeek API Key not provided."
     
-    # 伪代码示例
-    # ----------------------------------
+    # 伪代码示例（需根据你的 DeepSeek API 文档做实际实现）
     # import requests
-    # ...
     # response = requests.post(
     #     url=base_url or "https://api.deepseek.com/xxx", 
     #     headers={"Authorization": f"Bearer {api_key}"},
     #     json={"tags": tags}
     # )
     # return response.json()["description"]
-    # ----------------------------------
-    # 这里为了演示，就直接返回简单的字符串
     return "DeepSeek 生成的描述（此处为示例伪代码）"
 
 
@@ -84,7 +80,6 @@ def translate_text_with_gpt(text, target_language, api_key, base_url=None, model
         openai.api_base = base_url
 
     try:
-        # 通过系统提示，让 GPT 做翻译
         system_prompt = f"You are a professional translator. Translate the following text to {target_language}:"
         response = openai.ChatCompletion.create(
             model=model,
@@ -104,7 +99,7 @@ def translate_text_with_deepseek(text, target_language, api_key, base_url=None):
     """
     if not api_key:
         return "Error: DeepSeek Translation Key not provided."
-    # 类似上面的伪代码方式。
+    # 同样需要根据 DeepSeek 的文档来实现
     return f"DeepSeek翻译后的文本（示例）。目标语言：{target_language}"
 
 
@@ -114,15 +109,18 @@ def translate_text_with_deepseek(text, target_language, api_key, base_url=None):
 def transform_prompt(prompt, gender_option, furry_species, api_mode, api_key):
     """
     性别/物种转换的简单示例逻辑，然后调用相应 API。
+    你可在此处结合“关于Male/Female/ambiguous/intersex的details”添加更复杂的处理。
     """
     tags = {}
     
     # 根据选择设置性别或物种标签
     if gender_option == "Trans_to_Male":
+        # 这里可以参考你的细节 rules 做更加复杂的转换
         tags["gender"] = "male"
     elif gender_option == "Trans_to_Female":
         tags["gender"] = "female"
     elif gender_option == "Trans_to_Mannequin":
+        # 无明显二性征
         tags["gender"] = "genderless"
     elif gender_option == "Trans_to_Intersex":
         tags["gender"] = "intersex"
@@ -150,7 +148,7 @@ def do_translation(scene_desc, translate_language, api_mode, api_key):
     根据选择的 API（GPT/DeepSeek）进行翻译。
     """
     if not scene_desc.strip():
-        return ""
+        return ""  # 无内容则不翻译
 
     if api_mode == "GPT":
         return translate_text_with_gpt(scene_desc, translate_language, api_key)
@@ -205,7 +203,7 @@ def build_interface():
 
                 # 当性别选项切换时，如果选择 Furry，就显示物种下拉，否则隐藏
                 def show_furry_species(gender):
-                    return gr.update(visible=(gender == "Furry"))
+                    return gr.update(visible=(gender == "Trans_to_Furry"))
 
                 gender_option.change(
                     show_furry_species, 
@@ -218,7 +216,10 @@ def build_interface():
                 user_prompt = gr.Textbox(
                     label="提示词 (Prompt)",
                     lines=5,
-                    placeholder="Please Enter your prompt words. 在此输入你的提示词，例如：一位穿着红色连衣裙的少女，坐在落日余晖下的草地上..."
+                    placeholder=(
+                        "Please Enter your prompt words. \n"
+                        "在此输入你的提示词，例如：一位穿着红色连衣裙的少女，坐在落日余晖下的草地上..."
+                    )
                 )
 
                 # 输出场景描述
@@ -231,6 +232,7 @@ def build_interface():
         with gr.Row():
             translate_language = gr.Dropdown(
                 label="翻译语言 (Translation Language)",
+                # 可自行添加更多语言选项
                 choices=["English", "Chinese", "Japanese", "French", "German", "Dutch", "Arabic", "Russian", "Persian", "Italian"],
                 value="English",
             )
@@ -242,28 +244,36 @@ def build_interface():
         ######################################################################
         # 事件绑定
         ######################################################################
-        def on_generate(prompt, gender, furry, mode, key):
-            return transform_prompt(prompt, gender, furry, mode, key)
-
-        # 用户在 prompt 输入后按回车或失去焦点时，触发生成场景描述
+        
+        # 新增：生成时，直接返回「转换结果」和「翻译结果」并一起更新
+        def on_generate(prompt, gender, furry, mode, key, lang):
+            # 1) 先做性别/物种转换，拿到“转换后”提示词
+            trans_desc = transform_prompt(prompt, gender, furry, mode, key)
+            # 2) 立刻翻译
+            trans_result = do_translation(trans_desc, lang, mode, key)
+            # 返回两项
+            return trans_desc, trans_result
+
+        # 当用户在 prompt 输入后按回车时，触发生成场景描述 + 翻译
         user_prompt.submit(
             fn=on_generate,
-            inputs=[user_prompt, gender_option, furry_species, api_mode, api_key],
-            outputs=[generated_output],
+            inputs=[user_prompt, gender_option, furry_species, api_mode, api_key, translate_language],
+            outputs=[generated_output, translated_text],
         )
 
-        # 也可以加个按钮
+        # 点击按钮也触发同样的逻辑
         generate_button = gr.Button("生成 / Generate")
         generate_button.click(
             fn=on_generate,
-            inputs=[user_prompt, gender_option, furry_species, api_mode, api_key],
-            outputs=[generated_output],
+            inputs=[user_prompt, gender_option, furry_species, api_mode, api_key, translate_language],
+            outputs=[generated_output, translated_text],
         )
 
-        # 当用户切换翻译语言时，自动翻译当前生成文本
+        # 当用户切换翻译语言时，如果已经有转换后的内容，则再翻译一次
         def on_translate(scene_desc, lang, mode, key):
             return do_translation(scene_desc, lang, mode, key)
 
+        # 注意：这里只更新“翻译结果”一项
         translate_language.change(
             fn=on_translate,
             inputs=[generated_output, translate_language, api_mode, api_key],
@@ -276,4 +286,4 @@ def build_interface():
 # 在 Spaces 启动
 if __name__ == "__main__":
     demo = build_interface()
-    demo.launch()
+    demo.launch()