Spaces:
Sleeping
Sleeping
with open(f'{OUTPUT_PATH}/{video_id}_summary.json', 'w') as f:
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
# 逐字稿已存在,下载逐字稿内容
|