Update app.py

app.py

@@ -135,6 +135,11 @@ def infer_with_lora(input_image, prompt, selected_index, custom_lora, seed=42, r
     """Generate image with selected LoRA"""
     global current_lora, pipe
     
+    # Check if input image is provided
+    if input_image is None:
+        gr.Warning("Please upload an image first!")
+        return None, seed, gr.update(visible=False)
+    
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     
@@ -167,8 +172,14 @@ def infer_with_lora(input_image, prompt, selected_index, custom_lora, seed=42, r
         print(f"using already loaded lora: {lora_to_use}")
     
     input_image = input_image.convert("RGB")
+    
+    # Check if LoRA is selected
+    if lora_to_use is None:
+        gr.Warning("Please select a LoRA style from the gallery first!")
+        return None, seed, gr.update(visible=False)
+    
     # Add trigger word to prompt
-    trigger_word = lora_to_use["trigger_word"]
+    trigger_word = lora_to_use.get("trigger_word", "")
     if trigger_word == ", How2Draw":
         prompt = f"create a How2Draw sketch of the person of the photo {prompt}, maintain the facial identity of the person and general features"
     elif trigger_word == ", video game screenshot in the style of THSMS":