Update app.py

app.py

@@ -110,6 +110,36 @@ def download_notebook_from_drive(drive_service, temp_dir):
             downloader = MediaFileUpload(notebook_path)
             downloader.stream = request.execute()
             f.write(downloader.stream)
+            
+        # 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
+            },
+            "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}")
@@ -135,6 +165,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
             "competition_sources": [],
             "dataset_sources": [],
             "kernel_sources": []
@@ -157,18 +189,7 @@ 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