Added handling for input image and path

app.py

@@ -586,8 +586,11 @@ def generate_image(input_image_path, prompt):
     resized_input_img = resize_to_square(input_image_path, 512)
 
     # resized_input_img_path = resized_input_img.save('input_img.png')
+    resized_input_img_path = '/tmp/gradio/resized_input_img.png'
 
-    ai_gen_image = generate_ai_bg(input_image_path, prompt)
+    resized_input_img.convert("RGBA").save(resized_input_img_path, "PNG")
+
+    ai_gen_image = generate_ai_bg(resized_input_img_path, prompt)
 
     mask_input_image = run_grounded_sam(resized_input_img)