Update app.py

app.py

@@ -1,13 +1,7 @@
-import os
 import requests
 import gradio as gr
 
-api_key = os.getenv("API_KEY")
-
-def generate_image(prompt, code):
-    if code != os.getenv("code"):
-        raise gr.Error("❗ Не верный ключ!")
-        return None
+def generate_image(prompt, api_key, request_url):
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json",
@@ -16,25 +10,31 @@ def generate_image(prompt, code):
         "prompt": prompt,
         "num_images": 1,
     }
-    response = requests.post("https://api.openai.com/v1/images/generations", headers=headers, json=data)
+    response = requests.post(request_url, headers=headers, json=data)
     response_json = response.json()
     
-    # Получаем URL первого сгенерированного изображения
+    # Get the URL of the first generated image
     image_url = response_json["data"][0]["url"]
     
     return image_url
+
 css = """
 footer {visibility: hidden !important;}
 """
-# Создание интерфейса с помощью Gradio
+
+# Create the interface using Gradio
 with gr.Blocks(css=css, theme='YTheme/Sketch') as demo:
     with gr.Row():
-        code = gr.Textbox(label="Ключ доступа", type="password")
+        gr.Markdown("## Image Generation Tool")
+        gr.Markdown("Enter the required information below to generate an image.")
+    with gr.Row():
+        api_key_input = gr.Textbox(label="API Key", type="password")
+        request_url_input = gr.Textbox(label="Request URL", value="https://api.openai.com/v1/images/generations")
     with gr.Row():
-        prompt_input = gr.Textbox(label="Описание изображения")
-    submit_btn = gr.Button("Генерация", variant='primary')
-    image_output = gr.Image(label="Изображение")
+        prompt_input = gr.Textbox(label="Image Description")
+    submit_btn = gr.Button("Generate Image", variant='primary')
+    image_output = gr.Image(label="Generated Image")
     
-    submit_btn.click(fn=generate_image, inputs=[prompt_input, code], outputs=image_output)
+    submit_btn.click(fn=generate_image, inputs=[prompt_input, api_key_input, request_url_input], outputs=image_output)
 
-demo.launch()
+demo.launch()