remove language filter and update output fields
Browse files
app.py
CHANGED
@@ -35,12 +35,10 @@ def download_file(dataset_slug: str, file_name: str):
|
|
35 |
zip_path = pathlib.Path(tmp_dir) / f"{file_name}.zip"
|
36 |
return str(zip_path)
|
37 |
|
38 |
-
def search_kernels(query: str, max_results: int = 20
|
39 |
-
|
40 |
-
|
41 |
kernels = api.kernels_list(
|
42 |
search=query,
|
43 |
-
language=language_filter,
|
44 |
page_size=min(max_results, 20),
|
45 |
sort_by="voteCount",
|
46 |
)
|
@@ -59,9 +57,9 @@ def search_kernels(query: str, max_results: int = 20, language: str = "Python")
|
|
59 |
{
|
60 |
"title": k.title,
|
61 |
"ref": k.ref,
|
62 |
-
"language": k
|
63 |
-
"
|
64 |
-
"
|
65 |
"last_run": last_run,
|
66 |
}
|
67 |
)
|
@@ -71,7 +69,7 @@ def search_kernels(query: str, max_results: int = 20, language: str = "Python")
|
|
71 |
search_iface = gr.Interface(
|
72 |
fn=search_datasets,
|
73 |
inputs=[
|
74 |
-
gr.Textbox(label="Search term", placeholder="e.g.
|
75 |
gr.Slider(1, 50, step=1, value=20, label="Max results")
|
76 |
],
|
77 |
outputs=gr.JSON(label="Datasets"),
|
@@ -81,7 +79,7 @@ search_iface = gr.Interface(
|
|
81 |
|
82 |
list_files_iface = gr.Interface(
|
83 |
fn=list_files,
|
84 |
-
inputs=gr.Textbox(label="Dataset slug", placeholder="
|
85 |
outputs=gr.JSON(label="Files"),
|
86 |
title="List Dataset Files",
|
87 |
description="Given a dataset slug, returns its file list."
|
@@ -90,8 +88,8 @@ list_files_iface = gr.Interface(
|
|
90 |
download_file_iface = gr.Interface(
|
91 |
fn=download_file,
|
92 |
inputs=[
|
93 |
-
gr.Textbox(label="Dataset slug", placeholder="
|
94 |
-
gr.Textbox(label="File name", placeholder="
|
95 |
],
|
96 |
outputs=gr.File(label="Download file"),
|
97 |
title="Download a File",
|
@@ -99,15 +97,14 @@ download_file_iface = gr.Interface(
|
|
99 |
)
|
100 |
|
101 |
search_kernels_iface = gr.Interface(
|
102 |
-
fn
|
103 |
-
inputs
|
104 |
gr.Textbox(label="search term", placeholder="e.g. computer vision"),
|
105 |
gr.Slider(1, 50, step=1, value=20, label="Max results"),
|
106 |
-
gr.Dropdown(["All", "Python", "R"], value="Python", label="Language"),
|
107 |
],
|
108 |
outputs=gr.JSON(label="kernels"),
|
109 |
title="Search kaggle kernels",
|
110 |
-
description="Find notebook or script kernels by keyword
|
111 |
)
|
112 |
|
113 |
demo = gr.TabbedInterface(
|
|
|
35 |
zip_path = pathlib.Path(tmp_dir) / f"{file_name}.zip"
|
36 |
return str(zip_path)
|
37 |
|
38 |
+
def search_kernels(query: str, max_results: int = 20) -> List[Dict]:
|
39 |
+
|
|
|
40 |
kernels = api.kernels_list(
|
41 |
search=query,
|
|
|
42 |
page_size=min(max_results, 20),
|
43 |
sort_by="voteCount",
|
44 |
)
|
|
|
57 |
{
|
58 |
"title": k.title,
|
59 |
"ref": k.ref,
|
60 |
+
"language": getattr(k, "language", None),
|
61 |
+
"kernel_type": getattr(k, "kernelType", None),
|
62 |
+
"votes": k.total_votes,
|
63 |
"last_run": last_run,
|
64 |
}
|
65 |
)
|
|
|
69 |
search_iface = gr.Interface(
|
70 |
fn=search_datasets,
|
71 |
inputs=[
|
72 |
+
gr.Textbox(label="Search term", placeholder="e.g. fashion mnist"),
|
73 |
gr.Slider(1, 50, step=1, value=20, label="Max results")
|
74 |
],
|
75 |
outputs=gr.JSON(label="Datasets"),
|
|
|
79 |
|
80 |
list_files_iface = gr.Interface(
|
81 |
fn=list_files,
|
82 |
+
inputs=gr.Textbox(label="Dataset slug", placeholder="zalando-research/fashionmnist"),
|
83 |
outputs=gr.JSON(label="Files"),
|
84 |
title="List Dataset Files",
|
85 |
description="Given a dataset slug, returns its file list."
|
|
|
88 |
download_file_iface = gr.Interface(
|
89 |
fn=download_file,
|
90 |
inputs=[
|
91 |
+
gr.Textbox(label="Dataset slug", placeholder="zalando-research/fashionmnist"),
|
92 |
+
gr.Textbox(label="File name", placeholder="fashion-mnist_test.csv")
|
93 |
],
|
94 |
outputs=gr.File(label="Download file"),
|
95 |
title="Download a File",
|
|
|
97 |
)
|
98 |
|
99 |
search_kernels_iface = gr.Interface(
|
100 |
+
fn=search_kernels,
|
101 |
+
inputs=[
|
102 |
gr.Textbox(label="search term", placeholder="e.g. computer vision"),
|
103 |
gr.Slider(1, 50, step=1, value=20, label="Max results"),
|
|
|
104 |
],
|
105 |
outputs=gr.JSON(label="kernels"),
|
106 |
title="Search kaggle kernels",
|
107 |
+
description="Find notebook or script kernels by keyword."
|
108 |
)
|
109 |
|
110 |
demo = gr.TabbedInterface(
|