Update app.py

app.py

@@ -67,14 +67,44 @@ def find_fn(lvl_1,lvl_2,fetch_url,config="TEST",split="test",offset=0,length=10)
     return img_list
 
 
-def upd_drop(lvl_1,fetch_url,config="TEST",split="test",offset=0,length=10):
+def upd_drop(lvl_1,lvl_2,lvl_3,lvl_4,lvl_5,fetch_url,config="TEST",split="test",offset=0,length=10):
     out_json,_=query2(fetch_url,config,split,offset=10,length=20)
-    lvl_2=[]
-    for ea in (out_json[lvl_1]):
-        #print (i)
-        print (ea)
-        lvl_2.append(ea)    
-    return gr.update(choices=[l for l in lvl_2])
+    box_1=[]
+    box_2=[]
+    box_3=[]
+    box_4=[]
+    box_5=[]
+    if lvl_1 !="":
+        for ea in (out_json[lvl_1]):
+            #print (i)
+            print (ea)
+            box_1.append(ea)    
+        out = box_1
+    if lvl_2 !="":
+        for ea in (out_json[lvl_1][lvl_2]):
+            #print (i)
+            print (ea)
+            box_2.append(ea)    
+        out = box_2
+    if lvl_3 !="":
+        for ea in (out_json[lvl_1][lvl_2][lvl_3]):
+            #print (i)
+            print (ea)
+            box_3.append(ea)    
+        out = box_3  
+    if lvl_4 !="":
+        for ea in (out_json[lvl_1][lvl_2][lvl_3][lvl_4]):
+            #print (i)
+            print (ea)
+            box_4.append(ea)    
+        out = box_4  
+    if lvl_5 !="":
+        for ea in (out_json[lvl_1][lvl_2][lvl_3][lvl_4][lvl_5]):
+            #print (i)
+            print (ea)
+            box_5.append(ea)    
+        out = box_5          
+    return out
 
 with gr.Blocks() as app:
     with gr.Row():
@@ -86,11 +116,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.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")
+        lvl_1=gr.Textbox("lvl_1")
+        lvl_2=gr.Textbox("lvl_2")
+        lvl_3=gr.Textbox("lvl_3")
+        lvl_4=gr.Textbox("lvl_4")
+        lvl_5=gr.Textbox("lvl_5")
         
         find_btn=gr.Button("Search")
     with gr.Row():
@@ -101,11 +131,11 @@ with gr.Blocks() as app:
             out_gal = gr.Gallery()
 
 
-    lvl_1.change(upd_drop,[lvl_1,data_set_url,config_set,split_set],lvl_2)
+    find_btn.click(upd_drop,[lvl_1,lvl_2,lvl_3,lvl_4,lvl_5,data_set_url,config_set,split_set],out_find)
     view_data.click(query2,[data_set_url,config_set,split_set],[out_find,lvl_1])
     
     
     
-    find_btn.click(find_fn,[lvl_1,lvl_2,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()