zukijourney / app.py
2ch's picture
Create app.py
a7fe9c7 verified
import gradio as gr
from openai import OpenAI
def generate_images(token, size, quality, style, prompt):
client = OpenAI(api_key= token, base_url= "https://zukijourney.xyzbot.net/v1")
response = client.images.generate(prompt=prompt, model = 'dall-e-3', n=1, quality=quality, size=size, style=style)
return response.data[0].url
with gr.Blocks(analytics_enabled=False) as demo:
gr.Markdown("Generate images using OpenAI's API")
with gr.Row():
token_input = gr.Textbox(label="API Token", placeholder="Enter your OpenAI API token here")
size_input = gr.Radio(choices=["1024x1024", "1792x1024", "1024x1792"], label="Image Size", value="1024x1024")
quality_input = gr.Radio(choices=["standard", "hd"], label="Quality", value="standard")
style_input = gr.Radio(choices=["vivid", "natural"], label="Style", value="vivid")
prompt_input = gr.Textbox(label="Prompt", placeholder="Enter a prompt for the image generation")
submit_button = gr.Button("Generate")
output_images = gr.Gallery(label="Generated Images")
submit_button.click(
generate_images,
inputs=[token_input, size_input, quality_input, style_input, prompt_input],
outputs=output_images
)
demo.launch()