fix: photoshoot image path

app.py

@@ -13,19 +13,22 @@ genai.configure(api_key=os.getenv("GENAI_API_KEY"))
 
 def remove_background(image):
     if isinstance(image, np.ndarray):
-        img = Image.fromarray(image)
-        buffered = BytesIO()
-        img.save(buffered, format="PNG")
-        img_base64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
-
-        output = replicate.run(
-            "cjwbw/rembg:fb8af171cfa1616ddcf1242c093f9c46bcada5ad4cf6f2fbe8b81b330ec5c003",
-            input={"image": f"data:image/png;base64,{img_base64}"}
-        )
+        try:
+            img = Image.fromarray(image)
+            buffered = BytesIO()
+            img.save(buffered, format="PNG")
+            img_base64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
+
+            output = replicate.run(
+                "cjwbw/rembg:fb8af171cfa1616ddcf1242c093f9c46bcada5ad4cf6f2fbe8b81b330ec5c003",
+                input={"image": f"data:image/png;base64,{img_base64}"}
+            )
 
-        response = requests.get(output)
-        return Image.open(BytesIO(response.content))
-    return None
+            response = requests.get(output)
+            return Image.open(BytesIO(response.content))
+        except Exception as e:
+            print(f"Error removing background: {str(e)}")
+            return None
 
 
 def load_background_images(folder_path):
@@ -36,16 +39,33 @@ def load_background_images(folder_path):
             if file.lower().endswith(('.png', '.jpg', '.jpeg'))]
 
 
-def replace_background(input_image, background_image):
-    if input_image is None or background_image is None:
+def replace_background(input_image_path, background_image):
+    if input_image_path is None or background_image is None:
+        return None
+
+    try:
+        input_img = Image.open(input_image_path)
+    except Exception as e:
+        print(f"Error loading input image: {e}")
         return None
 
-    foreground = remove_background(input_image)
+    input_img_np = np.array(input_img)
+
+    foreground = remove_background(input_img_np)
     if foreground is None:
         return None
 
     width, height = foreground.size
-    background = Image.fromarray(background_image) if isinstance(background_image, np.ndarray) else background_image
+
+    if isinstance(background_image, np.ndarray):
+        background = Image.fromarray(background_image)
+    else:
+        try:
+            background = Image.open(background_image) if isinstance(background_image, str) else background_image
+        except Exception as e:
+            print(f"Error loading background image: {e}")
+            return None
+
     background = background.resize((width, height))
 
     background.paste(foreground, (0, 0), foreground)
@@ -126,6 +146,12 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue=gr.themes.colors.red)) as dem
                     value="Reset"
                 )
 
+            submit_btn.click(
+                fn=replace_background,
+                inputs=[input_image, background_image],
+                outputs=output_image
+            )
+
         with gr.Tab("Product Description"):
             with gr.Row():
                 description_input = gr.Image(label="Product Image", type='filepath')
@@ -141,11 +167,6 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue=gr.themes.colors.red)) as dem
                 inputs=description_input,
                 outputs=description_output
             )
-            submit_btn.click(
-                fn=replace_background,
-                inputs=[input_image, background_image],
-                outputs=output_image
-            )
 
 if __name__ == "__main__":
     demo.launch(share=True, debug=True)