Update app.py

app.py

@@ -4,22 +4,28 @@ import gradio as gr
 from openai import OpenAI
 
 ##############################################################################
-# 1. 读取外部文件: furry_species.json & gender_rules.json
+# 1. 读取外部文件
 ##############################################################################
 try:
-    with open("furry_species.json", "r", encoding="utf-8") as f:
-        FURRY_DATA = json.load(f)
+    with open("furry_species.json", "r", encoding="utf-8") as ff:
+        FURRY_DATA = json.load(ff)
 except:
     FURRY_DATA = {}
 
 try:
-    with open("gender_rules.json", "r", encoding="utf-8") as f:
-        GENDER_RULES = json.load(f)
+    with open("gender_rules.json", "r", encoding="utf-8") as gf:
+        GENDER_RULES = json.load(gf)
 except:
     GENDER_RULES = {}
 
+try:
+    with open("transform_rules.json", "r", encoding="utf-8") as tf:
+        TRANSFORM_DICT = json.load(tf)
+except:
+    TRANSFORM_DICT = {}
+
 ##############################################################################
-# 2. 多级菜单构造函数
+# 2. 多级菜单函数
 ##############################################################################
 def get_top_categories(furry_data):
     return sorted(list(furry_data.keys()))
@@ -35,50 +41,58 @@ def get_species_list(furry_data, top_category, sub_category):
     return []
 
 ##############################################################################
-# 3. 核心调用：GPT 或 DeepSeek
+# 3. 合并规则文本
 ##############################################################################
-def generate_transformed_output(
-    prompt,        # 原始 Prompt，如 "1girl, butterfly, solo..."
-    gender_option, # 转换目标
-    top_cat, sub_cat, species_item,
-    api_mode, api_key
-):
+def merge_transform_rules_into_prompt(rules_json):
+    """
+    将 transform_rules.json 中的相关字段转为统一文本，便于放到 system_prompt。
+    你也可以分段加入。
     """
-    根据指定的性别/物种等规则，让 GPT/DeepSeek 输出仅两段内容：
-     (转化后tags)
+    if not rules_json:
+        return "(No transform rules loaded)"
+    
+    # 1) 读取 gender_transform
+    gt = rules_json.get("gender_transform", {})
+    male_tag_rules = gt.get("male_tag_rules", {})
+    replacements = gt.get("tag_replacements", {})
+    additional = gt.get("additional_reminders", {})
+    
+    # 2) shared_preferences
+    sp = rules_json.get("shared_preferences", {})
+    
+    # 3) table_details
+    td = rules_json.get("table_details", {})
+
+    # 这只是一个简单示例，将 key-value 拼起来，真实项目可更精细
+    text_parts = []
+
+    text_parts.append("==== GENDER TRANSFORM RULES ====")
+    text_parts.append(str(gt))  # 直接转为字符串或更有条理地拼写
+    
+    text_parts.append("==== SHARED PREFERENCES ====")
+    text_parts.append(str(sp))
+
+    text_parts.append("==== TABLE DETAILS (PRO ACTIONS) ====")
+    text_parts.append(str(td))
+
+    return "\n".join(text_parts)
+
+RULES_TEXT_FULL = merge_transform_rules_into_prompt(TRANSFORM_DICT)
 
-     转化后描述 (3~6句)
-    不展示原始 prompt/base_prompt/gender: ... 等信息。
+##############################################################################
+# 4. 核心 GPT/DeepSeek 调用
+##############################################################################
+def generate_transformed_output(prompt, gender_option, top_cat, sub_cat, species_item, api_mode, api_key):
+    """
+    读取 transform_rules.json / GENDER_RULES / FurryData:
+    只输出两段：(tags)\n\n(description)
     """
     if not api_key:
         return "Error: No API Key provided."
 
-    # 根据 gender_option 选对应的 rule
-    if gender_option == "Trans_to_Male":
-        rule_text = GENDER_RULES.get("male", "")
-    elif gender_option == "Trans_to_Female":
-        rule_text = GENDER_RULES.get("female", "")
-    elif gender_option == "Trans_to_Mannequin":
-        rule_text = GENDER_RULES.get("genderless", "")
-    elif gender_option == "Trans_to_Intersex":
-        rule_text = GENDER_RULES.get("intersex", "")
-    else:
-        # Furry
-        # 你可以综合 male/female/intersex/genderless，也可以有专门 furry 的说明
-        rule_text = (
-            GENDER_RULES.get("male", "") + "\n\n"
-            + GENDER_RULES.get("female", "") + "\n\n"
-            + GENDER_RULES.get("intersex", "") + "\n\n"
-            + GENDER_RULES.get("genderless", "")
-        )
-        # 如果想在规则中附加选定的物种信息：
-        if top_cat and sub_cat and species_item:
-            rule_text += f"\nFurry species: {top_cat} > {sub_cat} > {species_item}\n"
-
-    # 选定 GPT or DeepSeek
     if api_mode == "GPT":
         base_url = None
-        model_name = "gpt-3.5-turbo"  # 可改成 "gpt-4"
+        model_name = "gpt-3.5-turbo"
     else:
         base_url = "https://api.deepseek.com"
         model_name = "deepseek-chat"
