Update app.py

app.py

@@ -22,14 +22,21 @@ text_encoder_client = Client("Gemini899/mistral-text-encoder")
 
 # ============================================================================
 # HARDCODED PROMPTS - EXACT match from depth_logic.py (bypasses Mistral API)
+# Now with 4 variants: face/no-face for both relief and details
 # ============================================================================
 
 HARDCODED_PROMPTS = {
-    # From call_flux2_dev_relief_generation() in depth_logic.py
-    "relief": "Ignore all shadows, Clay bas-relief sculpture. PRESERVE exact facial features and proportions. Uniform matte gray material, NO black areas, NO dark shadows, NO outlines. Soft smooth depth only. Light gray to white tones. Like carved marble or clay relief. Keep nostrils and area under nose BRIGHT and light-toned, NOT dark pits.",
+    # From call_flux2_dev_relief_generation() in depth_logic.py - WITH FACE
+    "relief_face": "Ignore all shadows, Clay bas-relief sculpture. PRESERVE exact facial features and proportions. Uniform matte gray material, NO black areas, NO dark shadows, NO outlines. Soft smooth depth only. Light gray to white tones. Like carved marble or clay relief. Nose area details soft and delicate, no high contrast.",
     
-    # From call_flux2_dev_detail_generation() in depth_logic.py
-    "details": "Preserve exact pixel alignment and IDENTICAL tonality as original image. Enhance this depth map with polished smooth skin texture (NOT pixelated), subtle fabric weave, fine hair strands, stone grain. Skin must appear smooth and refined and polished. Keep EXACT same outline, silhouette, and tonal range - output tonality must match input exactly. NO shadows, NO reflections, NO new light sources, NO dark areas, NO dark halos. Output overlays perfectly on original as bump map detail layer, Keep nostrils and area under nose BRIGHT and light-toned, NOT dark pits"
+    # From call_flux2_dev_relief_generation() in depth_logic.py - NO FACE
+    "relief": "Ignore all shadows, Clay bas-relief sculpture. PRESERVE exact proportions and features. Uniform matte gray material, NO black areas, NO dark shadows, NO outlines. Soft smooth depth only. Light gray to white tones. Like carved marble or clay relief.",
+    
+    # From call_flux2_dev_detail_generation() in depth_logic.py - WITH FACE
+    "details_face": "Preserve exact pixel alignment. Enhance this depth map with polished smooth skin texture (NOT pixelated), subtle fabric weave, fine hair strands, stone grain. Skin must appear smooth and refined like polished marble. Keep EXACT same outline, silhouette, and tonal range. NO shadows, NO reflections, NO new light sources, NO dark areas under nose/eyes/lips. Output must overlay perfectly on original as bump map detail layer.",
+    
+    # From call_flux2_dev_detail_generation() in depth_logic.py - NO FACE
+    "details": "Preserve exact pixel alignment. Enhance this depth map by adding surface micro-details (fabric texture, hair strands, stone grain) using ONLY tonal variations within ±10% of local gray values. Keep EXACT same outline, silhouette, and overall tonal range. NO shadows, NO reflections, NO new light sources, NO dark areas. Output must overlay perfectly on original as bump map detail layer.",
 }
 
 # Pre-load embeddings at startup
@@ -39,9 +46,12 @@ def load_cached_embeddings():
     """Load pre-generated embeddings at startup."""
     global _cached_embeddings
     
+    # Updated to include all 4 embedding files
     embedding_files = {
         "relief": "relief.pt",
-        "details": "details.pt"
+        "relief_face": "relief_face.pt",
+        "details": "details.pt",
+        "details_face": "details_face.pt",
     }
     
     for key, filename in embedding_files.items():