Update app.py

app.py

@@ -107,52 +107,36 @@ def download_notebook_from_drive(drive_service, temp_dir):
         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)
-            
+            f.write(request.execute())  # Directly write the downloaded content
+        
         # Add kernel specification to notebook
         with open(notebook_path, 'r') as f:
             notebook_content = json.load(f)
         
-        # Add proper kernel specification
         notebook_content['metadata']['kernelspec'] = {
             "display_name": "Python 3",
             "language": "python",
             "name": "python3"
         }
         
-        # Ensure language_info is complete
         notebook_content['metadata']['language_info'] = {
             "name": "python",
             "version": "3.10.12",
             "mimetype": "text/x-python",
-            "codemirror_mode": {
-                "name": "ipython",
-                "version": 3
-            },
+            "codemirror_mode": {"name": "ipython", "version": 3},
             "pygments_lexer": "ipython3",
             "nbconvert_exporter": "python",
             "file_extension": ".py"
         }
-            
-        # Write back the modified notebook
+        
         with open(notebook_path, 'w') as f:
             json.dump(notebook_content, f)
-            
+        
         return notebook_path
     except Exception as e:
         st.error(f"Failed to download notebook from Drive: {e}")
         raise
 
-def check_kernel_status(api, notebook_slug):
-    try:
-        status_response = api.kernel_status(notebook_slug)
-        return status_response
-    except Exception as e:
-        st.error(f"Error checking kernel status: {e}")
-        return None
-
 def push_kaggle_kernel(api, temp_dir, notebook_slug):
     try:
         drive_service = setup_drive_service()
@@ -161,6 +145,11 @@ def push_kaggle_kernel(api, temp_dir, notebook_slug):
         kernel_file = os.path.join(temp_dir, 'kernel.ipynb')
         shutil.copy(local_notebook_path, kernel_file)
         
+        # Verify kernel spec
+        with open(kernel_file, 'r') as f:
+            metadata_content = json.load(f)['metadata']
+            st.write("Notebook metadata:", metadata_content)  # Debug output
+        
         code_file = "kernel.ipynb"
         kernel_type = "notebook"
 
@@ -173,8 +162,8 @@ def push_kaggle_kernel(api, temp_dir, notebook_slug):
             "enable_gpu": True,
             "enable_internet": True,
             "is_private": True,
-            "accelerator": "gpu",  # Explicitly specify GPU
-            "gpu_product": "T4x2",  # Specify T4x2 GPU
+            "accelerator": "gpu",
+            "gpu_product": "T4x2",
             "competition_sources": [],
             "dataset_sources": [],
             "kernel_sources": []
@@ -186,8 +175,8 @@ def push_kaggle_kernel(api, temp_dir, notebook_slug):
         cmd = f"kaggle kernels push -p {temp_dir}"
         result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
         if result.returncode != 0:
-            st.error(f"kernel push failed: {result.stderr}")
-            raise Exception
+            st.error(f"Kernel push failed: {result.stderr}")
+            raise Exception(f"Push failed: {result.stderr}")
         
         os.remove(local_notebook_path)
         os.remove(kernel_file)
@@ -197,7 +186,18 @@ def push_kaggle_kernel(api, temp_dir, notebook_slug):
     except Exception as e:
         st.error(f"Failed to push kernel: {str(e)}")
         raise
-        
+
+def check_kernel_exists(api, notebook_slug):
+    try:
+        kernels = api.kernels_list(mine=True, search=notebook_slug)
+        for kernel in kernels:
+            if kernel.ref == notebook_slug:
+                return True
+        return False
+    except Exception as e:
+        st.error(f"Kernel check failed: {e}")
+        return False
+
 def process_video(api, drive_service, video_file):
     video_file_id = None
     bvh_file_id = None
@@ -233,7 +233,7 @@ def process_video(api, drive_service, video_file):
 
             while time.time() - start_time < overall_timeout:
                 try:
-                    status_response = check_kernel_status(api, notebook_slug)
+                    status_response = api.kernels_status(notebook_slug)
                     current_status = status_response.status if hasattr(status_response, 'status') else 'unknown'
                     
                     if current_status in ['queued', 'running']: