OPEN_AI_KEY_BOT2

app.py

@@ -45,7 +45,7 @@ print(f"is_env_local: {is_env_local}")
 print("===gr__version__===")
 print(gr.__version__)
 
-
+# KEY CONFIG
 if is_env_local:
     with open("local_config.json") as f:
         config = json.load(f)
@@ -53,6 +53,11 @@ if is_env_local:
         GCS_KEY = json.dumps(config["GOOGLE_APPLICATION_CREDENTIALS_JSON"])
         DRIVE_KEY = json.dumps(config["GOOGLE_APPLICATION_CREDENTIALS_JSON"])
         OPEN_AI_KEY = config["OPEN_AI_KEY"]
+        OPEN_AI_ASSISTANT_ID_GPT4_BOT1 = config["OPEN_AI_ASSISTANT_ID_GPT4_BOT1"]
+        OPEN_AI_ASSISTANT_ID_GPT3_BOT1 = config["OPEN_AI_ASSISTANT_ID_GPT3_BOT1"]
+        OPEN_AI_KEY_BOT2 = config["OPEN_AI_KEY_BOT2"]
+        OPEN_AI_ASSISTANT_ID_GPT4_BOT2 = config["OPEN_AI_ASSISTANT_ID_GPT4_BOT2"]
+        OPEN_AI_ASSISTANT_ID_GPT3_BOT2 = config["OPEN_AI_ASSISTANT_ID_GPT3_BOT2"]
         GROQ_API_KEY = config["GROQ_API_KEY"]
         JUTOR_CHAT_KEY = config["JUTOR_CHAT_KEY"]
         AWS_ACCESS_KEY = config["AWS_ACCESS_KEY"]
@@ -64,6 +69,11 @@ else:
     GCS_KEY = os.getenv("GOOGLE_APPLICATION_CREDENTIALS_JSON")
     DRIVE_KEY = os.getenv("GOOGLE_APPLICATION_CREDENTIALS_JSON")
     OPEN_AI_KEY = os.getenv("OPEN_AI_KEY")
+    OPEN_AI_ASSISTANT_ID_GPT4_BOT1 = os.getenv("OPEN_AI_ASSISTANT_ID_GPT4_BOT1")
+    OPEN_AI_ASSISTANT_ID_GPT3_BOT1 = os.getenv("OPEN_AI_ASSISTANT_ID_GPT3_BOT1")
+    OPEN_AI_KEY_BOT2 = os.getenv("OPEN_AI_KEY_BOT2")
+    OPEN_AI_ASSISTANT_ID_GPT4_BOT2 = os.getenv("OPEN_AI_ASSISTANT_ID_GPT4_BOT2")
+    OPEN_AI_ASSISTANT_ID_GPT3_BOT2 = os.getenv("OPEN_AI_ASSISTANT_ID_GPT3_BOT2")
     GROQ_API_KEY = os.getenv("GROQ_API_KEY")
     JUTOR_CHAT_KEY = os.getenv("JUTOR_CHAT_KEY")
     AWS_ACCESS_KEY = os.getenv("AWS_ACCESS_KEY")
@@ -74,7 +84,7 @@ else:
 TRANSCRIPTS = []
 CURRENT_INDEX = 0
 
-OPEN_AI_CLIENT = OpenAI(api_key=OPEN_AI_KEY)
+# CLIENTS CONFIG
 GROQ_CLIENT = Groq(api_key=GROQ_API_KEY)
 GCS_SERVICE = GoogleCloudStorage(GCS_KEY)
 GCS_CLIENT = GCS_SERVICE.client
@@ -85,6 +95,39 @@ BEDROCK_CLIENT = boto3.client(
     region_name=AWS_REGION_NAME,
 )
 
+# check open ai access
+def check_open_ai_access(open_ai_api_key):
+    # set key in OpenAI client and run to check status, if it is work, return True
+    client = OpenAI(api_key=open_ai_api_key)
+    try:
+        response = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=[
+                {"role": "user", "content": "This is a test."},
+            ],
+        )
+        if response.choices[0].message.content:
+            return True
+        else:
+            return False
+    except Exception as e:
+        print(f"Error: {str(e)}")
+        return False
+
+open_ai_api_key_assistant_id_list = [
+    {"account":"bot1", "open_ai_api_key": OPEN_AI_KEY, "assistant_gpt4_id": OPEN_AI_ASSISTANT_ID_GPT4_BOT1, "assistant_gpt3_id": OPEN_AI_ASSISTANT_ID_GPT3_BOT1},
+    {"account":"bot2", "open_ai_api_key": OPEN_AI_KEY_BOT2, "assistant_gpt4_id": OPEN_AI_ASSISTANT_ID_GPT4_BOT2, "assistant_gpt3_id": OPEN_AI_ASSISTANT_ID_GPT3_BOT2},
+]
+for open_ai_api_key_assistant_id in open_ai_api_key_assistant_id_list:
+    account = open_ai_api_key_assistant_id["account"]
+    open_ai_api_key = open_ai_api_key_assistant_id["open_ai_api_key"]
+    if check_open_ai_access(open_ai_api_key):
+        OPEN_AI_CLIENT = OpenAI(api_key=open_ai_api_key)
+        OPEN_AI_ASSISTANT_ID_GPT4 = open_ai_api_key_assistant_id["assistant_gpt4_id"]
+        OPEN_AI_ASSISTANT_ID_GPT3 = open_ai_api_key_assistant_id["assistant_gpt3_id"]
+        print(f"OpenAI access is OK, account: {account}")
+        break
+
 # 驗證 password
 def verify_password(password):
     if password == PASSWORD:
