Add Gradio interface for pulling Kaggle kernels and implement kernel download functionality

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-import os, json, pathlib, tempfile, datetime
+import os, json, pathlib, tempfile, datetime, shutil
 from typing import List, Dict, Optional
 
 from dotenv import load_dotenv
@@ -65,6 +65,16 @@ def search_kernels(query: str, max_results: int = 20) -> List[Dict]:
         )
     return out
     
+def pull_kernel(kernel_ref: str) -> str:
+    tmp_dir = tempfile.mkdtemp()
+    api.kernels_pull(kernel_ref, path=tmp_dir, quiet=False)
+
+    zip_path = shutil.make_archive(
+        base_name=os.path.join(tmp_dir, "kernel"),
+        format = "zip",
+        root_dir=tmp_dir,
+    )
+    return zip_path
 
 search_iface = gr.Interface(
     fn=search_datasets,
@@ -77,6 +87,17 @@ search_iface = gr.Interface(
     description="Resturns a JSON array of dataset metadata."
 )
 
+pull_kernel_iface = gr.Interface(
+    fn = pull_kernel,
+    inputs=gr.Textbox(
+        label="kernel reference",
+        placeholder="e.g. username/notebook-name",
+    ),
+    outputs=gr.File(label="Downlaod .zip"),
+    title="pull kaggle kernel",
+    description="Downlaods the notebook or script kernel and returns a ZIP archive."
+)
+
 list_files_iface = gr.Interface(
     fn=list_files,
     inputs=gr.Textbox(label="Dataset slug", placeholder="zalando-research/fashionmnist"),
@@ -108,8 +129,8 @@ search_kernels_iface = gr.Interface(
 )
 
 demo = gr.TabbedInterface(
-    [search_iface, list_files_iface, download_file_iface, search_kernels_iface],
-    tab_names=["Search Datasets", "Files", "Download File", "Search Kernels"],
+    [search_iface, list_files_iface, download_file_iface, search_kernels_iface, pull_kernel_iface],
+    tab_names=["Search Datasets", "Files", "Download File", "Search Kernels", "Pull kernel"],
 )
 
 def _bootstrap_kaggle_credentials():