key_moments_state

app.py

@@ -2012,7 +2012,7 @@ def init_params(text, request: gr.Request):
 
     return admin, reading_passage_admin, summary_admin, see_detail, password_text, youtube_link
 
-def update_state(content_subject, content_grade, trascript, question_1, question_2, question_3):
+def update_state(content_subject, content_grade, trascript, key_moments, question_1, question_2, question_3):
     # inputs=[content_subject, content_grade, df_string_output],
     # outputs=[content_subject_state, content_grade_state, trascript_state]
     content_subject_state = content_subject
@@ -2020,12 +2020,14 @@ def update_state(content_subject, content_grade, trascript, question_1, question
     trascript_json = json.loads(trascript)
     formatted_simple_transcript = create_formatted_simple_transcript(trascript_json)
     trascript_state = formatted_simple_transcript
+    key_moments_state = key_moments
     streaming_chat_thread_id_state = create_thread_id()
     ai_chatbot_question_1 = question_1
     ai_chatbot_question_2 = question_2
     ai_chatbot_question_3 = question_3
 
-    return content_subject_state, content_grade_state, trascript_state, streaming_chat_thread_id_state, \
+    return content_subject_state, content_grade_state, trascript_state, key_moments_state, \
+        streaming_chat_thread_id_state, \
         ai_chatbot_question_1, ai_chatbot_question_2, ai_chatbot_question_3
 
 
@@ -2106,6 +2108,7 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
         content_subject_state = gr.State()  # 使用 gr.State 存储 content_subject
         content_grade_state = gr.State()  # 使用 gr.State 存储 content_grade
         trascript_state = gr.State()  # 使用 gr.State 存储 trascript
+        key_moments_state = gr.State()  # 使用 gr.State 存储 key_moments
         streaming_chat_thread_id_state = gr.State()  # 使用 gr.State 存储 streaming_chat_thread_id
     with gr.Tab("AI小精靈"):
         with gr.Row():
@@ -2128,7 +2131,7 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
                     msg = gr.Textbox(label="訊息",scale=3)
                     send_button = gr.Button("送出", variant="primary", scale=1)
             with gr.Tab("飛特音速"):
-                additional_inputs = [password, streaming_chat_thread_id_state, trascript_state, content_subject_state, content_grade_state]
+                additional_inputs = [password, streaming_chat_thread_id_state, trascript_state, key_moments_state, content_subject_state, content_grade_state]
                 streaming_chat = gr.ChatInterface(
                     fn=streaming_chat_with_open_ai,
                     additional_inputs=additional_inputs,
@@ -2398,6 +2401,7 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
         content_subject,
         content_grade,
         df_string_output,
+        key_moments,
         btn_1, 
         btn_2, 
         btn_3
@@ -2405,7 +2409,8 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
     update_state_outputs = [
         content_subject_state, 
         content_grade_state, 
-        trascript_state, 
+        trascript_state,
+        key_moments_state,
         streaming_chat_thread_id_state,
         ai_chatbot_question_1,
         ai_chatbot_question_2,