add api key

app.py

@@ -1,39 +1,44 @@
+import os
 import io
-import gradio as gr
 import requests
 import base64
+
+import gradio as gr
 from PIL import Image
 
 # define the function that will be called when the user inputs text
 def get_images(text):
-  headers = {'Content-Type': 'application/json'}
-  params = {
-      'return-images': 'true',
-      'number-results': '4',
-  }
-  response = requests.post(
-                            "https://wjdr33c1id.execute-api.eu-west-1.amazonaws.com/dev/prediction", 
-                            params=params, 
-                            headers=headers, 
-                            json={"data": text}
-                          )
-  # get the list of image data from the response
-  image_data = response.json()["image"]
-  # decode the base64-encoded image data and convert it to PIL images
-  images = [
-    Image.open(io.BytesIO(base64.b64decode(data))) for data in image_data
-    ]
-  # first comes on the top
-  images.reverse()
-  # return the list of images
-  return images
+    headers = {
+        "Content-Type": "application/json",
+        "x-api-key": os.environ["API_KEY"],
+    }
+    params = {
+        "return-images": "true",
+        "number-results": "4",
+    }
+    response = requests.post(
+        "https://wjdr33c1id.execute-api.eu-west-1.amazonaws.com/dev/prediction",
+        params=params,
+        headers=headers,
+        json={"data": text},
+    )
+    # get the list of image data from the response
+    image_data = response.json()["image"]
+    # decode the base64-encoded image data and convert it to PIL images
+    images = [Image.open(io.BytesIO(base64.b64decode(data))) for data in image_data]
+    # first comes on the top
+    images.reverse()
+    # return the list of images
+    return images
+
 
 # create the gradio app, passing the function as the input and output
-app = gr.Interface(get_images,
-                   gr.components.Textbox(label="Description"),
-                  #  gr.components.Image(label="Images", type="pil")
-                   gr.Gallery(label="Images")
+app = gr.Interface(
+    get_images,
+    gr.components.Textbox(label="Description"),
+    #  gr.components.Image(label="Images", type="pil")
+    gr.Gallery(label="Images"),
 )
 
 # start the app
-app.launch()
+app.launch()