@@ -2200,8 +2243,8 @@ def chat_with_opan_ai_assistant(password, youtube_id, user_data, thread_id, tras
         raise gr.Error(error_msg)
 
     try:
-        assistant_id = "asst_Mk151eZmKhNxzG7L9Awqz6iZ" #GPT 4 turbo
-        # assistant_id = "asst_sCA7F5opi2g7AvGnYeRfoSfT" #GPT 3.5 turbo
+        assistant_id = OPEN_AI_ASSISTANT_ID_GPT4 #GPT 4 turbo
+        # assistant_id = OPEN_AI_ASSISTANT_ID_GPT3 #GPT 3.5 turbo
 
         client = OPEN_AI_CLIENT
         # 直接安排逐字稿資料 in instructions
@@ -2380,8 +2423,9 @@ def chat_with_opan_ai_assistant_streaming(user_message, chat_history, password,
         raise gr.Error(error_msg)
 
     try:
-        assistant_id = "asst_Mk151eZmKhNxzG7L9Awqz6iZ" #GPT 4 turbo
-        # assistant_id = "asst_sCA7F5opi2g7AvGnYeRfoSfT" #GPT 3.5 turbo
+        assistant_id = OPEN_AI_ASSISTANT_ID_GPT4 #GPT 4 turbo
+        # assistant_id = OPEN_AI_ASSISTANT_ID_GPT3 #GPT 3.5 turbo
+
         client = OPEN_AI_CLIENT
         # 直接安排逐字稿資料 in instructions
         # if isinstance(trascript, str):
@@ -2637,7 +2681,7 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
             with gr.Row():
                 user_avatar = "https://em-content.zobj.net/source/google/263/flushed-face_1f633.png"
                 ai_chatbot_bot_avatar = "https://junyitopicimg.s3.amazonaws.com/s4byy--icon.jpe?v=20200513013523726"
-                with gr.Column(scale=1, variant="panel", visible=False):
+                with gr.Column(scale=1, variant="panel", visible=True):
                     chatbot_avatar_url = "https://junyitopicimg.s3.amazonaws.com/s4byy--icon.jpe?v=20200513013523726"
                     chatbot_description = """Hi，我是你的AI學伴【飛特精靈】，\n
                     我可以陪你一起學習本次的內容，有什麼問題都可以問我喔！\n
@@ -2651,7 +2695,7 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
                     gr.Image(value=chatbot_avatar_url, height=100, width=100, show_label=False, show_download_button=False)
                     chatbot_open_ai_select_btn = gr.Button("👆選擇【飛特精靈】", elem_id="chatbot_btn", visible=True, variant="primary")
                     gr.Markdown(value=chatbot_description, visible=True)
-                with gr.Column(scale=1, variant="panel", visible=False):
+                with gr.Column(scale=1, variant="panel", visible=True):
                     streaming_chatbot_avatar_url = "https://storage.googleapis.com/wpassets.junyiacademy.org/1/2020/11/1-%E6%98%9F%E7%A9%BA%E9%A0%AD%E8%B2%BC-%E5%A4%AA%E7%A9%BA%E7%8B%90%E7%8B%B8%E8%B2%93-150x150.png"
                     streaming_chatbot_description = """Hi，我是【飛特音速】， \n
                     說話比較快，但有什麼問題都可以問我喔！ \n
@@ -2707,7 +2751,9 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
                     gr.Image(value=maimai_chatbot_avatar_url, height=100, width=100, show_label=False, show_download_button=False)
                     maimai_chatbot_select_btn = gr.Button("👆選擇【麥麥】", visible=True, variant="primary", elem_classes="chatbot_select_btn")
                     maimai_chatbot_description_value = gr.Markdown(value=maimai_chatbot_description, visible=True)
-                    
+                with gr.Column(scale=1, variant="panel"): 
+                    gr.Markdown(value="### 尚未開放", visible=True)
+
         with gr.Row("飛特精靈") as chatbot_open_ai:
             with gr.Column():
                 bot_avatar = "https://junyitopicimg.s3.amazonaws.com/s4byy--icon.jpe?v=20200513013523726"