Update app.py

app.py

@@ -1,53 +1,31 @@
 # app.py
-# FINAL STABLE VERSION using gr.Interface and an explicit api_name
+# A simple and reliable Gradio interface for background removal.
 
 import gradio as gr
 from rembg import remove
-from PIL import Image
-import os
-import traceback
-
-# --- Configuration ---
-MY_SECRET_KEY = os.environ.get("MY_SECRET_KEY", "Mbdh837hdjnfjsfujfeifjjweiojfe")
-
-# --- Core Function ---
-def process_image(input_image, secret_key):
-    """
-    The main function that processes the image.
-    """
-    # Security Check
-    if secret_key != MY_SECRET_KEY:
-        raise gr.Error("Access Denied: The secret key is incorrect.")
 
+# The core function that performs the background removal.
+def process_image(input_image):
     if input_image is None:
-        raise gr.Error("No image provided. Please upload an image.")
-
+        raise gr.Error("No image provided.")
     try:
-        # Use the default, stable rembg model
-        output_image = remove(input_image)
-        return output_image
-        
+        # Use the default rembg model for stability.
+        return remove(input_image)
     except Exception as e:
-        print(traceback.format_exc())
-        raise gr.Error(f"Failed to process image. Error: {e}")
+        raise gr.Error(f"Failed to process image: {e}")
 
-# --- Gradio Interface Definition ---
+# Create the Gradio interface.
+# The `api_name="predict"` explicitly creates an endpoint at `/run/predict`.
 demo = gr.Interface(
     fn=process_image,
-    inputs=[
-        gr.Image(type="pil", label="Upload Your Image"),
-        gr.Textbox(label="Secret Key", type="password", placeholder="API Key Required")
-    ],
-    outputs=[
-        gr.Image(type="pil", label="Background Removed")
-    ],
-    title="Avinya.Space Background Remover API",
-    description="This is the API backend for Avinya.Space. It is not intended for direct use.",
+    inputs=gr.Image(type="pil", label="Upload Image"),
+    outputs=gr.Image(type="pil", label="Background Removed"),
+    title="Background Remover API",
+    description="A simple API for removing image backgrounds.",
     allow_flagging="never",
-    # This explicitly creates the API endpoint at /run/predict
-    api_name="predict" 
+    api_name="predict"
 )
 
-# --- Launch the App ---
+# Launch the app.
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()