Update app.py

app.py

@@ -1,8 +1,8 @@
+import os
 import gradio as gr
 import time
 import datetime
 from zoneinfo import ZoneInfo
-import os
 from openai import (
     OpenAI, AuthenticationError, NotFoundError, BadRequestError
 )
@@ -11,6 +11,13 @@ from openai import (
 MAX_TRIAL = int(os.environ["MAX_TRIAL"])  # メッセージ取得最大試行数
 INTER_SEC = int(os.environ["INTER_SEC"])   # 試行間隔（秒）
 
+# アシスタント用設定
+DF_MODEL = "gpt-3.5-turbo-1106"
+ASSIST_NAME_JA = "Japanese Assistant"
+ASSIST_NAME_ET = "English Teacher Assistant"
+AST_SYS_PROMPT_JA = "あなたは日本人の優秀なアシスタントです。質問は日本語で回答して下さい。"
+AST_SYS_PROMPT_EN = "You are an English teacher. Please be sure to answer in English."
+
 # sys_prompt_default = "あなたは優秀なアシスタントです。日本語で質問に回答してください。"
 # lang_code = {'Japanese': "ja", 'English': "en"}
 auto_play_bl = {'ON': True, 'OFF': False}
@@ -43,11 +50,15 @@ def add_history(history, text_msg):
 def init(state, mode, text_msg, voice_msg):
     """ 初期処理（入力チェック・テキスト変換） """
 
-    print(state)
-
     err_msg = ""
     text = ""
 
+    if state["openai_key"] == "":
+
+        # OpenAI API Key未入力
+        err_msg = "OpenAI API Keyを入力してください。（設定タブ）"
+
+        return state, "", err_msg
 
     if text_msg == "" and voice_msg is None:
 
@@ -61,8 +72,8 @@ def init(state, mode, text_msg, voice_msg):
         if state["client"] is None:
 
             # 初回起動時は初期処理をする
-            os.environ["OPENAI_API_KEY"] = os.environ["TEST_OPENAI_KEY"]   # テスト時
-            # os.environ["OPENAI_API_KEY"] = state["openai_key"]
+            # os.environ["OPENAI_API_KEY"] = os.environ["TEST_OPENAI_KEY"]   # テスト時
+            os.environ["OPENAI_API_KEY"] = state["openai_key"]
 
             # クライアント新規作成
             client = OpenAI()
@@ -94,31 +105,40 @@ def init(state, mode, text_msg, voice_msg):
 
             if state["assistant_id"] == "":
 
-                # アシスタント作成
-                # assistant = client.beta.assistants.create(
-                #   name="codeinter_test",
-                #   instructions=state["system_prompt"],
-                #   # model="gpt-4-1106-preview",
-                #   model="gpt-3.5-turbo-1106",
-                #   tools=[{"type": "code_interpreter"}]
-                # )
-                # state["assistant_id"] = assistant.id
-
                 if mode == 0:
 
                     # 日本語アシスタントをセット
-                    state["assistant_id"] = os.environ["ASSIST_JA"]
+                    assistant_id = get_assist_id(client, ASSIST_NAME_JA)
+                    # state["assistant_id"] = os.environ["ASSIST_JA"]
+
+                    if assistant_id == "":
+
+                          # アシスタント新規作成
+                          assistant = client.beta.assistants.create(
+                            name=ASSIST_NAME_JA,
+                            instructions=AST_SYS_PROMPT_JA,
+                            model=DF_MODEL,
+                          )
+
+                          assistant_id = assistant.id
 
                 elif mode == 1:
 
                     # 英語教師アシスタントをセット
-                    state["assistant_id"] = os.environ["ASSIST_EN"]
+                    assistant_id = get_assist_id(client, ASSIST_NAME_ET)
 
-            else:
+                    if assistant_id == "":
 
-                # アシスタント確認（IDが存在しないならエラーとなる）
-                # assistant = client.beta.assistants.retrieve(state["assistant_id"])
-                pass
+                          # アシスタント新規作成
+                          assistant = client.beta.assistants.create(
+                            name=ASSIST_NAME_ET,
+                            instructions=AST_SYS_PROMPT_EN,
+                            model=DF_MODEL,
+                          )
+
+                          assistant_id = assistant.id
+
+                state["assistant_id"] = assistant_id
 
         # ユーザIDでフォルダ作成
         os.makedirs(state["user_id"], exist_ok=True)
@@ -146,6 +166,25 @@ def init(state, mode, text_msg, voice_msg):
     finally:
         return state, text, err_msg
 
+def get_assist_id(client, assist_name):
+
+    assist_id = ""
+
+    assist_list = client.beta.assistants.list()
+
+    if len(assist_list.data) == 0:
+
+        return assist_id
+
+    for assist in assist_list:
+
+        if assist.name == assist_name:
+
+            assist_id = assist.id
+
+            break
+
+    return assist_id
 
 def raise_exception(err_msg):
     """ エラーの場合例外を起こす関数 """
@@ -362,12 +401,12 @@ def reset_chat(state):
 
 with gr.Blocks() as demo:
 
-    title = "<h2>GPT音声対応チャット</h2>"
+    title = "<h2>GPT 音声対応チャット</h2>"
     message = "<h3>・WhisperとText to speechによ���チャットを行います。<br>"
     message += "・テキストまたは音声を入力して下さい。モードの違いは「利用上の注意」タブをご覧ください。<br>"
+    message += "・自動再生の音声は送信ボタンの下の”出力音声”から自動再生で流れています。（設定でOFFにできます）<br>"
     message += "・動画での紹介はこちら→https://www.youtube.com/watch?v=wMoAORg0Y5Q<br>"
-    message += "※自動再生の音声は送信ボタンの下の”出力音声”から自動再生で流れています。（設定でOFFにできます）<br>"
-    # message += "・テスト中でAPIKEY無しで動きます。フィードバックもお待ちしております。<br>"
+    message += "※OpenAIのAPIキーが必要となります。設定タブから入力して下さい。<br>"
     message += "</h3>"
 
     gr.Markdown(title + message)
@@ -420,7 +459,7 @@ with gr.Blocks() as demo:
       #     finally_proc, [state, chatbot, sys_msg], [text_msg, voice_msg, out_voice], queue=False)
 
     with gr.Tab("設定"):
-        openai_key = gr.Textbox(label="OpenAI API Key" ,visible=False)
+        openai_key = gr.Textbox(label="OpenAI API Key", visible=True)
         voice = gr.Dropdown(choices=voice_list, value = "nova", label="Voice", interactive = True)
         auto_play = gr.Dropdown(choices=["ON", "OFF"], value = "ON", label="Auto Play", interactive = True)
         speed = gr.Slider(0, 1, value=0.8, label="Speed", info="1に近づけるほど読むスピードが速くなります。", interactive = True)