Update app.py

app.py

@@ -1,3 +1,23 @@
+# -*- coding: utf-8 -*-
+"""【3種用】dall-e-3.ipynb
+
+Automatically generated by Colaboratory.
+
+Original file is located at
+    https://colab.research.google.com/drive/1_t-vIP8Q49XVOKPgUYrBg2_b6EymJt_p
+"""
+
+!pip install openai
+!pip install gradio
+
+# from openai import OpenAI
+import os
+from google.colab import userdata
+
+os.environ["TEST_API_KEY"] = userdata.get('OPENAI_KEY')
+
+# client = OpenAI()
+
 import os
 from openai import OpenAI, BadRequestError
 import gradio as gr
@@ -83,14 +103,14 @@ def create_image(state, text):
 
         err_msg = "OpenAIキーを入力してください。（設定タブ）"
 
-        return text, None, "", err_msg
+        return None, "", err_msg
 
     # 入力チェック
     if text.strip() == "":
 
         err_msg = "プロンプトを入力してください。"
 
-        return text, None, "", err_msg
+        return None, "", err_msg
 
     if user_id == "":
 
@@ -129,12 +149,12 @@ def create_image(state, text):
 
     if return_msg == "":
 
-        return "", image_path, prompt, ""
+        return image_path, prompt, ""
 
     else:
         err_msg = "画像の作成に失敗しました。\n" + return_msg
 
-        return text, None, "", err_msg
+        return None, "", err_msg
 
 def make_archive(state):
     """ 画像のZIP化・一括ダウンロード用関数 """
@@ -155,7 +175,11 @@ def make_archive(state):
 
 with gr.Blocks() as demo:
 
-    gr.Markdown("設定タブでOpenAIキーを入力してください。")
+    title = "<h2>DALL-E3デモアプリ</h2>"
+    message = "<h3>最初に[設定]タブからOpenAIキーを入力してください。"
+    message += "</h3>"
+
+    gr.Markdown(title + message)
 
     # セッションの宣言
     state = gr.State({
@@ -167,10 +191,10 @@ with gr.Blocks() as demo:
         "style" : "",
     })
 
-    with gr.Tab("whisperを利用する") as maintab:
+    with gr.Tab("DALLE3を利用する") as maintab:
 
       # 出力画像
-      out_image = gr.Image(label="生成画像", type="filepath", interactive = False)
+      out_image = gr.Image(label="生成画像", type="filepath", interactive=False)
 
       # 各コンポーネント定義
       text = gr.Textbox(label="プロンプト", lines=4, placeholder=PLACEHOLDER)
@@ -185,7 +209,7 @@ with gr.Blocks() as demo:
       sys_msg = gr.Text(label="システムメッセージ")
       out_file = gr.File(label="出力ファイル", type="filepath",interactive = False)
 
-      btn.click(create_image, [state, text], [text, out_image, out_prompt, sys_msg])
+      btn.click(create_image, [state, text], [out_image, out_prompt, sys_msg])
       btn_dl.click(make_archive, state, [out_file, sys_msg])
 
     with gr.Tab("設定") as settab:
@@ -201,11 +225,11 @@ with gr.Blocks() as demo:
 
         caution = "・1枚当たりの料金はサイズ：1024x1024で0.04ドル、それ以外のサイズは0.080ドルです。<br>"
         caution += "・こちらはクオリティがStandardの場合で、hdの場合は1.5～2倍とかかります。"
-        caution += "詳細→https://openai.com/pricing"
+        caution += "料金の詳細→https://openai.com/pricing"
         gr.Markdown("<h3>" + caution + "</h3>")
 
 
 if __name__ == '__main__':
 
     demo.queue()
-    demo.launch(debug=True)
+    demo.launch(debug=True)