Update app.py

app.py

@@ -4,7 +4,10 @@ import gradio as gr
 
 api_key = os.getenv("API_KEY")
 
-def generate_image(prompt):
+def generate_image(prompt, code):
+    if code != os.getenv("code"):
+        raise gr.Error("❗ Не верный ключ!")
+        return None
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json",
@@ -24,11 +27,12 @@ def generate_image(prompt):
 # Создание интерфейса с помощью Gradio
 with gr.Blocks() as demo:
     with gr.Row():
-        prompt_input = gr.Textbox(label="Описание изображения")
-        submit_btn = gr.Button("Генерация")
+        code = gr.Textbox(label="Ключ доступа")
     with gr.Row():
-        image_output = gr.Image(label="Изображение")
+        prompt_input = gr.Textbox(label="Описание изображения")
+    submit_btn = gr.Button("Генерация")
+    image_output = gr.Image(label="Изображение")
     
-    submit_btn.click(fn=generate_image, inputs=prompt_input, outputs=image_output)
+    submit_btn.click(fn=generate_image, inputs=[prompt_input, code], outputs=image_output)
 
 demo.launch()