added handling of no image provided

app.py

@@ -86,21 +86,35 @@ def apply_canny(image, low_threshold, high_threshold):
 def generate_image(prompt, input_image, low_threshold, high_threshold, strength, guidance, controlnet_conditioning_scale):
 
     print(pipe.get_active_adapters())
-    
-    # Apply edge detection
-    edge_detected = apply_canny(input_image, low_threshold, high_threshold)
-    
-    # Generate styled image using ControlNet
-    result = pipe(
-        prompt=prompt,
-        image=edge_detected,
-        num_inference_steps=30,
-        guidance_scale=guidance,
-        controlnet_conditioning_scale=float(controlnet_conditioning_scale),
-        strength=strength
-    ).images[0]
-    
-    return edge_detected, result
+    print(input_image)
+
+    ## execute if an image is provided
+    if input_image is not None:
+        # Apply edge detection
+        edge_detected = apply_canny(input_image, low_threshold, high_threshold)
+        
+        # Generate styled image using ControlNet
+        result = pipe(
+            prompt=prompt,
+            image=edge_detected,
+            num_inference_steps=30,
+            guidance_scale=guidance,
+            controlnet_conditioning_scale=float(controlnet_conditioning_scale),
+            strength=strength
+        ).images[0]
+        
+        return edge_detected, result
+
+    else:
+        # Generate styled image using ControlNet
+        result = pipe(
+            prompt=prompt,
+            num_inference_steps=30,
+            guidance_scale=guidance,
+            strength=strength
+        ).images[0]
+
+        return None, result
 
 # 🖥️ Gradio UI
 with gr.Blocks() as demo: