add secret key

app.py

@@ -1,3 +1,4 @@
+import os
 from io import BytesIO
 
 import gradio as gr
@@ -7,29 +8,33 @@ import requests
 from PIL import Image
 
 
-def generate(prompt: str):
+def generate(prompt: str, secret_key: str):
     """
     プロンプトから生成画像(PIL.Image.open)を取得
     """
-    output = replicate.run(
-        "stability-ai/sdxl:2b017d9b67edd2ee1401238df49d75da53c523f36e363881e057f5dc3ed3c5b2",
-        input={"prompt": prompt,
-               "seed": np.random.randint(1, 1001)},
-    )
-    # リンク取得
-    png_link = output[0]
-    # PNGファイルをリンクから取得
-    response = requests.get(png_link)
-    # イメージをメモリ上に開く
-    img = Image.open(BytesIO(response.content))
-    return img
+    if secret_key == os.environ["SECRET_KEY"]:
+        output = replicate.run(
+            "stability-ai/sdxl:2b017d9b67edd2ee1401238df49d75da53c523f36e363881e057f5dc3ed3c5b2",
+            input={"prompt": prompt,
+                "seed": np.random.randint(1, 1001)},
+        )
+        # リンク取得
+        png_link = output[0]
+        # PNGファイルをリンクから取得
+        response = requests.get(png_link)
+        # イメージをメモリ上に開く
+        img = Image.open(BytesIO(response.content))
+        return img
+
 
 
 demo = gr.Interface(
     generate,
-    inputs=[gr.Textbox(label='プロンプト')],
+    inputs=[gr.Textbox(label='プロンプト'), gr.Textbox(label='secret key')],
     outputs=["image"],
-    examples=["An astronaut riding a rainbow unicorn, cinematic, dramatic"],
+    examples=[["An astronaut riding a rainbow unicorn, cinematic, dramatic", ""]],
+    allow_flagging=None,
+    title="Stable Diffusion XL (SDXL 1.0)"
 )
 
 if __name__ == "__main__":