Update app.py

app.py

@@ -7,7 +7,9 @@ openai.api_key = os.getenv("API_KEY")
 app_password = os.getenv("APP_PASSWORD")
 app_username = os.getenv("APP_USERNAME")
 
-def generate(prompt):
+def generate(prompt, api_key, api_organization=None):
+    openai.organization = api_organization
+    openai.api_key = api_key
     response = openai.Image.create(
       prompt=prompt,
       n=1,
@@ -15,17 +17,19 @@ def generate(prompt):
     )
     return response['data'][0]['url']
 
-examples = [
-    ["きのこの山"],
-    ["たけのこの里"],
-]
-
-demo = gr.Interface(
-    fn=generate,
-    inputs=gr.components.Textbox(lines=5, label="Prompt"),
-    outputs=gr.components.Image(type="filepath", label="Generated Image"),
-    flagging_options=[],
-    examples=examples
-)
+with gr.Blocks() as demo:
+    gr.Markdown("Demo-app for image generation using DALL-E")
+    with gr.Accordion("OpenAI API Settings", open=False):
+        api_key = gr.Textbox(label="OpenAI API key", placeholder="OpenAI API key")
+        api_organization = gr.Textbox(label="OpenAI API organization", placeholder="OpenAI API organization (optional)")
+    with gr.Row():
+        inp = gr.Textbox(line=5, label="Input", placeholder="Image prompt")
+        out = gr.Image(type="filepath", label="Output")
+    examples = gr.Examples(
+        [["きのこの山"], ["たけのこの里"]],
+        [inp],
+    )
+    btn = gr.Button("Generate Image")
+    btn.click(fn=generate, inputs=[inp, api_key, api_organization], outputs=out)
 
-demo.launch(share=False, auth=(app_username, app_password))
+demo.launch()