Update app.py

app.py

@@ -59,6 +59,19 @@ def find_fn(lvl_1,lvl_2,fetch_url,config="TEST",split="test",offset=0,length=10)
     if lvl_2 == "":
         img_list.append(out_json[lvl_1])
     return img_list
+
+
+def upd_drop(inp1,fetch_url,config="TEST",split="test",offset=0,length=10):
+
+    #def upd_drop(inp1,inp2,inp3,inp4,inp5,fetch_url,config="TEST",split="test",offset=0,length=10):
+    out_json=query2(fetch_url,config,split,offset=10,length=20)
+    
+    for ea in out_json[inp1]:
+        out = [inp1]
+        
+        img_list.append(out_json[ea])    
+    return gr.update(choices=[c for c in img_list])
+
 with gr.Blocks() as app:
     with gr.Row():
         data_set_url=gr.Textbox(label="Dataset (repo/name)")
@@ -69,8 +82,11 @@ with gr.Blocks() as app:
         #config_drop = gr.Dropdown(label="Config/Split", choices=[])
         view_data=gr.Button(label="View")
     with gr.Row():
-        lvl_1=gr.Textbox("in:")
-        lvl_2=gr.Textbox("find every:")
+        lvl_1=gr.Dropdown("lvl_1")
+        lvl_2=gr.Dropdown("lvl_2")
+        lvl_3=gr.Dropdown("lvl_3")
+        lvl_4=gr.Dropdown("lvl_4")
+        lvl_5=gr.Dropdown("lvl_5")
         
         find_btn=gr.Button("Search")
     with gr.Row():
@@ -80,7 +96,12 @@ with gr.Blocks() as app:
             out_find = gr.JSON()
             out_gal = gr.Gallery()
 
+
+    lvl_1.change(upd_drop,[lvl_1,data_set_url,config_set,split_set],lvl_2)
     view_data.click(query2,[data_set_url,config_set,split_set],[out_find])
+    
+    
+    
     find_btn.click(find_fn,[lvl_1,lvl_2,data_set_url,config_set,split_set],[out_find])
     fetch_btn.click(query1,data_set_url,out_json)
 app.launch()