| | import requests |
| | import gradio as gr |
| |
|
| | def generate_image(prompt, quality, size, style, api_key, request_url): |
| | headers = { |
| | "Authorization": f"Bearer {api_key}", |
| | "Content-Type": "application/json", |
| | } |
| | data = { |
| | "prompt": prompt, |
| | "model": "dall-e-3", |
| | "quality": quality, |
| | "size": size, |
| | "style": style, |
| | "num_images": 1, |
| | } |
| | response = requests.post(request_url, headers=headers, json=data) |
| | response_json = response.json() |
| | |
| | |
| | image_url = response_json["data"][0]["url"] |
| | |
| | return image_url |
| |
|
| | |
| | with gr.Blocks() as demo: |
| | with gr.Row(): |
| | gr.Markdown("# <center> OpenAI dall-e-3 API with Gradio </center>") |
| | gr.Markdown("This demo uses the OpenAI dall-e-3 API to generate an image from text. You get a free session key from https://dongsiqie.me/sess") |
| | 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") |
| | quality_input = gr.Dropdown(label="Quality", choices=["standard", "hd"], value="standard") |
| | size_input = gr.Dropdown(label="Size", choices=["1792x1024", "1024x1024", "1024x1792"], value="1024x1024") |
| | style_input = gr.Dropdown(label="Style", choices=["vivid", "natural"], value="vivid") |
| | with gr.Row(): |
| | prompt_input = gr.Textbox(label="Image Description", value="A cute cat.") |
| | submit_btn = gr.Button("Generate Image", variant='primary') |
| | image_output = gr.Image(label="Generated Image") |
| | |
| | submit_btn.click(fn=generate_image, inputs=[prompt_input, quality_input, size_input, style_input, api_key_input, request_url_input], outputs=image_output) |
| |
|
| | demo.launch() |
| |
|
