add source search function

app.py

@@ -29,7 +29,7 @@ for data_json in json_data:
 data = pd.DataFrame(data)
 
 
-def show_search_results(language_filter, queries):
+def show_search_results(language_filter, queries, source_checkbox):
     queries = queries.lower()
     queries = queries.split()
 
@@ -43,6 +43,12 @@ def show_search_results(language_filter, queries):
             )
         ]
 
+    if "GitHub" not in source_checkbox:
+        df_search = df_search[df_search["source"] != "GitHub"]
+
+    if "Hugging Face" not in source_checkbox:
+        df_search = df_search[df_search["source"] != "Hugging Face"]
+
     for query in queries:
         contained_description = data["description"].str.contains(query)
 
@@ -90,15 +96,30 @@ with gr.Blocks() as demo:
         choices=languages,
     )
 
+    source_checkbox = gr.CheckboxGroup(
+        ["GitHub", "Hugging Face"],
+        value=["GitHub", "Hugging Face"],
+        label="Source",
+    )
+
     df = gr.DataFrame(
         value=data, type="pandas", datatype="markdown", height=1000
     )
 
     query.change(
-        fn=show_search_results, inputs=[language_selector, query], outputs=df
+        fn=show_search_results,
+        inputs=[language_selector, query, source_checkbox],
+        outputs=df,
     )
     language_selector.change(
-        fn=show_search_results, inputs=[language_selector, query], outputs=df
+        fn=show_search_results,
+        inputs=[language_selector, query, source_checkbox],
+        outputs=df,
+    )
+    source_checkbox.change(
+        fn=show_search_results,
+        inputs=[language_selector, query, source_checkbox],
+        outputs=df,
     )
 
 demo.launch()