chat_with_youtube_transcript

app.py

@@ -55,7 +55,7 @@ CURRENT_INDEX = 0
 VIDEO_ID = ""
 
 OPEN_AI_KEY = os.getenv("OPEN_AI_KEY")
-client = OpenAI(api_key=OPEN_AI_KEY)
+OPEN_AI_CLIENT = OpenAI(api_key=OPEN_AI_KEY)
 DRIVE_KEY = os.getenv("GOOGLE_APPLICATION_CREDENTIALS_JSON")
 GCS_KEY = os.getenv("GOOGLE_APPLICATION_CREDENTIALS_JSON")
 
@@ -659,7 +659,7 @@ def generate_mind_map(df_string):
         "max_tokens": 4000,
     }
 
-    response = client.chat.completions.create(**request_payload)
+    response = OPEN_AI_CLIENT.chat.completions.create(**request_payload)
     mind_map = response.choices[0].message.content.strip()
     print("=====mind_map=====")
     print(mind_map)
@@ -815,7 +815,7 @@ def generate_summarise(df_string):
         "max_tokens": 4000,
     }
 
-    response = client.chat.completions.create(**request_payload)
+    response = OPEN_AI_CLIENT.chat.completions.create(**request_payload)
     df_summarise = response.choices[0].message.content.strip()
     print("=====df_summarise=====")
     print(df_summarise)
@@ -847,7 +847,7 @@ def generate_questions(df_string):
         "response_format": response_format
     }
 
-    response = client.chat.completions.create(**request_payload)
+    response = OPEN_AI_CLIENT.chat.completions.create(**request_payload)
     questions = json.loads(response.choices[0].message.content)["questions"]
     print("=====json_response=====")
     print(questions)
@@ -990,7 +990,7 @@ def respond(user_message, data, chat_history, socratic_mode=False):
         "messages": messages,
         "max_tokens": 4000  # 設定一個較大的值，可根據需要調整
     }
-    response = client.chat.completions.create(**request_payload)
+    response = OPEN_AI_CLIENT.chat.completions.create(**request_payload)
     response_text = response.choices[0].message.content.strip()
 
     # 更新聊天历史
@@ -1003,6 +1003,63 @@ def respond(user_message, data, chat_history, socratic_mode=False):
     # 返回聊天历史和空字符串清空输入框
     return "", chat_history
 
+def chat_with_youtube_transcript(youtube_id, thread_id, user_message, chat_history, socratic_mode=False):
+    assistant_id = "asst_kmvZLNkDUYaNkMNtZEAYxyPq"
+    client = OPEN_AI_CLIENT
+    # 创建线程
+    if thread_id is None:
+        thread = client.beta.threads.create()
+        # 向线程添加初始系统消息，指定使用的 YouTube 视频ID
+        client.beta.threads.messages.create(
+            thread_id=thread.id,
+            role="system",
+            content=f"""
+                使用 YouTube 视频逐字稿数据：youtube_id = {youtube_id}，
+                如果沒有找到逐字稿，就告訴 user 你有沒有逐字稿的資料。
+            """
+        )
+    else:
+        thread = client.beta.threads.retrieve(thread_id)
+    
+    # 向线程添加用户的消息
+    client.beta.threads.messages.create(
+        thread_id=thread.id,
+        role="user",
+        content=user_message
+    )
+
+    # 运行助手，生成响应
+    run = client.beta.threads.runs.retrieve(
+        thread_id=thread.id,
+        assistant_id=assistant_id
+    )
+
+    # check run status
+    while run.status == "running":
+        time.sleep(3)
+        run = client.beta.threads.runs.retrieve(
+            thread_id=thread.id,
+            assistant_id=assistant_id
+        )
+
+    # 获取助手的响应消息
+    messages = client.beta.threads.messages.list(thread_id=thread.id)
+    assistant_responses = [msg for msg in messages.data if msg.role == "assistant"]
+    
+    # 返回最新的助手响应
+    response_text = assistant_responses[-1].content if assistant_responses else "没有响应。"
+
+    # 更新聊天历史
+    new_chat_history = (user_message, response_text)
+    if chat_history is None:
+        chat_history = [new_chat_history]
+    else:
+        chat_history.append(new_chat_history)
+
+    # 返回聊天历史和空字符串清空输入框
+    return "", chat_history, thread.id
+
+
 def update_slide(direction):
     global TRANSCRIPTS
     global CURRENT_INDEX
@@ -1070,6 +1127,7 @@ with gr.Blocks() as demo:
             youtube_link_btn = gr.Button("Submit_YouTube_Link")
             web_link = gr.Textbox(label="Enter Web Page Link", visible=False)
             chatbot = gr.Chatbot()
+            thread_id = gr.Textbox(label="thread_id")
             socratic_mode_btn = gr.Checkbox(label="蘇格拉底家教助理模式", value=False)
             msg = gr.Textbox(label="Message")
             send_button = gr.Button("Send")
@@ -1103,16 +1161,40 @@ with gr.Blocks() as demo:
                 mind_map = gr.Textbox(container=True, show_copy_button=True, lines=40, elem_id="mind_map_markdown")
             with gr.Tab("心智圖",elem_id="mind_map_tab"):
                 mind_map_html = gr.HTML()
-
+    
+    # 傳統模式
+    # send_button.click(
+    #     respond, 
+    #     inputs=[msg, df_string_output, chatbot, socratic_mode_btn], 
+    #     outputs=[msg, chatbot]
+    # )
+    # # 连接按钮点击事件
+    # btn_1.click(respond, inputs=[btn_1, df_string_output, chatbot, socratic_mode_btn], outputs=[msg, chatbot])
+    # btn_2.click(respond, inputs=[btn_2, df_string_output, chatbot, socratic_mode_btn], outputs=[msg, chatbot])
+    # btn_3.click(respond, inputs=[btn_3, df_string_output, chatbot, socratic_mode_btn], outputs=[msg, chatbot])
+
+    # chat_with_youtube_transcript
     send_button.click(
-        respond, 
-        inputs=[msg, df_string_output, chatbot, socratic_mode_btn], 
-        outputs=[msg, chatbot]
+        chat_with_youtube_transcript,
+        inputs=[get_video_id, thread_id, msg, chatbot, socratic_mode_btn],
+        outputs=[msg, chatbot, thread_id]
     )
     # 连接按钮点击事件
-    btn_1.click(respond, inputs=[btn_1, df_string_output, chatbot, socratic_mode_btn], outputs=[msg, chatbot])
-    btn_2.click(respond, inputs=[btn_2, df_string_output, chatbot, socratic_mode_btn], outputs=[msg, chatbot])
-    btn_3.click(respond, inputs=[btn_3, df_string_output, chatbot, socratic_mode_btn], outputs=[msg, chatbot])
+    btn_1.click(
+        chat_with_youtube_transcript,
+        inputs=[get_video_id, thread_id, btn_1, chatbot, socratic_mode_btn],
+        outputs=[msg, chatbot, thread_id]
+    )
+    btn_2.click(
+        chat_with_youtube_transcript,
+        inputs=[get_video_id, thread_id, btn_2, chatbot, socratic_mode_btn],
+        outputs=[msg, chatbot, thread_id]
+    )
+    btn_3.click(
+        chat_with_youtube_transcript,
+        inputs=[get_video_id, thread_id, btn_3, chatbot, socratic_mode_btn],
+        outputs=[msg, chatbot, thread_id]
+    )
 
     btn_create_question.click(change_questions, inputs = [df_string_output], outputs = [btn_1, btn_2, btn_3])