token client

app.py

@@ -42,24 +42,36 @@ def process_image(image: np.ndarray):
     image_pil.save(buffered, format="PNG")
     img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
 
-    # Send image to backend
+    # Get API key from environment variable
+    api_key = os.getenv('API_KEY')
+
+    if not api_key:
+        raise ValueError("API_KEY is not set in the environment variables")
+
+        # Send image to backend with API key in headers
     response = requests.post(
-        os.getenv('BACKEND_URL'),
+        os.getenv('BACKEND_URL') + "/process_image/",
+        headers={"access_token": api_key},
         files={"file": ("image.png", base64.b64decode(img_str), "image/png")}
     )
 
-    # Process the response
+    # Check if the response is successful
+    if response.status_code != 200:
+        raise Exception(f"Request failed with status code {response.status_code}: {response.text}")
+
+        # Process the response
     result = response.json()
     processed_image_b64 = result["processed_image"]
     processed_image = Image.open(BytesIO(base64.b64decode(processed_image_b64)))
 
     # Save the processed image
+    output_folder = Path("output")  # Make sure this folder exists or create it
+    output_folder.mkdir(parents=True, exist_ok=True)
     image_path = output_folder / f"no_bg_image_{uuid.uuid4().hex}.png"
     processed_image.save(image_path)
 
     return (processed_image, image_pil), str(image_path)
 
-
 # Define inputs and outputs for the Gradio interface
 image = gr.Image(label="Upload a photo")
 output_slider = ImageSlider(label="Processed photo", type="pil")