Update config.py

config.py

@@ -15,8 +15,12 @@ HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN", None)
 
 # Model files
 MODEL_FILES = {
-    "checkpoint": "horizon.safensors",
-    "lora": "retroart.safetensors",
+    "checkpoint": "horizon.safetensors",
+    # --- START FIX: Define all LORA files ---
+    "lora_retroart": "retroart.safetensors",
+    "lora_vga": "vga.safetensors",
+    "lora_lucasart": "lucasart.safetensors",
+    # --- END FIX ---
     "vae": "pixelate.safetensors"
 }
 
@@ -29,8 +33,15 @@ INSTANTID_CONFIG = {
     "face_model_repo": "DIAMONIK7777/antelopev2"
 }
 
+# --- START FIX: Update TRIGGER_WORD to be a dictionary ---
 # LORA configuration
-TRIGGER_WORD = "p1x3l4rt, pixel art"
+TRIGGER_WORD = {
+    "RetroArt": "p1x3l4rt, pixel art",
+    "VGA": "dosvga style",
+    "LucasArt": "lucasarts style",
+    "None": ""  # No trigger word when no LORA is selected
+}
+# --- END FIX ---
 
 # Face detection configuration
 FACE_DETECTION_CONFIG = {
@@ -62,10 +73,8 @@ DEFAULT_PARAMS = {
     "enable_color_matching": False,
     "consistency_mode": True,
     "seed": -1,
-    # --- START FIX: Add missing default parameters ---
     "expression_control_scale": 0.6,
     "lora_choice": "RetroArt"
-    # --- END FIX ---
 }
 
 # Optimized preset configurations
@@ -77,9 +86,7 @@ PRESETS = {
         "lora_scale": 0.8,
         "depth_control_scale": 0.65,
         "identity_control_scale": 0.95,
-        # --- START FIX: Add missing parameters ---
         "expression_control_scale": 0.5,
-        # --- END FIX ---
         "description": "Maximum face preservation - 96-98% similarity (Level 3)"
     },
     "Premium Portrait": {
@@ -89,9 +96,7 @@ PRESETS = {
         "lora_scale": 1.1,
         "depth_control_scale": 0.75,
         "identity_control_scale": 0.85,
-        # --- START FIX: Add missing parameters ---
         "expression_control_scale": 0.6,
-        # --- END FIX ---
         "description": "Optimized balanced - strong pixel art + excellent face (92-94% similarity)"
     },
     "Balanced Portrait": {
@@ -101,9 +106,7 @@ PRESETS = {
         "lora_scale": 1.0,
         "depth_control_scale": 0.75,
         "identity_control_scale": 0.85,
-        # --- START FIX: Add missing parameters ---
         "expression_control_scale": 0.6,
-        # --- END FIX ---
         "description": "Good balance between fidelity and style - 90-93% similarity"
     },
     "Artistic Excellence": {
@@ -113,9 +116,7 @@ PRESETS = {
         "lora_scale": 1.2,
         "depth_control_scale": 0.78,
         "identity_control_scale": 0.75,
-        # --- START FIX: Add missing parameters ---
         "expression_control_scale": 0.7,
-        # --- END FIX ---
         "description": "Creative with strong likeness - 88-91% similarity"
     },
     "Style Focus": {
@@ -125,9 +126,7 @@ PRESETS = {
         "lora_scale": 1.4,
         "depth_control_scale": 0.82,
         "identity_control_scale": 0.65,
-        # --- START FIX: Add missing parameters ---
         "expression_control_scale": 0.7,
-        # --- END FIX ---
         "description": "Maximum pixel art style - 83-87% similarity"
     },
     "Subtle Enhancement": {
@@ -137,9 +136,7 @@ PRESETS = {
         "lora_scale": 0.75,
         "depth_control_scale": 0.60,
         "identity_control_scale": 0.98,
-        # --- START FIX: Add missing parameters ---
         "expression_control_scale": 0.4,
-        # --- END FIX ---
         "description": "Minimal transformation, photo-realistic - 97-99% similarity"
     }
 }
@@ -229,7 +226,5 @@ print(f"  HuggingFace Token: {'Set' if HUGGINGFACE_TOKEN else 'Not set (using IP
 print(f"  InstantID: Enabled")
 
 
-# --- START FIX: Add missing LORA_CHOICES list ---
 # This list must match the LORA adapter names loaded in models.py
-LORA_CHOICES = ["RetroArt", "VGA", "LucasArt", "None"]
-# --- END FIX ---
+LORA_CHOICES = ["RetroArt", "VGA", "LucasArt", "None"]