Update app.py

app.py

@@ -14,13 +14,13 @@ from googleapiclient.errors import HttpError
 
 def setup_kaggle_api():
     try:
-        os.environ["KAGGLE_USERNAME"] = st.secrets["KAGGLE_USERNAME"]
-        os.environ["KAGGLE_KEY"] = st.secrets["KAGGLE_KEY"]
+        os.environ["KAGGLE_USERNAME"] = os.environ.get("KAGGLE_USERNAME")
+        os.environ["KAGGLE_KEY"] = os.environ.get("KAGGLE_KEY")
         api = KaggleApi()
         api.authenticate()
         return api
     except KeyError as e:
-        st.error(f"Missing Kaggle secret: {e}")
+        st.error(f"Missing Kaggle environment variable: {e}")
         st.stop()
     except Exception as e:
         st.error(f"Kaggle API setup failed: {e}")
@@ -28,7 +28,8 @@ def setup_kaggle_api():
 
 def setup_drive_service():
     try:
-        credentials_dict = dict(st.secrets["GOOGLE_SERVICE_ACCOUNT"])
+        credentials_json = os.environ.get("GOOGLE_SERVICE_ACCOUNT")
+        credentials_dict = json.loads(credentials_json)
         credentials = Credentials(
             token=credentials_dict["access_token"],
             refresh_token=credentials_dict["refresh_token"],
@@ -42,14 +43,18 @@ def setup_drive_service():
         drive_service = build('drive', 'v3', credentials=credentials)
         return drive_service
     except KeyError as e:
-        st.error(f"Missing key in GOOGLE_SERVICE_ACCOUNT: {e}")
+        st.error(f"Missing GOOGLE_SERVICE_ACCOUNT environment variable: {e}")
+        st.stop()
+    except json.JSONDecodeError as e:
+        st.error(f"Invalid JSON in GOOGLE_SERVICE_ACCOUNT: {e}")
         st.stop()
     except Exception as e:
         st.error(f"Google Drive auth failed: {e}")
         st.stop()
 
-def upload_to_drive(drive_service, file_path, title, folder_id="1T6v7Iqc90-NA-F3I-HeHDSvEaIyFibKd"):
+def upload_to_drive(drive_service, file_path, title, folder_id=None):
     try:
+        folder_id = os.environ.get("DRIVE_FOLDER_ID", "1T6v7Iqc90-NA-F3I-HeHDSvEaIyFibKd")
         file_metadata = {'name': title, 'parents': [folder_id]}
         media = MediaFileUpload(file_path, resumable=True)
         file = drive_service.files().create(
@@ -76,8 +81,9 @@ def delete_from_drive(drive_service, file_id):
     except Exception as e:
         st.error(f"Failed to delete file {file_id} from Drive: {e}")
 
-def get_bvh_from_folder(drive_service, folder_id="1T6v7Iqc90-NA-F3I-HeHDSvEaIyFibKd"):
+def get_bvh_from_folder(drive_service, folder_id=None):
     try:
+        folder_id = os.environ.get("DRIVE_FOLDER_ID", "1T6v7Iqc90-NA-F3I-HeHDSvEaIyFibKd")
         query = f"'{folder_id}' in parents and name contains '.bvh'"
         response = drive_service.files().list(q=query, fields="files(id, name, mimeType)", pageSize=1).execute()
         files = response.get('files', [])
@@ -155,7 +161,7 @@ def process_video(api, drive_service, video_file):
         video_file_id, video_shareable_link = upload_to_drive(drive_service, video_path, "input_video.mp4")
         st.success(f"Video uploaded to Drive: {video_shareable_link}")
 
-        username = st.secrets['KAGGLE_USERNAME']
+        username = os.environ.get("KAGGLE_USERNAME")
         notebook_slug = f"{username}/video-to-bvh-converter"
         
         kernel_exists = check_kernel_exists(api, notebook_slug)
@@ -292,7 +298,7 @@ def main():
     status_col1, status_col2 = st.columns(2)
     with status_col1:
         try:
-            username = st.secrets['KAGGLE_USERNAME']
+            username = os.environ.get("KAGGLE_USERNAME")
             notebook_slug = f"{username}/video-to-bvh-converter"
             kernel_exists = check_kernel_exists(api, notebook_slug)
             if kernel_exists: