Update app.py

app.py

@@ -12,6 +12,8 @@ app_password = os.getenv("APP_PASSWORD")
 app_username = os.getenv("APP_USERNAME")
 
 
+NUM_OUTPUTS=4
+
 PROMPT_1 = """\
 論文のアブストラクトに対し、そこから抽象的なアイコンやイメージを生成してスライドやポスターに使いたいです。生成には DALL・Eを使います。アブストラクトと、DALL・E の画像プロンプトを例として与えるので、最後に入力されるアブストラクトに対する画像プロンプトを考えて、1つ出力してください。プロンプト以外の文字は含まないでください
 
@@ -36,14 +38,13 @@ def generate(text, select_radio):
             temperature=0.0,
         )
         response = openai.Image.create(
-            prompt="Draw an easy-to-read system configuration diagram under the following conditions Use only two colors, black and white. Use arrows and squares. This diagram is a pictogram. Write no alphabets. The following is the system configuration."
-            + str(response["choices"][0]["message"]["content"]),
-            n=1,
+            str(response["choices"][0]["message"]["content"]),
+            n=NUM_OUTPUTS,
             size="256x256",
         )
     # icon
     # 後でプロンプトを変更する
-    else:
+    elif str(select_radio) == "icon":
         system_prompt = PROMPT_1
         response = openai.ChatCompletion.create(
             model="gpt-3.5-turbo-0613",
@@ -56,27 +57,25 @@ def generate(text, select_radio):
         )
         response = openai.Image.create(
             prompt=str(response["choices"][0]["message"]["content"]),
-            n=1,
+            n=NUM_OUTPUTS,
             size="256x256",
         )
-    return response["data"][0]["url"]
-
+    else:
+        raise gr.Error("Choose output type!")
+    return [response["data"][i]["url"] for i in range(NUM_OUTPUTS)]
 
-examples = [
-    ["きのこの山"],
-    ["たけのこの里"],
-]
 
 demo = gr.Interface(
     fn=generate,
     inputs=[
-        gr.components.Textbox(lines=5, label="Prompt"),
-        gr.Radio(["icon", "pictogram"], label="radio"),
+        gr.components.Textbox(lines=5, label="Abstract of your paper"),
+        gr.Radio(["icon", "pictogram"], label="Output type", value="icon"),
     ],
-    outputs=gr.components.Image(type="filepath", label="Generated Image"),
+    outputs=[gr.components.Image(type="filepath", label="Generated Image") for _ in range(NUM_OUTPUTS)],
     flagging_options=[],
-    # examples=examples,
+    title="IdeaIllustrate"
 )
 
 demo.launch(share=False, auth=(app_username, app_password))
 # https://www.gradio.app/docs/radio ラジオボタンの入力要素をリストで取得する
+