Update app.py

app.py

@@ -1,3 +1,24 @@
+# -*- coding: utf-8 -*-
+"""gradio_chat_image.ipynb
+
+Automatically generated by Colaboratory.
+
+Original file is located at
+    https://colab.research.google.com/drive/1UOcwXwg1bHrPbhkM9tB5ivx6OnI9lB7q
+"""
+
+!pip install gradio
+!pip install openai
+
+import os
+
+os.environ["TEST_OPENAI_KEY"] = "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"
+os.environ["ASSIST_ID"] = "asst_ePkuRLY0DUmChyCv8tXtf0gC"
+os.environ["MAX_TRIAL"] = "50"
+os.environ["INTER_SEC"] = "1"   # 画像は1
+os.environ["MAX_TOKENS"] = "300"
+
+from logging import DEBUG
 import os
 import json
 import base64
@@ -25,6 +46,7 @@ file_format = {".png", ".jpeg", ".jpg", ".webp", ".gif", ".PNG", ".JPEG", ".JPG"
 # 各種メッセージ
 PLACEHOLDER = "DaLL-E3を利用の場合「『○○』で作ってください。」\nVisionを利用の場合「この画像について説明して下さい。」など入力して下さい。"
 # IMG_MSG = "(画像ファイルを追加しました。リセットボタンの上に表示されています。)"
+DEL_MSG = "こちらをクリックして表示"
 ANT_MSG = "（下部の[出力ファイル]にファイルを追加しました。）"
 
 # 各種設定値
@@ -33,7 +55,7 @@ INTER_SEC = int(os.environ["INTER_SEC"])   # 試行間隔（秒）
 MAX_TOKENS = int(os.environ["MAX_TOKENS"])  # Vison最大トークン
 
 # 正規表現用パターン
-pt = r".*!\[(.*)\]\((.*)\)"
+pt = r".*\[(.*)\]\((.*)\)"
 
 # サンプル用情報
 examples = ["1980s anime girl with straight bob-cut in school uniform, roughly drawn drawing"
@@ -353,6 +375,9 @@ def bot(state, history, image_path):
                                 # パターン一致の場合はプロンプトだけ抜き出す
                                 res_text = result.group(1)
 
+                                # 「こちらをクリックして表示」は削除
+                                res_text = res_text.replace(DEL_MSG, "")
+
                             # Chat画面更新
                             if history[-1][1] is not None:
 
@@ -579,10 +604,14 @@ with gr.Blocks() as demo:
     with gr.Tab("注意事項") as notes:
             caution = "・[画像一括ダウンロード]を押すと、下部[出力ファイル]にZIPファイルができます。<br>"
             caution += "・画像を生成した際、チャット画面に「こちらをクリック」となる場合がありますが画像は[出力画像]に表示されます。<br>"
+            caution += "・テスト中はDaLL-E3用設定は固定となっております。"
             gr.Markdown("<h3>" + caution + "</h3>")
 
 
+if __name__ == '__main__':
 
+    demo.queue()
+    demo.launch(debug=True)
 
 def request_DallE3(client, prompt, size, quality, out_image_path):
 
@@ -689,7 +718,28 @@ def request_Vision(client, prompt, image_path, detail, max_tokens):
         }
         return json.dumps(vision_result)
 
-if __name__ == '__main__':
+def request_DallE3(client, prompt, size, quality, out_image_path):
+    """ DallE3を呼び出す """
 
-    demo.queue()
-    demo.launch(debug=True)
+    err_msg = ""
+
+    imgage_path = "/content/sample_datacat1.png"
+
+    dalle3_result = {
+        "imgage_path" : imgage_path,
+        "error_message" : err_msg
+    }
+
+    return json.dumps(dalle3_result)
+
+
+def request_Vision(prompt, detail, image_path):
+    """ GPT4 Visionを呼び出す """
+
+    response_text = "この画像は驚いた表情をしている人物を写した写真です。"
+
+    vision_result = {
+        "answer" : response_text
+    }
+
+    return json.dumps(vision_result)