Update app.py

app.py

@@ -1,27 +1,22 @@
 import os
-import requests
 import gradio as gr
-
-api_key = os.getenv("API_KEY")
+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
-    headers = {
-        "Authorization": f"Bearer {api_key}",
-        "Content-Type": "application/json",
-    }
-    data = {
-        "prompt": prompt,
-        "num_images": 1,
-    }
-    response = requests.post("https://api.openai.com/v1/images/generations", headers=headers, json=data)
-    response_json = response.json()
-    
-    # Получаем URL первого сгенерированного изображения
-    image_url = response_json["data"][0]["url"]
-    
+    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;}