upload_content_directly

app.py

@@ -96,6 +96,18 @@ def upload_to_drive(service, file_name, folder_id, content):
     
     service.files().create(body=file_metadata, media_body=media, fields='id').execute()
 
+def upload_content_directly(service, file_name, folder_id, content):
+    """
+    直接将内容上传到Google Drive中的新文件。
+    """
+    file_metadata = {'name': file_name, 'parents': [folder_id]}
+    # 使用io.StringIO为文本内容创建一个内存中的文件对象
+    fh = io.BytesIO(content.encode('utf-8'))
+    media = MediaIoBaseUpload(fh, mimetype='text/plain', resumable=True)
+    
+    # 执行上传
+    service.files().create(body=file_metadata, media_body=media, fields='id').execute()
+
 
 def process_file(file):
     # 读取文件
@@ -172,7 +184,7 @@ def process_youtube_link(link):
         transcript = YouTubeTranscriptApi.get_transcript(video_id, languages=['zh-TW'])
         transcript_text = "\n".join([f"{item['start']}: {item['text']}" for item in transcript])
         # 上传到Google Drive
-        upload_to_drive(service, file_name, folder_id, transcript_text)
+        upload_content_directly(service, file_name, folder_id, transcript_text)
         print("逐字稿已上传到Google Drive")
     else:
         print("逐字稿已存在于Google Drive中")