Create main.py

main.py

@@ -0,0 +1,38 @@
+import os
+import gradio as gr
+from openai import OpenAI
+client = OpenAI(api_key=os.getenv("API_KEY"))
+
+def generate_image(prompt, code):
+    if code != os.getenv("code"):
+        raise gr.Error("❗ Не верный ключ!")
+        return None
+    response = client.images.generate(
+    model="dall-e-3",
+    prompt=prompt,
+    size="1024x1024",
+    quality="hd",
+    n=1,
+    )
+
+    image_url = response.data[0].url
+    print(image_url)
+    return image_url
+css = """
+footer {visibility: hidden !important;}
+"""
+# Создание интерфейса с помощью Gradio
+with gr.Blocks(css=css, theme='YTheme/Sketch') as demo:
+    with gr.Row():
+        with gr.Column():
+            with gr.Row():
+                code = gr.Textbox(label="Ключ доступа", type="password")
+            with gr.Row():
+                prompt_input = gr.Textbox(label="Описание изображения", lines=3)
+            submit_btn = gr.Button("Генерация", variant='primary')
+        with gr.Column():
+            image_output = gr.Image(label="Изображение")
+    
+    submit_btn.click(fn=generate_image, inputs=[prompt_input, code], outputs=image_output)
+
+demo.launch()