update app

app.py

@@ -62,16 +62,16 @@ orange_red_theme = OrangeRedTheme()
 MAX_SEED = np.iinfo(np.int32).max
 
 ADAPTER_SPECS = {
-    "h4rd8are": {
-        "repo": "siraxe/h4rd8are_klein9b",
-        "weights": "turn_to_h4rd8are_klein9b.safetensors",
-        "adapter_name": "h4rd8are_klein9b"
-    },
     "Klein-Delight-Style": {
         "repo": "linoyts/Flux2-Klein-Delight-LoRA",
         "weights": "pytorch_lora_weights.safetensors",
         "adapter_name": "klein-delight"
-    }
+    },
+    # "h4rd8are": {
+    #    "repo": "siraxe/h4rd8are_klein9b",
+    #    "weights": "turn_to_h4rd8are_klein9b.safetensors",
+    #    "adapter_name": "h4rd8are_klein9b"
+    # },
 }
 
 LOADED_ADAPTERS = set()
@@ -116,29 +116,33 @@ def infer(
     if not input_image:
         raise gr.Error("Please upload an image to apply a style to.")
 
-    spec = ADAPTER_SPECS.get(lora_adapter)
-    if spec:
-        adapter_name = spec["adapter_name"]
-        
-        if adapter_name not in LOADED_ADAPTERS:
-            print(f"--- Downloading and Loading Adapter: {lora_adapter} ---")
-            try:
-                pipe.load_lora_weights(
-                    spec["repo"], 
-                    weight_name=spec["weights"], 
-                    adapter_name=adapter_name
-                )
-                LOADED_ADAPTERS.add(adapter_name)
-            except Exception as e:
-                raise gr.Error(f"Failed to load adapter {lora_adapter}: {e}")
-        else:
-            print(f"--- Adapter {lora_adapter} is already loaded. ---")
-            
-        print(f"Activating LoRA: {adapter_name}")
-        pipe.set_adapters([adapter_name], adapter_weights=[1.0])
-    else:
-        print("No valid LoRA selected or found. Disabling adapters.")
+    if lora_adapter == "None":
+        print("Selection is None. Disabling LoRA adapters.")
         pipe.disable_lora()
+    else:
+        spec = ADAPTER_SPECS.get(lora_adapter)
+        if spec:
+            adapter_name = spec["adapter_name"]
+            
+            if adapter_name not in LOADED_ADAPTERS:
+                print(f"--- Downloading and Loading Adapter: {lora_adapter} ---")
+                try:
+                    pipe.load_lora_weights(
+                        spec["repo"], 
+                        weight_name=spec["weights"], 
+                        adapter_name=adapter_name
+                    )
+                    LOADED_ADAPTERS.add(adapter_name)
+                except Exception as e:
+                    raise gr.Error(f"Failed to load adapter {lora_adapter}: {e}")
+            else:
+                print(f"--- Adapter {lora_adapter} is already loaded. ---")
+                
+            print(f"Activating LoRA: {adapter_name}")
+            pipe.set_adapters([adapter_name], adapter_weights=[1.0])
+        else:
+            print("Adapter not found in specs. Disabling LoRA.")
+            pipe.disable_lora()
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -202,7 +206,7 @@ with gr.Blocks() as demo:
                 with gr.Row():
                     prompt = gr.Text(
                         label="Edit Prompt", 
-                        max_lines=2,
+                        max_lines=1,
                         show_label=True, 
                         placeholder="e.g., a man with a red superhero mask"
                     )
@@ -222,14 +226,14 @@ with gr.Blocks() as demo:
                 with gr.Row():
                     lora_adapter = gr.Dropdown(
                         label="Choose Editing Style",
-                        choices=list(ADAPTER_SPECS.keys()),
+                        choices=["None"] + list(ADAPTER_SPECS.keys()),
                         value="Klein-Delight-Style"
                     )
                     
         gr.Examples(
             examples=[
-                ["examples/1.jpg", "Relight the image to remove all existing lighting conditions and replace them with neutral, uniform illumination. Apply soft, evenly distributed lighting with no directional shadows, no harsh highlights, and no dramatic contrast. Maintain the original identity of all subjects exactly—preserve facial structure, skin tone, proportions, expressions, hair, clothing, and textures. Do not alter pose, camera angle, background geometry, or image composition. Lighting should appear balanced, and studio-neutral, similar to diffuse overcast or a soft lightbox setup. Ensure consistent exposure across the entire image with realistic depth and subtle shading only where necessary for form.Relight the image to remove all existing lighting conditions and replace them with neutral, uniform illumination. Apply soft, evenly distributed lighting with no directional shadows, no harsh highlights, and no dramatic contrast. Maintain the original identity of all subjects exactly—preserve facial structure, skin tone, proportions, expressions, hair, clothing, and textures. Do not alter pose, camera angle, background geometry, or image composition. Lighting should appear balanced, and studio-neutral, similar to diffuse overcast or a soft lightbox setup. Ensure consistent exposure across the entire image with realistic depth and subtle shading only where necessary for form.", "Klein-Delight-Style"],
-                ["examples/portrait.jpg", "cinematic lighting, high contrast", "h4rd8are"],
+                ["examples/1.jpg", "Relight the image to remove all existing lighting conditions and replace them with neutral, uniform illumination. Apply soft, evenly distributed lighting with no directional shadows, no harsh highlights, and no dramatic contrast. Maintain the original identity of all subjects exactly—preserve facial structure, skin tone, proportions, expressions, hair, clothing, and textures. Do not alter pose, camera angle, background geometry, or image composition. Lighting should appear balanced, and studio-neutral, similar to diffuse overcast or a soft lightbox setup. Ensure consistent exposure across the entire image with realistic depth and subtle shading only where necessary for form.", "Klein-Delight-Style"],
+                # ["examples/2.jpg", "Make it look cinematic", "None"], 
             ],
             inputs=[input_image, prompt, lora_adapter],
             outputs=[output_image, used_seed],