Update app.py

app.py

@@ -5,7 +5,7 @@ import spaces
 import gradio as gr
 import os
 
-from config import PRESETS, DEFAULT_PARAMS, TRIGGER_WORD
+from config import PRESETS, DEFAULT_PARAMS, TRIGGER_WORD, LORA_CHOICES
 from generator import RetroArtConverter
 
 
@@ -42,6 +42,7 @@ def process_image(
     depth_control_scale,
     identity_control_scale,
     expression_control_scale, # <-- NEW
+    lora_choice,
     lora_scale,
     identity_preservation,
     strength,
@@ -65,6 +66,7 @@ def process_image(
             depth_control_scale=depth_control_scale,
             identity_control_scale=identity_control_scale,
             expression_control_scale=expression_control_scale, # <-- NEW
+            lora_choice=lora_choice,
             lora_scale=lora_scale,
             identity_preservation=identity_preservation,
             strength=strength,
@@ -107,10 +109,29 @@ def get_model_status():
     if converter.models_loaded:
         status_text = "**[OK] Loaded Models:**\n"
         status_text += f"- Custom Checkpoint (Horizon): {'[OK] Loaded' if converter.models_loaded['custom_checkpoint'] else '[OK] Using SDXL base'}\n"
-        status_text += f"- LORA (RetroArt): {'[OK] Loaded' if converter.models_loaded['lora'] else ' Disabled'}\n"
+        
+        # Updated LORA status
+        lora_status = 'Disabled'
+        if converter.models_loaded['lora']:
+            loaded_count = sum(1 for loaded in converter.loaded_loras.values() if loaded)
+            if loaded_count > 0:
+                lora_status = f"[OK] Loaded {loaded_count}/3"
+            else:
+                lora_status = "[ERROR] All failed"
+        status_text += f"- LORAs (Retro, VGA, ...): {lora_status}\n"
+        
         status_text += f"- InstantID: {'[OK] Loaded' if converter.models_loaded['instantid'] else ' Disabled'}\n"
-        status_text += f"- Zoe Depth: {'[OK] Loaded' if converter.models_loaded['zoe_depth'] else ' Fallback'}\n"
-        status_text += f"- OpenPose (Expression): {'[OK] Loaded' if converter.models_loaded.get('openpose', False) else ' Disabled'}\n" # <-- NEW
+        
+        # Show depth detector type
+        depth_type = converter.models_loaded.get('depth_type', 'unknown')
+        depth_loaded = converter.models_loaded.get('depth_detector', False)
+        if depth_loaded and depth_type:
+            status_text += f"- Depth Detector: [OK] {depth_type.upper()} Loaded\n"
+        else:
+            status_text += f"- Depth Detector: Fallback (grayscale)\n"
+        
+        status_text += f"- OpenPose (Expression): {'[OK] Loaded' if converter.models_loaded.get('openpose', False) else ' Disabled'}\n"
+        status_text += f"- MediapipeFace: {'[OK] Loaded' if converter.models_loaded.get('mediapipe_face', False) else ' Disabled'}\n"
         status_text += f"- IP-Adapter (Face Embeddings): {'[OK] Loaded' if converter.models_loaded.get('ip_adapter', False) else ' Keypoints only'}\n"
         return status_text
     return "**Model status unavailable**"
@@ -177,7 +198,8 @@ with gr.Blocks(title="Pixagram - AI Pixel Art Generator", theme=gr.themes.Soft()
     scheduler_info = f"""
     **[CONFIG] Advanced Configuration:**
     - Pipeline: **Img2Img** (structure preservation)
-    - Face System: **CLIP + InsightFace** (dual embeddings)
+    - Face System: **CLIP + InsightFace + MediapipeFace** (triple detection)
+    - **Depth Detection:** Hierarchical (Leres → Zoe → Midas) - best available automatically selected
     - **[NEW] Expression Control:** OpenPose-Face (68 keypoints)
     - **[ADVANCED] Enhanced Resampler:** 10 layers, 20 heads (+3-5% quality)
     - **[ADVANCED] Adaptive Attention:** Context-aware scaling (+2-3% quality)
@@ -268,12 +290,21 @@ with gr.Blocks(title="Pixagram - AI Pixel Art Generator", theme=gr.themes.Soft()
                         label="Depth ControlNet Scale"
                     )
                     
+                    # --- NEW: LORA Dropdown ---
+                    lora_choice = gr.Dropdown(
+                        label="LORA Style",
+                        choices=LORA_CHOICES,
+                        value=DEFAULT_PARAMS['lora_choice'],
+                    )
+                    # --- END NEW ---
+                    
+                with gr.Row():
                     lora_scale = gr.Slider(
-                        minimum=0.5,
+                        minimum=0.0,
                         maximum=2.0,
                         value=DEFAULT_PARAMS['lora_scale'],
                         step=0.05,
-                        label="RetroArt LORA Scale\nLower = more realistic"
+                        label="LORA Scale\nIntensity for selected style" # <-- Modified Label
                     )
             
             with gr.Accordion(" InstantID Settings (for portraits)", open=True):
@@ -367,7 +398,7 @@ with gr.Blocks(title="Pixagram - AI Pixel Art Generator", theme=gr.themes.Soft()
             **[ADAPTIVE] Automatic Adjustments:**
             - Small faces (< 50K px): Boosts identity preservation to 1.8
             - Low confidence (< 80%): Increases identity control to 0.9
-            - Profile views (> 20° yaw): Enhances preservation to 1.7
+            - Profile views (> 20° yaw): Enhances preservation to 1.7
             - Good quality faces: Uses your selected parameters
             
             **[PARAMETERS] Parameter Relationships:**
@@ -447,13 +478,13 @@ with gr.Blocks(title="Pixagram - AI Pixel Art Generator", theme=gr.themes.Soft()
         outputs=all_sliders
     )
     
-    # --- MODIFIED: generate_btn.click (added expression_control_scale) ---
+    # --- MODIFIED: generate_btn.click (added lora_choice, expression_control_scale) ---
     generate_btn.click(
         fn=process_image,
         inputs=[
             input_image, prompt, negative_prompt, steps, guidance_scale,
             depth_control_scale, identity_control_scale, expression_control_scale, 
-            lora_scale, identity_preservation, strength, enable_color_matching, 
+            lora_choice, lora_scale, identity_preservation, strength, enable_color_matching, 
             consistency_mode, seed_input, enable_captions
         ],
         outputs=[output_image, caption_output]