Handle image formats and paths

app.py

@@ -583,13 +583,15 @@ def blend_details(input_image, relit_image, masked_image, scaling_factor=1):
 @spaces.GPU
 def generate_image(input_image_path, prompt):
 
-    input_img = resize_to_square(input_image_path, 512)
+    resized_input_img = resize_to_square(input_image_path, 512)
 
-    ai_gen_image = generate_ai_bg(input_img, prompt)
+    resized_input_img_path = resized_input_img.save('input_img.jpg')
 
-    mask_input_image = run_grounded_sam(input_img)
+    ai_gen_image = generate_ai_bg(resized_input_img_path, prompt)
 
-    final_image = blend_details(input_img, ai_gen_image, mask_input_image)
+    mask_input_image = run_grounded_sam(resized_input_img)
+
+    final_image = blend_details(resized_input_img, ai_gen_image, mask_input_image)
 
     # final_image = ai_gen_image