Update app.py

app.py

@@ -1,21 +1,26 @@
 import gradio as gr
 from rembg import remove
 from PIL import Image
-import io
 
-def remove_background(input_image):
-    if isinstance(input_image, str):
-        input_image = Image.open(input_image)
+# Set your secret API key here
+API_KEY = "mysecret123"
+
+def remove_background(input_image, key):
+    if key != API_KEY:
+        raise gr.Error("Invalid API Key!")
+    
     output_image = remove(input_image)
     return output_image
 
 demo = gr.Interface(
     fn=remove_background,
-    inputs=gr.Image(type="pil"),
+    inputs=[
+        gr.Image(type="pil", label="Upload Image"),
+        gr.Text(label="API Key")
+    ],
     outputs=gr.Image(type="pil"),
-    title="Background Remover",
-    description="Upload an image to remove its background",
-    examples=[],
+    title="Protected Background Remover",
+    description="Enter your API key to use this app",
     theme=gr.themes.Soft()
 )