youngtsai commited on
Commit
38a8c25
1 Parent(s): 143b0eb

with open(f'{OUTPUT_PATH}/{video_id}_summary.json', 'w') as f:

Browse files
Files changed (1) hide show
  1. app.py +21 -7
app.py CHANGED
@@ -133,6 +133,19 @@ def upload_content_directly(service, file_name, folder_id, content):
133
  print(f"上传文件时发生错误: {e}")
134
  raise # 重新抛出异常,调用者可以根据需要处理或忽略
135
 
 
 
 
 
 
 
 
 
 
 
 
 
 
136
  def download_file_as_string(service, file_id):
137
  """
138
  从Google Drive下载文件并将其作为字符串返回。
@@ -147,12 +160,6 @@ def download_file_as_string(service, file_id):
147
  content = fh.read().decode('utf-8')
148
  return content
149
 
150
- def upload_img_directly(service, file_name, folder_id, file_path):
151
- file_metadata = {'name': file_name, 'parents': [folder_id]}
152
- media = MediaFileUpload(file_path, mimetype='image/jpeg')
153
- file = service.files().create(body=file_metadata, media_body=media, fields='id').execute()
154
- return file.get('id') # 返回文件ID
155
-
156
  def set_public_permission(service, file_id):
157
  service.permissions().create(
158
  fileId=file_id,
@@ -460,7 +467,14 @@ def get_video_id_summary(video_id, df_string):
460
  # processed_summary = processed_video_summary_to_json(summary)
461
  processed_summary = { "summary": str(summary) }
462
  summary_text = json.dumps(processed_summary, ensure_ascii=False, indent=2)
463
- file_id = upload_content_directly(service, file_name, folder_id, summary_text)
 
 
 
 
 
 
 
464
  print("summary已上傳到Google Drive")
465
  else:
466
  # 逐字稿已存在,下载逐字稿内容
 
133
  print(f"上传文件时发生错误: {e}")
134
  raise # 重新抛出异常,调用者可以根据需要处理或忽略
135
 
136
+ def upload_file_directly(service, file_name, folder_id, file_path):
137
+ # 上傳 .json to Google Drive
138
+ file_metadata = {'name': file_name, 'parents': [folder_id]}
139
+ media = MediaFileUpload(file_path, mimetype='application/json')
140
+ file = service.files().create(body=file_metadata, media_body=media, fields='id').execute()
141
+ return file.get('id') # 返回文件ID
142
+
143
+ def upload_img_directly(service, file_name, folder_id, file_path):
144
+ file_metadata = {'name': file_name, 'parents': [folder_id]}
145
+ media = MediaFileUpload(file_path, mimetype='image/jpeg')
146
+ file = service.files().create(body=file_metadata, media_body=media, fields='id').execute()
147
+ return file.get('id') # 返回文件ID
148
+
149
  def download_file_as_string(service, file_id):
150
  """
151
  从Google Drive下载文件并将其作为字符串返回。
 
160
  content = fh.read().decode('utf-8')
161
  return content
162
 
 
 
 
 
 
 
163
  def set_public_permission(service, file_id):
164
  service.permissions().create(
165
  fileId=file_id,
 
467
  # processed_summary = processed_video_summary_to_json(summary)
468
  processed_summary = { "summary": str(summary) }
469
  summary_text = json.dumps(processed_summary, ensure_ascii=False, indent=2)
470
+
471
+ # 存在 local at OUTPUT_PATH as {video_id}_summary.json
472
+ with open(f'{OUTPUT_PATH}/{video_id}_summary.json', 'w') as f:
473
+ f.write(summary_text)
474
+ print(f"summary已存在 local at {OUTPUT_PATH}/{video_id}_summary.json")
475
+
476
+ # file_id = upload_content_directly(service, file_name, folder_id, summary_text)
477
+ file_id = upload_file_directly(service, file_name, folder_id, f'{OUTPUT_PATH}/{video_id}_summary.json')
478
  print("summary已上傳到Google Drive")
479
  else:
480
  # 逐字稿已存在,下载逐字稿内容