Update app.py

app.py

@@ -22,21 +22,30 @@ import numpy as np
 # Configuration Section (Modify here to expand)
 # ======================
 
-# 1. Base Model - Illustrious XL v1.0
-BASE_MODEL = "OnomaAIResearch/Illustrious-XL-v1.0"
+# 1. Base Model - Illustrious XL v1.0 (Multiple options for stability)
+BASE_MODELS = {
+    "illustrious_xl_v1": "Liberata/illustrious-xl-v1.0",  # Primary choice - more stable
+    "illustrious_xl_v2": "OnomaAIResearch/Illustrious-XL-v2.0",  # Backup option
+    "sdxl_base": "stabilityai/stable-diffusion-xl-base-1.0",  # Fallback to standard SDXL
+    "realistic_vision": "SG161222/RealVisXL_V4.0"  # Alternative realistic model
+}
+
+# Current model selection (change this to switch models)
+CURRENT_MODEL_KEY = "illustrious_xl_v1"
+BASE_MODEL = BASE_MODELS[CURRENT_MODEL_KEY]
 
-# 2. Fixed LoRAs (Auto-loaded, not user-selectable)
+# 2. Fixed LoRAs (Auto-loaded, not user-selectable) - Using reliable SDXL LoRAs
 FIXED_LORAS = {
     "quality_enhancer": {
-        "repo_id": "stabilityai/stable-diffusion-xl-base-1.0",
-        "filename": "sd_xl_base_1.0.safetensors",
-        "weight": 0.8,
-        "trigger_words": "high quality, detailed, sharp"
+        "repo_id": "ByteDance/SDXL-Lightning",
+        "filename": None,
+        "weight": 0.6,
+        "trigger_words": "high quality, detailed, masterpiece"
     },
     "pose_control": {
         "repo_id": "latent-consistency/lcm-lora-sdxl",
         "filename": None,
-        "weight": 0.7,
+        "weight": 0.5,
         "trigger_words": "perfect anatomy, natural pose"
     }
 }
@@ -50,7 +59,7 @@ STYLE_PROMPTS = {
     "Watercolor": "watercolor painting, soft brush strokes, translucent layers, artistic, painterly, paper texture, traditional art, masterpiece, ",
 }
 
-# 4. Optional LoRAs (User-selectable via dropdown, can select multiple)
+# 4. Optional LoRAs (User-selectable via dropdown, can select multiple) - Using verified SDXL LoRAs
 OPTIONAL_LORAS = {
     "None": {
         "repo_id": None,
@@ -58,35 +67,35 @@ OPTIONAL_LORAS = {
         "trigger_words": "",
         "description": "No additional LoRA"
     },
-    "Detail Enhancement": {
+    "Lightning Speed": {
         "repo_id": "ByteDance/SDXL-Lightning",
         "weight": 0.8,
-        "trigger_words": "extremely detailed, intricate details, hyperdetailed",
-        "description": "Enhances fine details and textures"
+        "trigger_words": "high quality, detailed, fast generation",
+        "description": "Fast generation with quality enhancement"
     },
-    "Portrait Master": {
-        "repo_id": "ostris/face-helper-sdxl-lora",
+    "LCM LoRA": {
+        "repo_id": "latent-consistency/lcm-lora-sdxl",
         "weight": 0.9,
-        "trigger_words": "perfect face, beautiful eyes, detailed skin texture",
-        "description": "Specializes in realistic portraits"
+        "trigger_words": "lcm, high quality, detailed",
+        "description": "Latent Consistency Model for faster inference"
     },
-    "Cinematic Style": {
-        "repo_id": "goofyai/cyborg_style_xl",
+    "Offset Noise": {
+        "repo_id": "stabilityai/stable-diffusion-xl-offset-example-lora",
         "weight": 0.7,
-        "trigger_words": "cinematic lighting, dramatic shadows, film noir",
-        "description": "Adds cinematic atmosphere"
-    },
-    "Vintage Photo": {
-        "repo_id": "artificialguybr/LogoRedmond-LogoLoraForSDXL-V2",
-        "weight": 0.6,
-        "trigger_words": "vintage photo, retro style, film photography",
-        "description": "Vintage photography effects"
+        "trigger_words": "high contrast, dramatic lighting",
+        "description": "Enhanced contrast and lighting effects"
     },
-    "Art Nouveau": {
-        "repo_id": "ostris/super-cereal-sdxl-lora",
+    "AnimeDiff": {
+        "repo_id": "Linaqruf/anime-detailer-xl-lora",
         "weight": 0.8,
-        "trigger_words": "art nouveau style, ornate decorations, flowing lines",
-        "description": "Art Nouveau artistic style"
+        "trigger_words": "anime style, detailed anime, cel shading",
+        "description": "Anime and manga style enhancement"
+    },
+    "PhotoReal": {
+        "repo_id": "Jovian-Experiments/photorealistic-xl-lora",
+        "weight": 0.9,
+        "trigger_words": "photorealistic, ultra realistic, 8k uhd",
+        "description": "Photorealistic enhancement"
     }
 }