@@ -87,26 +101,46 @@ def generate_transformed_output(
     if base_url:
         client.base_url = base_url
 
-    # ��造 System Prompt：要求只输出两段；第一行(转化后tags)，空行，随后3~6句描述
-    # 让它把 prompt 中的 tags 进行「合并、替换、去重、增加」等处理
+    # 如果选 Furry:
+    if gender_option == "Trans_to_Furry":
+        # 在 system prompt 内也可写点提示：
+        furry_path = f"{top_cat} > {sub_cat} > {species_item}" if (top_cat and sub_cat and species_item) else "unknown"
+        extra_line = f"\nFurry chosen: {furry_path}\n"
+    else:
+        extra_line = ""
+
+    # 这里把 "RULES_TEXT_FULL" (来自 transform_rules.json) + GENDER_RULES + 
+    # "extra_line" 一并放到 system_prompt
+    gender_specific_rule = ""
+    if gender_option == "Trans_to_Male":
+        gender_specific_rule = GENDER_RULES.get("male", "")
+    elif gender_option == "Trans_to_Female":
+        gender_specific_rule = GENDER_RULES.get("female", "")
+    elif gender_option == "Trans_to_Mannequin":
+        gender_specific_rule = GENDER_RULES.get("genderless", "")
+    elif gender_option == "Trans_to_Intersex":
+        gender_specific_rule = GENDER_RULES.get("intersex", "")
+
     system_prompt = f"""
-You are a creative assistant that modifies the user's base prompt tags 
-to reflect the correct gender/furry transformations, following these rules:
-
-{rule_text}
-
-Steps:
-1) The original prompt tags are: {prompt}
-2) Convert them into NEW combined tags that reflect the requested transformation. 
-   (Remove or replace conflicting tags, unify synonyms, add any essential tags 
-    for {gender_option} or for the selected furry species.)
-3) Output EXACTLY two parts:
-   - First line: the final, consolidated tags in parentheses (e.g. (male, solo, ...)).
-   - Then a blank line.
-   - Then a short imaginative scene description (3 to 6 sentences). 
-4) Do NOT include 'gender:' or 'base_prompt:' or any headings or extra lines. 
-5) Output everything in English.
-6) Do not reference these steps in the final answer.
+You are a creative assistant that transforms the user's base prompt 
+to reflect correct gender/furry transformations. Follow these references:
+
+1) Detailed Transform Rules (transform_rules.json):
+{RULES_TEXT_FULL}
+
+2) Additional short gender rules (gender_rules.json):
+{gender_specific_rule}
+
+{extra_line}
+Instructions:
+- Original prompt tags: {prompt}
+- Convert them into NEW combined tags, removing or replacing conflicting ones.
+- Only output two parts:
+  1) One line of final tags in parentheses, e.g. (male, short hair, dynamic pose, ...)
+  2) A blank line.
+  3) Then 3~6 sentences of imaginative scene description in English.
+- No extra lines, no headings, no 'gender:' or 'base_prompt:'.
+- End of instructions.
 """.strip()
 
     try:
@@ -114,20 +148,14 @@ Steps:
             model=model_name,
             messages=[
                 {"role": "system", "content": system_prompt},
-                {"role": "user", "content": "Generate the final tags and description now."},
+                {"role": "user", "content": "Generate final tags and description now."}
             ],
         )
-        # 结果中仅包含 (tags)\n\n(description)
-        result = resp.choices[0].message.content.strip()
-        return result
-
+        return resp.choices[0].message.content.strip()
     except Exception as e:
         return f"{api_mode} generation failed. Error: {e}"
 
 def translate_text(content, lang, api_mode, api_key):
-    """
-    对上一步的 (tags)\n\n(description) 做翻译，保持格式
-    """
     if not api_key:
         return "Error: No API Key provided."
     if not content.strip():
@@ -146,8 +174,8 @@ def translate_text(content, lang, api_mode, api_key):
 
     translate_system_prompt = f"""
 You are a translator. Translate the following text to {lang}, 
-preserving the parentheses line and blank lines if present. 
-Do not add extra headings.
+keeping parentheses line and blank line if present. 
+No extra headings.
 """.strip()
 
     try:
@@ -155,7 +183,7 @@ Do not add extra headings.
             model=model_name,
             messages=[
                 {"role": "system", "content": translate_system_prompt},
-                {"role": "user", "content": content},
+                {"role": "user", "content": content}
             ],
         )
         return resp.choices[0].message.content.strip()
@@ -163,30 +191,27 @@ Do not add extra headings.
         return f"{api_mode} translation failed. Error: {e}"
 
 ##############################################################################
-# 4. Gradio 前端
+# 5. Gradio 界面
 ##############################################################################
 def build_interface():
     with gr.Blocks() as demo:
-        gr.Markdown("## Prompt Transformer Tool- 提示词物种性别转化器(GPT/DeepSeek)")
+        gr.Markdown("## Prompt Transformer - (male/female/furry) & Scene Design with transform_rules.json")
 
         with gr.Row():
             with gr.Column():
