| 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() |
|
|
