Update google_drive.py

google_drive.py

@@ -3,6 +3,7 @@ from googleapiclient.http import MediaFileUpload
 from google.oauth2 import service_account
 import json
 import os
+import traceback
 
 def upload_to_drive(local_path, filename):
     creds_dict = json.loads(os.environ["GOOGLE_CREDS_JSON"])
@@ -14,6 +15,7 @@ def upload_to_drive(local_path, filename):
         "parents": ["1RQx-f1UGOAY248bCIRasenJiQCHsD4Eq"]
     }
     media = MediaFileUpload(local_path, mimetype="application/zip")
+    
     try:
         uploaded_file = service.files().create(
             body=file_metadata,
@@ -21,9 +23,8 @@ def upload_to_drive(local_path, filename):
             fields="id"
         ).execute()
     except Exception as e:
-        import traceback
-        st.error("Upload failed:")
-        st.code(traceback.format_exc())
-        raise
+        print("Upload failed:")
+        print(traceback.format_exc())  # Or log it as needed
+        raise Exception(f"Upload to Google Drive failed: {e}")
 
     return uploaded_file.get("id")