Update app.py

app.py

@@ -97,16 +97,32 @@ def get_bvh_from_folder(drive_service, folder_id=None):
         st.error(f"Error checking folder for BVH: {e}")
         return None, None
 
-def push_kaggle_kernel(api, temp_dir, notebook_slug):
+def download_notebook_from_drive(drive_service, temp_dir):
     try:
-        local_notebook_path = os.path.join(os.path.dirname(__file__), 'video-to-bvh-converter.ipynb')
-        if not os.path.exists(local_notebook_path):
-            st.error(f"Notebook file not found")
-            raise FileNotFoundError
+        notebook_file_id = os.environ.get("NOTEBOOK_FILE_ID")
+        if not notebook_file_id:
+            st.error("NOTEBOOK_FILE_ID environment variable not set")
+            raise KeyError("NOTEBOOK_FILE_ID not set")
+        
+        request = drive_service.files().get_media(fileId=notebook_file_id)
+        notebook_path = os.path.join(temp_dir, 'video-to-bvh-converter.ipynb')
+        with open(notebook_path, 'wb') as f:
+            downloader = MediaFileUpload(notebook_path)
+            downloader.stream = request.execute()
+            f.write(downloader.stream)
+        return notebook_path
+    except Exception as e:
+        st.error(f"Failed to download notebook from Drive: {e}")
+        raise
 
+def push_kaggle_kernel(api, temp_dir, notebook_slug):
+    try:
+        drive_service = setup_drive_service()
+        local_notebook_path = download_notebook_from_drive(drive_service, temp_dir)
+        
         kernel_file = os.path.join(temp_dir, 'kernel.ipynb')
         shutil.copy(local_notebook_path, kernel_file)
-
+        
         code_file = "kernel.ipynb"
         kernel_type = "notebook"
 
@@ -133,6 +149,9 @@ def push_kaggle_kernel(api, temp_dir, notebook_slug):
             st.error(f"kernel push failed: {result.stderr}")
             raise Exception
         
+        os.remove(local_notebook_path)
+        os.remove(kernel_file)
+        
         kernel_url = f"https://www.kaggle.com/code/{notebook_slug}"
         return kernel_url
     except Exception as e: