com3dian commited on
Commit
e35a7d2
·
verified ·
1 Parent(s): 123b195

Update src/google_drive.py

Browse files
Files changed (1) hide show
  1. src/google_drive.py +11 -5
src/google_drive.py CHANGED
@@ -14,10 +14,16 @@ def upload_to_drive(local_path, filename):
14
  "parents": ["1RQx-f1UGOAY248bCIRasenJiQCHsD4Eq"]
15
  }
16
  media = MediaFileUpload(local_path, mimetype="application/zip")
17
- uploaded_file = service.files().create(
18
- body=file_metadata,
19
- media_body=media,
20
- fields="id"
21
- ).execute()
 
 
 
 
 
 
22
 
23
  return uploaded_file.get("id")
 
14
  "parents": ["1RQx-f1UGOAY248bCIRasenJiQCHsD4Eq"]
15
  }
16
  media = MediaFileUpload(local_path, mimetype="application/zip")
17
+ try:
18
+ uploaded_file = service.files().create(
19
+ body=file_metadata,
20
+ media_body=media,
21
+ fields="id"
22
+ ).execute()
23
+ except Exception as e:
24
+ import traceback
25
+ st.error("Upload failed:")
26
+ st.code(traceback.format_exc())
27
+ raise
28
 
29
  return uploaded_file.get("id")