summary_json = {"summary": str(summary)}

app.py

@@ -527,7 +527,7 @@ def process_youtube_link(link):
     simple_html_content = format_simple_transcript_to_html(formatted_simple_transcript)
     first_image = formatted_transcript[0]['screenshot_path']
     first_text = formatted_transcript[0]['text']
-    mind_map_json = get_mind_map(video_id, formatted_simple_transcript)
+    mind_map_json = get_mind_map(video_id, formatted_simple_transcript, source)
     mind_map = mind_map_json["mind_map"]
     mind_map_html = get_mind_map_html(mind_map)
 
@@ -596,27 +596,47 @@ def process_web_link(link):
     soup = BeautifulSoup(response.content, 'html.parser')
     return soup.get_text()
 
-def get_mind_map(video_id, df_string):
-    # 先抓 g drive 看看有沒有 {video_id}_mind_map.json
-    print("===get_mind_map===")
-    service = init_drive_service()
-    parent_folder_id = '1GgI4YVs0KckwStVQkLa1NZ8IpaEMurkL'
-    folder_id = create_folder_if_not_exists(service, video_id, parent_folder_id)
-    file_name = f'{video_id}_mind_map.json'
+def get_mind_map(video_id, df_string, source):
+    if source == "gcs":
+        print("===get_mind_map on gcs===")
+        gcs_client = init_gcs_client(GCS_KEY)
+        bucket_name = 'video_ai_assistant'
+        file_name = f'{video_id}_mind_map.json'
+        blob_name = f"{video_id}/{file_name}"
+        # 检查檔案是否存在
+        is_file_exists = gcs_check_file_exists(gcs_client, bucket_name, blob_name)
+        if not is_file_exists:
+            mind_map = generate_mind_map(df_string)
+            mind_map_json = {"mind_map": str(mind_map)}
+            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("mind_map已上傳到GCS")
+        else:
+            # mindmap已存在，下载内容
+            print("mind_map已存在于GCS中")
+            mind_map_text = download_blob_to_string(gcs_client, bucket_name, blob_name)
+            mind_map_json = json.loads(mind_map_text)
 
-    # 检查檔案是否存在
-    exists, file_id = check_file_exists(service, folder_id, file_name)
-    if not exists:
-        mind_map = generate_mind_map(df_string)
-        mind_map_json = {"mind_map": str(mind_map)}
-        mind_map_text = json.dumps(mind_map_json, ensure_ascii=False, indent=2)
-        upload_content_directly(service, file_name, folder_id, mind_map_text)
-        print("mind_map已上傳到Google Drive")
-    else:
-        # mindmap已存在，下载内容
-        print("mind_map已存在于Google Drive中")
-        mind_map_text = download_file_as_string(service, file_id)
-        mind_map_json = json.loads(mind_map_text)
+    elif source == "drive":
+        print("===get_mind_map on drive===")
+        service = init_drive_service()
+        parent_folder_id = '1GgI4YVs0KckwStVQkLa1NZ8IpaEMurkL'
+        folder_id = create_folder_if_not_exists(service, video_id, parent_folder_id)
+        file_name = f'{video_id}_mind_map.json'
+
+        # 检查檔案是否存在
+        exists, file_id = check_file_exists(service, folder_id, file_name)
+        if not exists:
+            mind_map = generate_mind_map(df_string)
+            mind_map_json = {"mind_map": str(mind_map)}
+            mind_map_text = json.dumps(mind_map_json, ensure_ascii=False, indent=2)
+            upload_content_directly(service, file_name, folder_id, mind_map_text)
+            print("mind_map已上傳到Google Drive")
+        else:
+            # mindmap已存在，下载内容
+            print("mind_map已存在于Google Drive中")
+            mind_map_text = download_file_as_string(service, file_id)
+            mind_map_json = json.loads(mind_map_text)
 
     return mind_map_json
 
@@ -712,7 +732,8 @@ def get_video_id_summary(video_id, df_string, source):
         # 检查 summary_file 是否存在
         is_summary_file_exists = gcs_check_file_exists(gcs_client, bucket_name, summary_file_blob_name)
         if not is_summary_file_exists:
-            summary_json = processed_video_summary_to_json(df_string)
+            summary = generate_summarise(df_string)
+            summary_json = {"summary": str(summary)}
             summary_text = json.dumps(summary_json, ensure_ascii=False, indent=2)
             upload_file_to_gcs_with_json_string(gcs_client, bucket_name, summary_file_blob_name, summary_text)
             print("summary已上传到GCS")
@@ -733,7 +754,6 @@ def get_video_id_summary(video_id, df_string, source):
         exists, file_id = check_file_exists(service, folder_id, file_name)
         if not exists:
             summary = generate_summarise(df_string)
-            # processed_summary = processed_video_summary_to_json(summary)
             summary_json = {"summary": str(summary)}
             summary_text = json.dumps(summary_json, ensure_ascii=False, indent=2)
             
@@ -746,14 +766,6 @@ def get_video_id_summary(video_id, df_string, source):
                 print("===get_video_id_summary error===")
                 print(error_msg)
                 print("===get_video_id_summary error===")
-
-
-
-            # 存在 local at OUTPUT_PATH as {video_id}_summary.json
-            # with open(f'{OUTPUT_PATH}/{video_id}_summary.json', 'w') as f:
-            #     f.write(summary_text)
-            #     print(f"summary已存在 local at {OUTPUT_PATH}/{video_id}_summary.json")
-            # file_id = upload_file_directly(service, file_name, folder_id, f'{OUTPUT_PATH}/{video_id}_summary.json')
         else:
             # 逐字稿已存在，下载逐字稿内容
             print("summary已存在Google Drive中")