CHAT_LIMIT = 10

app.py

@@ -83,6 +83,7 @@ else:
 
 TRANSCRIPTS = []
 CURRENT_INDEX = 0
+CHAT_LIMIT = 10
 
 # CLIENTS CONFIG
 GROQ_CLIENT = Groq(api_key=GROQ_API_KEY)
@@ -2085,7 +2086,7 @@ def get_instructions(content_subject, content_grade, key_moments):
     """
     return instructions
 
-def chat_with_ai(ai_name, password, video_id, user_data, trascript_state, key_moments, user_message, chat_history, content_subject, content_grade, questions_answers_json, socratic_mode=False):
+def chat_with_ai(ai_name, password, video_id, user_data, trascript_state, key_moments, user_message, chat_history, content_subject, content_grade, questions_answers_json, send_btn, socratic_mode=False):
     verify_password(password)
 
     print("=====user_data=====")
@@ -2108,10 +2109,17 @@ def chat_with_ai(ai_name, password, video_id, user_data, trascript_state, key_mo
 
             # 等待 3 秒
             time.sleep(3)
+            send_count = len(chat_history)-1
+            if len(chat_history) > CHAT_LIMIT:
+                send_btn_value = f"對話已達上限 ({send_count}/{CHAT_LIMIT})"
+                send_btn = gr.update(value=send_btn_value, interactive=False)
+            else:
+                send_btn_value = f"發送 ({send_count}/{CHAT_LIMIT})"
+                send_btn = gr.update(value=send_btn_value, interactive=True)
 
-            return "", chat_history
+            return "", chat_history, send_btn
 
-    if chat_history is not None and len(chat_history) > 11:
+    if chat_history is not None and len(chat_history) > CHAT_LIMIT:
         error_msg = "此次對話超過上限（對話一輪10次）"
         raise gr.Error(error_msg)
 
@@ -2197,14 +2205,22 @@ def chat_with_ai(ai_name, password, video_id, user_data, trascript_state, key_mo
         else:
             chat_history.append(new_chat_history)
 
+        send_count = len(chat_history)-1
+        if len(chat_history) > CHAT_LIMIT:
+            send_btn_value = f"對話已達上限 ({send_count}/{CHAT_LIMIT})"
+            send_btn = gr.update(value=send_btn_value, interactive=False)
+        else:
+            send_btn_value = f"發送 ({send_count}/{CHAT_LIMIT})"
+            send_btn = gr.update(value=send_btn_value, interactive=True)
+
         # 返回聊天历史和空字符串清空输入框
-        return "", chat_history
+        return "", chat_history, send_btn
     except Exception as e:
         # 处理错误情况
         print(f"Error: {e}")
         return "请求失败，请稍后再试！", chat_history
 
-def chat_with_opan_ai_assistant(password, youtube_id, user_data, thread_id, trascript_state, key_moments, user_message, chat_history, content_subject, content_grade, questions_answers_json, socratic_mode=False):
+def chat_with_opan_ai_assistant(password, youtube_id, user_data, thread_id, trascript_state, key_moments, user_message, chat_history, content_subject, content_grade, questions_answers_json, send_btn, socratic_mode=False):
     verify_password(password)
 
     print("=====user_data=====")
@@ -2233,12 +2249,19 @@ def chat_with_opan_ai_assistant(password, youtube_id, user_data, thread_id, tras
 
             # 等待 3 秒
             time.sleep(3)
+            send_count = len(chat_history)-1
+            if len(chat_history) > CHAT_LIMIT:
+                send_btn_value = f"對話已達上限 ({send_count}/{CHAT_LIMIT})"
+                send_btn = gr.update(value=send_btn_value, interactive=False)
+            else:
+                send_btn_value = f"發送 ({send_count}/{CHAT_LIMIT})"
+                send_btn = gr.update(value=send_btn_value, interactive=True)
 
-            return "", chat_history, thread_id
+            return "", chat_history, thread_id, send_btn
 
     
     # 如果 chat_history 超過 10 則訊息，直接 return "對話超過上限"
-    if chat_history is not None and len(chat_history) > 10:
+    if chat_history is not None and len(chat_history) > CHAT_LIMIT:
         error_msg = "此次對話超過上限（對話一輪10次）"
         raise gr.Error(error_msg)
 
@@ -2329,8 +2352,17 @@ def chat_with_opan_ai_assistant(password, youtube_id, user_data, thread_id, tras
         print(f"Error: {e}")
         raise gr.Error(f"Error: {e}")
 
+    send_count = len(chat_history)-1
+    if len(chat_history) > CHAT_LIMIT:
+        send_btn_value = f"對話已達上限 ({send_count}/{CHAT_LIMIT})"
+        send_btn = gr.update(value=send_btn_value, interactive=False)
+    else:
+        send_btn_value = f"發送 ({send_count}/{CHAT_LIMIT})"
+        send_btn = gr.update(value=send_btn_value, interactive=True)
+
+
     # 返回聊天历史和空字符串清空输入框
-    return "", chat_history, thread.id
+    return "", chat_history, thread.id, send_btn
 
 def process_open_ai_audio_to_chatbot(password, audio_url):
     verify_password(password)
@@ -2418,8 +2450,8 @@ def chat_with_opan_ai_assistant_streaming(user_message, chat_history, password,
         raise gr.Error(error_msg)
     
     # 如果 chat_history 超過 10 則訊息，直接 return "對話超過上限"
-    if chat_history is not None and len(chat_history) > 11:
-        error_msg = "此次對話超過上限（對話一輪10次）"
+    if chat_history is not None and len(chat_history) > CHAT_LIMIT:
+        error_msg = f"此次對話超過上限（對話一輪{CHAT_LIMIT}次）"
         raise gr.Error(error_msg)
 
     try:
@@ -3076,8 +3108,8 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
     # OPENAI ASSISTANT CHATBOT 模式
     send_button.click(
         chat_with_opan_ai_assistant,
-        inputs=[password, video_id, user_data, thread_id, trascript_state, key_moments, msg, chatbot, content_subject, content_grade, questions_answers_json, socratic_mode_btn],
-        outputs=[msg, chatbot, thread_id],
+        inputs=[password, video_id, user_data, thread_id, trascript_state, key_moments, msg, chatbot, content_subject, content_grade, questions_answers_json, send_button, socratic_mode_btn],
+        outputs=[msg, chatbot, thread_id, send_button],
         scroll_to_output=True
     )
     openai_chatbot_audio_input.change(
@@ -3095,8 +3127,8 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
     )
     question_buttons = [btn_1, btn_2, btn_3]
     for question_btn in question_buttons:
-        inputs_list = [password, video_id, user_data, thread_id, trascript_state, key_moments, question_btn, chatbot, content_subject, content_grade, questions_answers_json, ai_chatbot_socratic_mode_btn]
-        outputs_list = [msg, chatbot, thread_id]
+        inputs_list = [password, video_id, user_data, thread_id, trascript_state, key_moments, question_btn, chatbot, content_subject, content_grade, questions_answers_json, send_button, socratic_mode_btn]
+        outputs_list = [msg, chatbot, thread_id, send_button]
         setup_question_button_click(question_btn, inputs_list, outputs_list, chat_with_opan_ai_assistant)
 
     # 為生成問題按鈕設定特殊的點擊事件
@@ -3109,15 +3141,15 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
     # 其他精靈 ai_chatbot 模式
     ai_send_button.click(
         chat_with_ai,
-        inputs=[ai_name, password, video_id, user_data, trascript_state, key_moments, ai_msg, ai_chatbot, content_subject, content_grade, questions_answers_json, ai_chatbot_socratic_mode_btn],
-        outputs=[ai_msg, ai_chatbot],
+        inputs=[ai_name, password, video_id, user_data, trascript_state, key_moments, ai_msg, ai_chatbot, content_subject, content_grade, questions_answers_json, ai_send_button, ai_chatbot_socratic_mode_btn],
+        outputs=[ai_msg, ai_chatbot, ai_send_button],
         scroll_to_output=True
     )
     # 其他精靈 ai_chatbot 连接按钮点击事件
     ai_chatbot_buttons = [ai_chatbot_question_1, ai_chatbot_question_2, ai_chatbot_question_3]
     for ai_question_btn in ai_chatbot_buttons:
-        inputs_list = [ai_name, password, video_id, user_data, trascript_state, key_moments, ai_question_btn, ai_chatbot, content_subject, content_grade, questions_answers_json, ai_chatbot_socratic_mode_btn]
-        outputs_list = [ai_msg, ai_chatbot]
+        inputs_list = [ai_name, password, video_id, user_data, trascript_state, key_moments, ai_question_btn, ai_chatbot, content_subject, content_grade, questions_answers_json, ai_send_button, ai_chatbot_socratic_mode_btn]
+        outputs_list = [ai_msg, ai_chatbot, ai_send_button]
         setup_question_button_click(ai_question_btn, inputs_list, outputs_list, chat_with_ai)
 
     # file_upload.change(process_file, inputs=file_upload, outputs=df_string_output)