Update app.py

app.py

@@ -92,10 +92,62 @@ def forced_replace(prompt, direction):
         prompt = re.sub(pattern, new, prompt)
 
     # 针对复杂句子的补充替换
-    if direction == "female_to_male":
+    if direction == "Trans_to_male":
+        # 女性到男性
         prompt = re.sub(r"\bshe\b", "he", prompt, flags=re.IGNORECASE)
         prompt = re.sub(r"\bher\b", "his", prompt, flags=re.IGNORECASE)
         prompt = re.sub(r"\bherself\b", "himself", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bwomen\b", "men", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\blady\b", "gentleman", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bqueen\b", "king", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bfemale\b", "male", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bgirl\b", "boy", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bprincess\b", "prince", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bactress\b", "actor", prompt, flags=re.IGNORECASE)
+    
+    elif direction == "Trans_to_female":
+        # 男性到女性
+        prompt = re.sub(r"\bhe\b", "she", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bhis\b", "her", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bhimself\b", "herself", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bmen\b", "women", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bgentleman\b", "lady", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bking\b", "queen", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bmale\b", "female", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bboy\b", "girl", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bprince\b", "princess", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bactor\b", "actress", prompt, flags=re.IGNORECASE)
+    
+    elif direction == "Trans_to_mannequin":
+        # 转换为无性别或人偶化
+        prompt = re.sub(r"\bshe\b|\bhe\b", "it", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bher\b|\bhis\b", "its", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bherself\b|\bhimself\b", "itself", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bwoman\b|\bman\b", "mannequin-like figure", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bgirl\b|\bboy\b", "character", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bqueen\b|\bking\b", "figurehead", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bfemale\b|\bmale\b", "genderless", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bprincess\b|\bprince\b", "androgynous heir", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bactress\b|\bactor\b", "performer", prompt, flags=re.IGNORECASE)
+    
+    elif direction == "Trans_to_intersex":
+        # 转换为双性化特征
+        prompt = re.sub(r"\bshe\b", "they", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bhe\b", "they", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bher\b", "their", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bhis\b", "their", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bherself\b", "themself", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bhimself\b", "themself", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bwoman\b", "androgynous individual", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bman\b", "androgynous individual", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bgirl\b", "intersex character", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bboy\b", "intersex character", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bfemale\b", "intersex", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bmale\b", "intersex", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bqueen\b", "intersex ruler", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bking\b", "intersex ruler", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bprincess\b", "intersex heir", prompt, flags=re.IGNORECASE)
+        prompt = re.sub(r"\bprince\b", "intersex heir", prompt, flags=re.IGNORECASE)
 
     return prompt
 
@@ -119,30 +171,30 @@ def generate_transformed_output(prompt, gender_option, top_cat, sub_cat, species
 
     if gender_option == "Trans_to_Furry":
         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"
+        extra_line = f"\nFurry chosen: {furry_path} anthro furry\n"
     else:
         extra_line = ""
 
     gender_specific_rule = ""
     if gender_option == "Trans_to_Male":
         gender_specific_rule = GENDER_RULES.get("male", "")
+        extra_line = f"\n1boy\n"
     elif gender_option == "Trans_to_Female":
         gender_specific_rule = GENDER_RULES.get("female", "")
+        extra_line = f"\n1girl\n"
     elif gender_option == "Trans_to_Mannequin":
         gender_specific_rule = GENDER_RULES.get("genderless", "")
+        extra_line = f"\nmannequin or statue or doll or robot or android\n"
     elif gender_option == "Trans_to_Intersex":
         gender_specific_rule = GENDER_RULES.get("intersex", "")
+        extra_line = f"\nhanve both penis_testicles and pussy\n"
 
     system_prompt = f"""
 You are a creative assistant that transforms the user's base prompt 
-生成将提示词中的主体转化为{gender_specific_rule}{extra_line}的提示词，注意性别，其服装和装饰细节，皮肤和第二性征的转变，并且 Follow references below:
-1) Detailed Transform Rules (transform_rules.json):
+生成将提示词中的主体转化为{extra_line}{gender_specific_rule}的提示词，注意性别，其服装和装饰细节，皮肤和第二性征的转变，并且 Follow references below:
+Detailed Transform Rules (transform_rules.json):
 {RULES_TEXT_FULL}
-2) Additional short gender rules (gender_rules.json):
-{gender_specific_rule}
-{extra_line}
 Instructions:
-- Remove any female-specific terms if the target is male.
 - Ensure all gender-specific terms are replaced correctly based on the rules.
 - Original prompt tags: {prompt}
 - Convert them into NEW combined tags, removing or replacing conflicting ones.
@@ -188,7 +240,7 @@ def translate_text(content, lang, api_mode, api_key):
 
     translate_system_prompt = f"""
 You are a translator. Translate the following text to {lang}, 
-keeping parentheses line and blank line if present. 
+keeping parentheses line, tags and blank line if present. 
 No extra headings.
 """.strip()