response_text = "抱歉，助手有點累，好像睡著了，請稍後再試。"

app.py

@@ -1044,27 +1044,14 @@ def chat_with_youtube_transcript(youtube_id, thread_id, user_message, chat_histo
     )
 
     # 等待助手响应，设定最大等待时间为 30 秒
-    max_wait_time = 30
-    start_time = time.time()
-    while time.time() - start_time < max_wait_time:
-        print("=== 等待助手响应 ===")
-        print(f"run.status: {run.status}")
-        print(f"time.time() - start_time: {time.time() - start_time}")
-        print("=== 等待助手响应 ===")
-        run_status = client.beta.threads.runs.retrieve(
-            thread_id=thread_id,
-            run_id=run.id
-        )
-        if run_status.status != "running":
-            break
-        time.sleep(3)  # 每3秒检查一次状态
-
+    run_status = poll_run_status(run.id, thread.id, timeout=30)
     # 获取助手的响应消息
-    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 "没有响应。"
+    if run_status == "completed":
+        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:
+        response_text = "抱歉，助手有點累，好像睡著了，請稍後再試。"
 
     # 更新聊天历史
     new_chat_history = (user_message, response_text)
@@ -1076,6 +1063,54 @@ def chat_with_youtube_transcript(youtube_id, thread_id, user_message, chat_histo
     # 返回聊天历史和空字符串清空输入框
     return "", chat_history, thread.id
 
+def poll_run_status(run_id, thread_id, timeout=600, poll_interval=5):
+    """
+    Polls the status of a Run and handles different statuses appropriately.
+
+    :param run_id: The ID of the Run to poll.
+    :param thread_id: The ID of the Thread associated with the Run.
+    :param timeout: Maximum time to wait for the Run to complete, in seconds.
+    :param poll_interval: Time to wait between each poll, in seconds.
+    """
+    client = OPEN_AI_CLIENT
+    start_time = time.time()
+    while time.time() - start_time < timeout:
+        run = client.beta.threads.runs.retrieve(thread_id=thread_id, run_id=run_id)
+
+        if run.status in ["completed", "cancelled", "failed"]:
+            print(f"Run completed with status: {run.status}")
+            break
+        elif run.status == "requires_action":
+            print("Run requires action. Performing required action...")
+            # Here, you would perform the required action, e.g., running functions
+            # and then submitting the outputs. This is simplified for this example.
+            # After performing the required action, you'd complete the action:
+            # OPEN_AI_CLIENT.beta.threads.runs.complete_required_action(...)
+        elif run.status == "expired":
+            print("Run expired. Exiting...")
+            break
+        else:
+            print(f"Run status is {run.status}. Waiting for updates...")
+        
+        time.sleep(poll_interval)
+    else:
+        print("Timeout reached. Run did not complete in the expected time.")
+
+    # Once the Run is completed, handle the result accordingly
+    if run.status == "completed":
+        # Retrieve and handle messages or run steps as needed
+        messages = client.beta.threads.messages.list(thread_id=thread_id)
+        for message in messages.data:
+            if message.role == "assistant":
+                print(f"Assistant response: {message.content}")
+    elif run.status in ["cancelled", "failed"]:
+        # Handle cancellation or failure
+        print(f"Run ended with status: {run.status}")
+    elif run.status == "expired":
+        # Handle expired run
+        print("Run expired without completion.")
+    
+    return run.status
 
 def update_slide(direction):
     global TRANSCRIPTS