-                # 选择 GPT/DeepSeek
                 api_mode = gr.Radio(
-                    label="Choose API (GPT or DeepSeek)",
+                    label="Select API (GPT/DeepSeek)",
                     choices=["GPT", "DeepSeek"],
                     value="GPT"
                 )
-                # 输入 API Key
                 api_key = gr.Textbox(
-                    label="API Key-API密匙",
+                    label="API Key",
                     type="password",
-                    placeholder="Enter your GPT or DeepSeek Key here"
+                    placeholder="Input your GPT or DeepSeek Key"
                 )
 
-                # 性别/Furry
                 gender_option = gr.Radio(
-                    label="Gender / Furry Conversion 选择物种性别",
+                    label="Gender / Furry Option",
                     choices=[
                         "Trans_to_Male",
                         "Trans_to_Female",
@@ -197,7 +222,6 @@ def build_interface():
                     value="Trans_to_Male"
                 )
 
-                # 若选 Furry -> 多级菜单
                 top_cat_dd = gr.Dropdown(
                     label="Furry: Top Category",
                     choices=get_top_categories(FURRY_DATA),
@@ -217,9 +241,8 @@ def build_interface():
                     visible=False
                 )
 
-                # 当性别选项切到 Furry 时，显示下拉，否则隐藏
-                def show_furry_options(chosen):
-                    if chosen == "Trans_to_Furry":
+                def show_furry_options(opt):
+                    if opt == "Trans_to_Furry":
                         return (gr.update(visible=True),
                                 gr.update(visible=True),
                                 gr.update(visible=True))
@@ -227,83 +250,77 @@ def build_interface():
                         return (gr.update(visible=False),
                                 gr.update(visible=False),
                                 gr.update(visible=False))
+
                 gender_option.change(
                     fn=show_furry_options,
                     inputs=[gender_option],
                     outputs=[top_cat_dd, sub_cat_dd, species_dd]
                 )
 
-                # 主分类 -> 子分类
                 def on_top_cat_select(selected):
                     subs = get_sub_categories(FURRY_DATA, selected)
                     return gr.update(choices=subs, value=None)
+
                 top_cat_dd.change(
                     fn=on_top_cat_select,
                     inputs=[top_cat_dd],
                     outputs=[sub_cat_dd]
                 )
 
-                # 子分类 -> 物种
                 def on_sub_cat_select(top_c, sub_c):
                     sp = get_species_list(FURRY_DATA, top_c, sub_c)
                     return gr.update(choices=sp, value=None)
+
                 sub_cat_dd.change(
                     fn=on_sub_cat_select,
                     inputs=[top_cat_dd, sub_cat_dd],
                     outputs=[species_dd]
                 )
 
-            # 输入 prompt & 输出
             with gr.Column():
                 user_prompt = gr.Textbox(
-                    label="原始 Prompt (Base Tags)",
-                    lines=5,
-                    placeholder="Your Prompts"
+                    label="Original Prompt (e.g. 1girl, butterfly, solo, ...)",
+                    lines=5
                 )
                 final_output = gr.Textbox(
-                    label="(转化后Tags)\n\n(转化后描述)",
+                    label="Transformed Output (tags + description)",
                     lines=10
                 )
 
-        # 翻译
         with gr.Row():
             translate_lang = gr.Dropdown(
-                label="翻译语言 Translater",
+                label="Translate to Language",
                 choices=["English", "Chinese", "Japanese", "French", "German", "Spanish"],
                 value="English"
             )
-            translated_result = gr.Textbox(
-                label="翻译后的结果 Trans-Language Result",
+            translated_text = gr.Textbox(
+                label="Translated Result",
                 lines=10
             )
 
         ######################################################################
-        # 事件
+        # 生成
         ######################################################################
-        def on_generate(prompt, gender, tc, sc, sp, mode, key, lang):
-            # 1) 生成新的 (tags) + 描述
-            merged = generate_transformed_output(prompt, gender, tc, sc, sp, mode, key)
-            # 2) 翻译
+        def on_generate(prompt, gender, tc, sc, spc, mode, key, lang):
+            merged = generate_transformed_output(prompt, gender, tc, sc, spc, mode, key)
             trans = translate_text(merged, lang, mode, key)
             return merged, trans
 
-        # 回车提交
         user_prompt.submit(
             fn=on_generate,
             inputs=[user_prompt, gender_option, top_cat_dd, sub_cat_dd, species_dd, api_mode, api_key, translate_lang],
-            outputs=[final_output, translated_result]
+            outputs=[final_output, translated_text]
         )
-        # 点击按钮
-        btn = gr.Button("生成 / Generate")
-        btn.click(
+
+        gen_btn = gr.Button("Generate")
+        gen_btn.click(
             fn=on_generate,
             inputs=[user_prompt, gender_option, top_cat_dd, sub_cat_dd, species_dd, api_mode, api_key, translate_lang],
-            outputs=[final_output, translated_result]
+            outputs=[final_output, translated_text]
         )
 
     return demo
 
-
 if __name__ == "__main__":
     demo = build_interface()
     demo.launch()