Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
 import requests
 
-def search_huggingface_datasets(query):
+def search_huggingface_datasets(query, type):
     if not query:  # If the query is empty, return an empty list
         return []
-    url = f"https://huggingface.co/api/quicksearch?q={query}&type=dataset&limit=20"
+    url = f"https://huggingface.co/api/quicksearch?q={query}&type={type}&limit=20"
     response = requests.get(url)
     if response.status_code == 200:
         data = response.json()
@@ -13,31 +13,17 @@ def search_huggingface_datasets(query):
     else:
         return ["Error fetching datasets"]
 
-def update_dropdown(query, key_up_data: gr.KeyUpData):
-    datasets = search_huggingface_datasets(key_up_data.input_value)
-    print(key_up_data.input_value)
+def update_dropdown(query, type, key_up_data: gr.KeyUpData):
+    datasets = search_huggingface_datasets(key_up_data.input_value, type)
     return gr.update(choices=datasets, visible=True)
 
-def update_dropdown_value(query, key_up_data: gr.KeyUpData):
-    datasets = search_huggingface_datasets(key_up_data.input_value)
-    print(key_up_data.input_value)
-    return gr.update(choices=datasets, value=key_up_data.input_value, visible=True)
+
 
 with gr.Blocks() as demo:
-    with gr.Row():
-        with gr.Column():
-            dataset_dropdown = gr.Dropdown(label="Datasets Auto-Complete", choices=[""], elem_id="dataset_list", interactive=True, filterable=True, allow_custom_value=True)
-            gr.Markdown('''Here the return is:
-```
-gr.update(choices=datasets, visible=True)
-```
-''')
-        with gr.Column():
-            dataset_dropdown_update = gr.Dropdown(label="Datasets Auto-Complete", choices=[""], elem_id="dataset_list", interactive=True, filterable=True, allow_custom_value=True)
-            gr.Markdown('''Here the return is:
-```
-gr.update(choices=datasets, value=key_up_data.input_value, visible=True)
-```''')
-    dataset_dropdown.key_up(fn=update_dropdown, inputs=dataset_dropdown, outputs=dataset_dropdown, queue=False, show_progress="hidden")
-    dataset_dropdown_update.key_up(fn=update_dropdown_value, inputs=dataset_dropdown_update, outputs=dataset_dropdown_update, queue=False, show_progress="hidden")
+    model_dropdown = gr.Dropdown(label="Models Auto-Complete", choices=[""], allow_custom_value=True)        
+    dataset_dropdown = gr.Dropdown(label="Datasets Auto-Complete", choices=[""], allow_custom_value=True)
+    spaces_dropdown = gr.Dropdown(label="Spaces Auto-Complete", choices=[""], allow_custom_value=True)
+    model_dropdown.key_up(fn=update_dropdown, inputs=[dataset_dropdown, gr.State("model")], outputs=dataset_dropdown, queue=False, show_progress="hidden")
+    dataset_dropdown.key_up(fn=update_dropdown, inputs=[dataset_dropdown, gr.State("dataset")], outputs=dataset_dropdown, queue=False, show_progress="hidden")
+    spaces_dropdown.key_up(fn=update_dropdown, inputs=[dataset_dropdown, gr.State("space")], outputs=dataset_dropdown, queue=False, show_progress="hidden")
 demo.launch()