Update app.py

app.py

@@ -320,6 +320,14 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_stre
     st = time.time()
     face_image = resize_image_aspect_ratio(face_image)
 
+    # Validate selection FIRST before any array access
+    if not selected_state and not custom_lora_path:
+        raise gr.Error("❌ You must select a style before generating")
+    
+    # Validate index is within bounds
+    if not custom_lora_path and (selected_state_index < 0 or selected_state_index >= len(sdxl_loras)):
+        raise gr.Error(f"❌ Invalid selection index: {selected_state_index}")
+
     # Enhanced face detection
     face_info_list = detect_faces(face_image, use_multiple_faces)
     face_detected = len(face_info_list) > 0
@@ -359,9 +367,7 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_stre
     
     print("Custom Loaded LoRA:", custom_lora_path)
 
-    if not selected_state and not custom_lora_path:
-        raise gr.Error("You must select a style")
-    elif custom_lora_path:
+    if custom_lora_path:
         repo_name = custom_lora_path
         full_path_lora = custom_lora_path
     else: