Update app.py

app.py

@@ -1,9 +1,10 @@
 import os 
 import gradio as gr
 import openai
+import requests
 
 ##############################################################################
-# 1. GPT 或 DeepSeek 调用示例函数
+# 1. GPT 调用示例函数
 ##############################################################################
 def generate_natural_language_description_gpt(tags, api_key, base_url=None, model="gpt-4"):
     """
@@ -12,7 +13,6 @@ def generate_natural_language_description_gpt(tags, api_key, base_url=None, mode
     if not api_key:
         return "Error: GPT API Key not provided."
 
-    # 设置 API
     openai.api_key = api_key
     if base_url:
         openai.api_base = base_url
@@ -45,35 +45,48 @@ def generate_natural_language_description_gpt(tags, api_key, base_url=None, mode
     except Exception as e:
         return f"GPT generation failed. Error: {e}"
 
-
+##############################################################################
+# 2. DeepSeek 调用示例函数 (真实请求)
+##############################################################################
 def generate_natural_language_description_deepseek(tags, api_key, base_url=None):
     """
-    使用 DeepSeek API 生成自然语言描述的示例函数。
-    这里留作你在实际部署时，替换为真正的 DeepSeek API 调用逻辑。
+    使用 DeepSeek API 生成自然语言描述的真实示例函数。
+    假设 DeepSeek 文档里的生成接口是:
+      POST https://api.deepseek.com/v1/generate
+      Headers: {"Authorization": "Bearer <api_key>"}
+      Body(JSON): {"tags": {...}}
+    返回: 
+      {"success": true, "data": {"description": "..."}}
     """
     if not api_key:
         return "Error: DeepSeek API Key not provided."
     
-    # 伪代码示例（请用真实 DeepSeek 调用替换此处）
-    # ----------------------------------
-    # import requests
-    # response = requests.post(
-    #     url=base_url or "https://api.deepseek.com/xxx", 
-    #     headers={"Authorization": f"Bearer {api_key}"},
-    #     json={"tags": tags}
-    # )
-    # if response.status_code == 200:
-    #     return response.json().get("description", "No description returned.")
-    # else:
-    #     return f"DeepSeek generation failed: {response.text}"
-    # ----------------------------------
-    
-    # 目前只是一个占位返回：
-    return "[DeepSeek: 此处应返回真实API结果]"
+    # 如果有自定义 base_url 就用，没有就用假设的默认
+    url = base_url or "https://api.deepseek.com/v1/generate"
 
+    try:
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {api_key}"
+        }
+        payload = {
+            "tags": tags  # 具体字段要根据实际 DeepSeek 文档
+        }
+        resp = requests.post(url, json=payload, headers=headers, timeout=30)
+        if resp.status_code == 200:
+            j = resp.json()
+            if j.get("success"):
+                # 假设描述存放在 data.description
+                return j["data"].get("description", "No 'description' found in data.")
+            else:
+                return f"DeepSeek generation failed, success=false. {j}"
+        else:
+            return f"DeepSeek generation failed with status {resp.status_code}. {resp.text}"
+    except Exception as e:
+        return f"DeepSeek generation request error: {e}"
 
 ##############################################################################
-# 2. 翻译示例函数（使用 GPT 或 DeepSeek）
+# 3. GPT 翻译函数
 ##############################################################################
 def translate_text_with_gpt(text, target_language, api_key, base_url=None, model="gpt-4"):
     """
@@ -99,37 +112,52 @@ def translate_text_with_gpt(text, target_language, api_key, base_url=None, model
     except Exception as e:
         return f"GPT translation failed. Error: {e}"
 
-
+##############################################################################
+# 4. DeepSeek 翻译函数 (真实请求)
+##############################################################################
 def translate_text_with_deepseek(text, target_language, api_key, base_url=None):
     """
-    使用 DeepSeek 来进行翻译的简单示例（仍需你在此处写入实际逻辑）。
+    使用 DeepSeek 来进行翻译的真实示例函数。
+    假设 DeepSeek 文档里的翻译接口是:
+      POST https://api.deepseek.com/v1/translate
+      Headers: {"Authorization": "Bearer <api_key>"}
+      Body(JSON): {"text": "...", "target_language": "..."}
+    返回: 
+      {"success": true, "data": {"translated_text": "..."}}
     """
     if not api_key:
         return "Error: DeepSeek Translation Key not provided."
     
-    # 伪代码示例（请用真实 DeepSeek 翻译调用替换）
-    # ----------------------------------
-    # import requests
-    # response = requests.post(
-    #     url=base_url or "https://api.deepseek.com/translate", 
-    #     headers={"Authorization": f"Bearer {api_key}"},
-    #     json={"text": text, "target_language": target_language}
-    # )
-    # if response.status_code == 200:
-    #     return response.json().get("translated_text", "No translation returned.")
-    # else:
-    #     return f"DeepSeek translation failed: {response.text}"
-    # ----------------------------------
-    
-    return "[DeepSeek: 此处应返回翻译API的真实结果]"
+    url = base_url or "https://api.deepseek.com/v1/translate"
 
+    try:
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {api_key}"
+        }
+        payload = {
+            "text": text,
+            "target_language": target_language
+        }
+        resp = requests.post(url, json=payload, headers=headers, timeout=30)
+        if resp.status_code == 200:
+            j = resp.json()
+            if j.get("success"):
+                # 假设翻译结果存放在 data.translated_text
+                return j["data"].get("translated_text", "No 'translated_text' found in data.")
+            else:
+                return f"DeepSeek translation failed, success=false. {j}"
+        else:
+            return f"DeepSeek translation failed with status {resp.status_code}. {resp.text}"
+    except Exception as e:
+        return f"DeepSeek translation request error: {e}"
 
 ##############################################################################
