reading_passage_admin

app.py

@@ -770,41 +770,6 @@ def generate_reading_passage(df_string):
 
     return reading_passage
 
-def enable_edit_mode():
-    return gr.update(interactive=True)
-
-def delete_reading_passage(video_id):
-    print("===delete_reading_passage===")
-    gcs_client = GCS_CLIENT
-    bucket_name = 'video_ai_assistant'
-    file_name = f'{video_id}_reading_passage.json'
-    blob_name = f"{video_id}/{file_name}"
-    # 检查 reading_passage 是否存在
-    is_file_exists = GCS_SERVICE.check_file_exists(bucket_name, blob_name)
-    if is_file_exists:
-        delete_blob(gcs_client, bucket_name, blob_name)
-        print("reading_passage已从GCS中删除")
-    return gr.update(value="", interactive=False)
-
-def update_reading_passage(video_id, new_content):
-    print("===upfdate_reading_passage on gcs===")
-    gcs_client = GCS_CLIENT
-    bucket_name = 'video_ai_assistant'
-    file_name = f'{video_id}_reading_passage.json'
-    blob_name = f"{video_id}/{file_name}"
-    reading_passage_json = {"reading_passage": str(new_content)}
-    reading_passage_text = json.dumps(reading_passage_json, ensure_ascii=False, indent=2)
-    upload_file_to_gcs_with_json_string(gcs_client, bucket_name, blob_name, reading_passage_text)
-    print("reading_passage已更新到GCS")
-    return gr.update(value=new_content, interactive=False)
-
-def create_reading_passage(video_id, df_string):
-    print("===create_reading_passage===")
-    content = generate_reading_passage(df_string)
-    update_reading_passage(video_id, content)
-    return gr.update(value=content, interactive=False)
-
-
 def text_to_speech(video_id, text):
     tts = gTTS(text, lang='en')
     filename = f'{video_id}_reading_passage.mp3'
@@ -1092,6 +1057,58 @@ def change_questions(password, df_string):
     print("=====get_questions=====")
     return q1, q2, q3
 
+# ---- LLM CRUD ----
+def enable_edit_mode():
+    return gr.update(interactive=True)
+
+def delete_LLM_content(video_id, kind):
+    print(f"===delete_{kind}===")
+    gcs_client = GCS_CLIENT
+    bucket_name = 'video_ai_assistant'
+    file_name = f'{video_id}_{kind}.json'
+    blob_name = f"{video_id}/{file_name}"
+    # 检查 reading_passage 是否存在
+    is_file_exists = GCS_SERVICE.check_file_exists(bucket_name, blob_name)
+    if is_file_exists:
+        delete_blob(gcs_client, bucket_name, blob_name)
+        print("reading_passage已从GCS中删除")
+    return gr.update(value="", interactive=False)
+
+def update_LLM_content(video_id, new_content, kind):
+    print(f"===upfdate kind on gcs===")
+    gcs_client = GCS_CLIENT
+    bucket_name = 'video_ai_assistant'
+    file_name = f'{video_id}_{kind}.json'
+    blob_name = f"{video_id}/{file_name}"
+
+    if kind == "reading_passage":
+        reading_passage_json = {"reading_passage": str(new_content)}
+        reading_passage_text = json.dumps(reading_passage_json, ensure_ascii=False, indent=2)
+        upload_file_to_gcs_with_json_string(gcs_client, bucket_name, blob_name, reading_passage_text)
+    elif kind == "summary":
+        summary_json = {"summary": str(new_content)}
+        summary_text = json.dumps(summary_json, ensure_ascii=False, indent=2)
+        upload_file_to_gcs_with_json_string(gcs_client, bucket_name, blob_name, summary_text)
+    elif kind == "mind_map":
+        mind_map_json = {"mind_map": str(new_content)}
+        mind_map_text = json.dumps(mind_map_json, ensure_ascii=False, indent=2)
+        upload_file_to_gcs_with_json_string(gcs_client, bucket_name, blob_name, mind_map_text)
+
+    print(f"{kind} 已更新到GCS")
+    return gr.update(value=new_content, interactive=False)
+
+def create_LLM_content(video_id, df_string, kind):
+    print(f"===create_{kind}===")
+    if kind == "reading_passage":
+        content = generate_reading_passage(df_string)
+    elif kind == "summary":
+        content = generate_summarise(df_string)
+    elif kind == "mind_map":
+        content = generate_mind_map(df_string)
+
+    update_LLM_content(video_id, content, kind)
+    return gr.update(value=content, interactive=False)
+
 
 # AI 生成教學素材
 def get_meta_data(video_id, source="gcs"):
