Update app.py

app.py

@@ -129,13 +129,28 @@ def query(prompt, negative_prompt, steps=4, cfg_scale=0, seed=-1, width=1024, he
         print("") # Add newline
 
     # --- Proceed with Generation ---
-    final_prompt = f"{translated_prompt} | ultra detail, ultra elaboration, ultra quality, perfect, adult"
+    # Define a threshold for modifying prompts based on sexual content score
+    SEXUAL_SCORE_THRESHOLD = 0.6
+    prompt_suffix = " | ultra detail, ultra elaboration, ultra quality, perfect" # Base quality enhancers
+    negative_prompt_addition = "" # Start with no additions
+
+    # Check moderation scores if available and adjust prompts
+    if openai_client and 'mod_response' in locals(): # Check if moderation ran and response exists
+        result = mod_response.results[0]
+        if hasattr(result, 'category_scores') and result.category_scores.sexual > SEXUAL_SCORE_THRESHOLD:
+            print(f"INFO: High 'sexual' score detected ({result.category_scores.sexual:.4f}). Modifying prompts.")
+            prompt_suffix += "adult face, mature face, grown-up, of age, " 
+            negative_prompt_addition = "child, minor, underage, teenager, "
+
+    final_prompt = f"{prompt_suffix}{translated_prompt}"
+    final_negative_prompt = f"{negative_prompt_addition}{negative_prompt}".strip(', ') # Combine and clean up potential leading comma
 
     payload = {
         "inputs": final_prompt,
         "parameters": {
             "width": width, "height": height, "num_inference_steps": steps,
-            "negative_prompt": negative_prompt, "guidance_scale": cfg_scale,
+            "negative_prompt": final_negative_prompt, # Use the potentially modified negative prompt
+            "guidance_scale": cfg_scale,
             "seed": seed if seed != -1 else random.randint(1, 1000000000),
         }
     }