CRUD

app.py

@@ -114,7 +114,7 @@ def upload_file_to_gcs_with_json_string(gcs_client, bucket_name, destination_blo
     bucket = gcs_client.bucket(bucket_name)
     blob = bucket.blob(destination_blob_name)
     blob.upload_from_string(json_string)
-    print(f"JSON string uploaded to {destination_blob_name} in GCS.")    
+    print(f"JSON string uploaded to {destination_blob_name} in GCS.")
 
 def download_blob_to_string(gcs_client, bucket_name, source_blob_name):
     """从 GCS 下载文件内容到字符串"""
@@ -175,6 +175,13 @@ def copy_file_from_drive_to_gcs(drive_service, gcs_client, file_id, bucket_name,
     blob.upload_from_string(file_content)
     print(f"File {file_id} copied to GCS at {gcs_destination_path}.")
 
+def delete_blob(gcs_client, bucket_name, blob_name):
+    """删除指定的 GCS 对象"""
+    bucket = gcs_client.bucket(bucket_name)
+    blob = bucket.blob(blob_name)
+    blob.delete()
+    print(f"Blob {blob_name} deleted from GCS.")
+
 # # ====drive====初始化
 def init_drive_service():
     credentials_json_string = DRIVE_KEY
@@ -763,16 +770,39 @@ def generate_reading_passage(df_string):
 
     return reading_passage
 
-def edit_generate_reading():
-    # 將 reading_passage 設置為可編輯模式並顯示更新按鈕
-    return gr.update(interactive=True), gr.update(visible=True)
+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 update_reading_passage(new_content, video_id):
-    # 更新 GCS 上的 reading_passage 內容
-    # 此處假設已有函數 update_reading_passage_to_gcs 用於更新 GCS 內容
-    # update_reading_passage_to_gcs(video_id, new_content)
-    # 將 reading_passage 設置回唯讀模式並隱藏更新按鈕
-    return gr.update(value=new_content, interactive=False), gr.update(visible=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):
@@ -1545,12 +1575,16 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
                     ai_msg = gr.Textbox(label="Message")
                     ai_send_button = gr.Button("Send", variant="primary")    
     with gr.Tab("文章模式"):
-        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)
         with gr.Row() as reading_passage_admin:
-            reading_passage_edit_button = gr.Button("編輯")
-            reading_passage_update_button = gr.Button("更新")
+            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")
+            reading_passage_create_button = gr.Button("建立", size="sm", variant="primary")
+        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)
+        
     with gr.Tab("重點摘要"):
         df_summarise = gr.Textbox(container=True, show_copy_button=True, lines=40, show_label=False)    
     with gr.Tab("教學備課"):
@@ -1759,6 +1793,28 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue=gr.themes.colors.orange, seconda
     # 当输入网页链接时触发
     # web_link.change(process_web_link, inputs=web_link, outputs=[btn_1, btn_2, btn_3, df_summarise, df_string_output])
 
+    # reading_passage event
+    reading_passage_create_button.click(
+        create_reading_passage,
+        inputs=[video_id, df_string_output],
+        outputs=[reading_passage]
+    )
+    reading_passage_delete_button.click(
+        delete_reading_passage,
+        inputs=[video_id],
+        outputs=[reading_passage]
+    )
+    reading_passage_edit_button.click(
+        enable_edit_mode,
+        inputs=[],
+        outputs=[reading_passage]
+    )
+    reading_passage_update_button.click(
+        update_reading_passage,
+        inputs=[video_id, reading_passage],
+        outputs=[reading_passage]
+    )
+
     # 教師版
     worksheet_content_btn.click(
         get_ai_content,