update app.py

app.py

@@ -12,7 +12,15 @@ def read_json(file_name):
 
 json_file = "awesome-japanese-nlp-resources-search.json"
 json_data = read_json(json_file)
-data = {"project_name": [], "source": [], "description": [], "languages": []}
+data = {
+    "project_name": [],
+    "downloads": [],
+    "stars": [],
+    "description": [],
+    "source": [],
+    "languages": [],
+    "type": [],
+}
 
 for data_json in json_data:
     url = data_json["url"]
@@ -20,16 +28,30 @@ for data_json in json_data:
     project_name = data_json["project_name"]
     source = data_json["source"]
     languages = data_json["languages"]
+    repo_type = data_json["model_or_dataset"]
+
+    if "stargazers_count" in data_json:
+        data["stars"].append(data_json["stargazers_count"])
+    else:
+        data["stars"].append(None)
+
+    if "downloads" in data_json:
+        data["downloads"].append(data_json["downloads"])
+    else:
+        data["downloads"].append(None)
 
     data["project_name"].append(f"[{project_name}]({url})")
     data["source"].append(source)
     data["description"].append(description)
     data["languages"].append(", ".join(languages))
+    data["type"].append(repo_type)
 
 data = pd.DataFrame(data)
 
 
-def show_search_results(language_filter, queries, source_checkbox):
+def show_search_results(
+    language_filter, queries, type_checkbox, source_checkbox
+):
     queries = queries.lower()
     queries = queries.split()
 
@@ -43,6 +65,12 @@ def show_search_results(language_filter, queries, source_checkbox):
             )
         ]
 
+    if "Dataset" in type_checkbox:
+        df_search = df_search[df_search["type"] == "dataset"]
+
+    if "Model" in type_checkbox:
+        df_search = df_search[df_search["type"] == "model"]
+
     if "GitHub" not in source_checkbox:
         df_search = df_search[df_search["source"] != "GitHub"]
 
@@ -65,9 +93,7 @@ with gr.Blocks() as demo:
     """
     )
 
-    query = gr.Textbox(
-        label="Search English or Japanese words", placeholder="llm"
-    )
+    query = gr.Textbox(label="Search words", placeholder="llm")
     languages = [
         "Python",
         "Jupyter Notebook",
@@ -102,23 +128,33 @@ with gr.Blocks() as demo:
         label="Source",
     )
 
+    type_checkbox = gr.CheckboxGroup(
+        ["Dataset", "Model"],
+        label="Type",
+    )
+
     df = gr.DataFrame(
-        value=data, type="pandas", datatype="markdown", height=1000
+        value=data, type="pandas", datatype="markdown", height=1250
     )
 
     query.change(
         fn=show_search_results,
-        inputs=[language_selector, query, source_checkbox],
+        inputs=[language_selector, query, type_checkbox, source_checkbox],
         outputs=df,
     )
     language_selector.change(
         fn=show_search_results,
-        inputs=[language_selector, query, source_checkbox],
+        inputs=[language_selector, query, type_checkbox, source_checkbox],
         outputs=df,
     )
     source_checkbox.change(
         fn=show_search_results,
-        inputs=[language_selector, query, source_checkbox],
+        inputs=[language_selector, query, type_checkbox, source_checkbox],
+        outputs=df,
+    )
+    type_checkbox.change(
+        fn=show_search_results,
+        inputs=[language_selector, query, type_checkbox, source_checkbox],
         outputs=df,
     )