update approach + add logs

app.py

@@ -79,19 +79,30 @@ def calculate_optimal_dimensions(image: Image.Image):
 @spaces.GPU(durations=300)
 def infer(edit_images, prompt, lora_model, strength, seed=42, randomize_seed=False, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
     # pipe.enable_xformers_memory_efficient_attention()
+    gr.Info("Infering")
 
     if lora_model != "None":
         pipe.load_lora_weights(lora_models[lora_model])
         pipe.enable_lora()
 
-    image = edit_images["background"]
+    image_path = edit_images["background"]
+    mask_path = edit_images["layers"][0]
+
+    gr.info("starting checks")
+
+    image: Image.Image = Image.open(image_path)
+    mask: Image.Image = Image.open(mask_path)
+
+    if not image:
+        gr.Info("Please upload an image.")
+        return None, None
+
     width, height = calculate_optimal_dimensions(image)
-    mask = edit_images["layers"][0]
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
     # controlImage = processor(image)
-    print("generating image")
+    gr.info("generating image")
     image = pipe(
         # mask_image_latent=vae.encode(controlImage),
         prompt=prompt,