Roobick commited on
Commit
d855e11
·
1 Parent(s): 7b0d4d1

remove language filter and update output fields

Browse files
Files changed (1) hide show
  1. app.py +12 -15
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, language: str = "Python") -> List[Dict]:
39
- language_filter : Optional[str] = None if language == "all" else language.lower()
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.language,
63
- "votes": k.totalVotes,
64
- "status": k.kernelStatus,
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. titanic"),
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="zynicide/wine-reviews"),
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="zynicide/wine-reviews"),
94
- gr.Textbox(label="File name", placeholder="winemag-data_first150k.csv")
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 = search_kernels,
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, language, etc."
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(