@@ -1539,9 +1556,9 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
         with gr.Row():
             with gr.Column(scale=1):
                 gr.Markdown("## 你也有類似的問題想問嗎？")
-                btn_1 = gr.Button()
-                btn_2 = gr.Button()
-                btn_3 = gr.Button()
+                btn_1 = gr.Button("問題一")
+                btn_2 = gr.Button("問題一")
+                btn_3 = gr.Button("問題一")
                 gr.Markdown("## 重新生成問題")
                 btn_create_question = gr.Button("生成其他問題", variant="primary")
             with gr.Column(scale=2):
@@ -1576,6 +1593,7 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
                     ai_send_button = gr.Button("Send", variant="primary")    
     with gr.Tab("文章模式"):
         with gr.Row() as reading_passage_admin:
+            reading_passage_kind = gr.Textbox(value="reading_passage")
             reading_passage_edit_button = gr.Button("編輯", size="sm", variant="primary")
             reading_passage_update_button = gr.Button("更新", size="sm", variant="primary")
             reading_passage_delete_button = gr.Button("刪除", size="sm", variant="primary")
@@ -1583,10 +1601,16 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
         with gr.Row():
             reading_passage = gr.Textbox(label="Reading Passage", lines=40, show_label=False)
             reading_passage_speak_button = gr.Button("Speak", visible=False)
-            reading_passage_audio_output = gr.Audio(label="Audio Output", visible=False)
-        
+            reading_passage_audio_output = gr.Audio(label="Audio Output", visible=False) 
     with gr.Tab("重點摘要"):
-        df_summarise = gr.Textbox(container=True, show_copy_button=True, lines=40, show_label=False)    
+        with gr.Row() as summary_admmin:
+            summary_kind = gr.Textbox(value="summary")
+            summary_edit_button = gr.Button("編輯", size="sm", variant="primary")
+            summary_update_button = gr.Button("更新", size="sm", variant="primary")
+            summary_delete_button = gr.Button("刪除", size="sm", variant="primary")
+            summary_create_button = gr.Button("建立", size="sm", variant="primary")
+        with gr.Row():
+            df_summarise = gr.Textbox(container=True, show_copy_button=True, lines=40, show_label=False)    
     with gr.Tab("教學備課"):
         with gr.Row():
             content_subject = gr.Dropdown(label="選擇主題", choices=["數學", "自然", "國文", "英文", "社會","物理", "化學", "生物", "地理", "歷史", "公民"], value="", visible=False)             
@@ -1795,13 +1819,13 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
 
     # reading_passage event
     reading_passage_create_button.click(
-        create_reading_passage,
-        inputs=[video_id, df_string_output],
+        create_LLM_content,
+        inputs=[video_id, df_string_output, reading_passage_kind],
         outputs=[reading_passage]
     )
     reading_passage_delete_button.click(
-        delete_reading_passage,
-        inputs=[video_id],
+        delete_LLM_content,
+        inputs=[video_id, reading_passage_kind],
         outputs=[reading_passage]
     )
     reading_passage_edit_button.click(
@@ -1810,11 +1834,33 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
         outputs=[reading_passage]
     )
     reading_passage_update_button.click(
-        update_reading_passage,
-        inputs=[video_id, reading_passage],
+        update_LLM_content,
+        inputs=[video_id, reading_passage, reading_passage_kind],
         outputs=[reading_passage]
     )
 
+    # summary event
+    summary_create_button.click(
+        create_LLM_content,
+        inputs=[video_id, df_string_output, summary_kind],
+        outputs=[df_summarise]
+    )
+    summary_delete_button.click(
+        delete_LLM_content,
+        inputs=[video_id, summary_kind],
+        outputs=[df_summarise]
+    )
+    summary_edit_button.click(
+        enable_edit_mode,
+        inputs=[],
+        outputs=[df_summarise]
+    )
+    summary_update_button.click(
+        update_LLM_content,
+        inputs=[video_id, df_summarise, summary_kind],
+        outputs=[df_summarise]
+    )
+
     # 教師版
     worksheet_content_btn.click(
         get_ai_content,