-# 3. 根据用户选择进行提示词转换并调用 GPT/DeepSeek 生成描述
+# 5. 根据用户选择进行提示词转换并调用 GPT/DeepSeek 生成描述
 ##############################################################################
 def transform_prompt(prompt, gender_option, furry_species, api_mode, api_key):
     """
-    性别/物种转换的简单示例逻辑，然后调用相应 API。
+    性别/物种转换的简单示例逻辑，然后调用相应 API 进行生成。
     """
     tags = {}
     
@@ -149,7 +177,6 @@ def transform_prompt(prompt, gender_option, furry_species, api_mode, api_key):
     # 原始提示词
     tags["base_prompt"] = prompt
 
-    # 根据选择的 API 调用对应的函数
     if api_mode == "GPT":
         scene_description = generate_natural_language_description_gpt(tags, api_key)
     else:  # DeepSeek
@@ -157,9 +184,8 @@ def transform_prompt(prompt, gender_option, furry_species, api_mode, api_key):
     
     return scene_description
 
-
 ##############################################################################
-# 4. 调用翻译函数
+# 6. 调用翻译函数
 ##############################################################################
 def do_translation(scene_desc, translate_language, api_mode, api_key):
     """
@@ -173,9 +199,8 @@ def do_translation(scene_desc, translate_language, api_mode, api_key):
     else:
         return translate_text_with_deepseek(scene_desc, translate_language, api_key)
 
-
 ##############################################################################
-# 5. 搭建 Gradio 界面
+# 7. 搭建 Gradio 界面
 ##############################################################################
 def build_interface():
     with gr.Blocks() as demo:
@@ -216,7 +241,7 @@ def build_interface():
                     label="Furry 物种 (Furry Species)",
                     choices=["Wolf", "Fox", "Tiger", "Lion"],  
                     value=None,
-                    visible=False  # 初始不可见
+                    visible=False
                 )
 
                 # 当性别选项切换时，如果选择 Furry，就显示物种下拉，否则隐藏
@@ -246,11 +271,9 @@ def build_interface():
                     lines=7
                 )
 
-        # 翻译区域
         with gr.Row():
             translate_language = gr.Dropdown(
                 label="翻译语言 (Translation Language)",
-                # 可自行添加更多语言选项
                 choices=["English", "Chinese", "Japanese", "French", "German", "Dutch", "Arabic", "Russian", "Persian", "Italian"],
                 value="English",
             )
@@ -263,23 +286,19 @@ def build_interface():
         # 事件绑定
         ######################################################################
         
-        # 生成函数：一次性返回「转换结果」和「翻译结果」
         def on_generate(prompt, gender, furry, mode, key, lang):
-            # 1) 先做性别/物种转换，拿到“转换后”提示词
+            # 1) 调用 transform_prompt 拿到转换后的描述
             trans_desc = transform_prompt(prompt, gender, furry, mode, key)
-            # 2) 立刻翻译
+            # 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, translate_language],
             outputs=[generated_output, translated_text],
         )
 
-        # 点击���钮也触发同样的逻辑
         generate_button = gr.Button("生成 / Generate")
         generate_button.click(
             fn=on_generate,
@@ -287,7 +306,6 @@ def build_interface():
             outputs=[generated_output, translated_text],
         )
 
-        # 当用户切换翻译语言时，如果已经有转换后的内容，则再次翻译
         def on_translate(scene_desc, lang, mode, key):
             return do_translation(scene_desc, lang, mode, key)
 
@@ -299,8 +317,6 @@ def build_interface():
 
     return demo
 
-
-# 在 Spaces 启动
 if __name__ == "__main__":
     demo = build_interface()
-    demo.launch()
+    demo.launch()