Update app.py

app.py

@@ -56,7 +56,7 @@ def find_fn(lvl_1,lvl_2,fetch_url,config="TEST",split="test",offset=0,length=10)
     #print(out_json['rows'])
     #print (inp)
     img_list=[]
-    out_json=query2(fetch_url,config,split,offset=10,length=20)
+    out_json,_=query2(fetch_url,config,split,offset=10,length=20)
     lvl_1=lvl_1.strip("[]")
 
     lvl_2=lvl_2.strip('""').strip("''")
@@ -74,9 +74,36 @@ 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(lvl_1,lvl_2,lvl_3,lvl_4,lvl_5,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)
+    box_1=[]
+    box_2=[]
+    box_3=[]
+    box_4=[]
+    box_5=[]
+    out1=None 
+    out2=None 
+    out3=None 
+    out4=None 
+    
+    if lvl_1 !="":
+        for ea in (out_json[lvl_1]):
+            box_1.append(ea)    
+        out= gr.update(choices = [m for m in box_1]
+    if lvl_2 !="":
+        for ea in (out_json[lvl_1]):
+            box_2.append(ea[lvl_2])    
+        out= gr.update(choices = [m for m in box_2]
+    if lvl_3 !="":
+        for ea in (out_json[lvl_1]):
+            box_3.append(ea[lvl_2][lvl_3])    
+        out= gr.update(choices = [m for m in box_3] 
+    if lvl_4 !="":
+        for ea in (out_json[lvl_1]):
+            box_4.append(ea[lvl_2][lvl_3][lvl_4])    
+        out= gr.update(choices = [m for m in box_4]  
+    return out
+def find_items(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)
     box_1=[]
     box_2=[]
@@ -124,10 +151,10 @@ with gr.Blocks() as app:
                     view_data=gr.Button(label="View")
                 with gr.Row():
                     lvl_1=gr.Dropdown(label="lvl_1")
-                    lvl_2=gr.Textbox(label="lvl_2")
-                    lvl_3=gr.Textbox(label="lvl_3")
-                    lvl_4=gr.Textbox(label="lvl_4")
-                    lvl_5=gr.Textbox(label="lvl_5")
+                    lvl_2=gr.Dropdown(label="lvl_2")
+                    lvl_3=gr.Dropdown(label="lvl_3")
+                    lvl_4=gr.Dropdown(label="lvl_4")
+                    lvl_5=gr.Dropdown(label="lvl_5")
                     
                     find_btn=gr.Button("Search")
                 with gr.Tab("JSON"):
@@ -139,8 +166,13 @@ with gr.Blocks() as app:
             with gr.Column(scale=1):
                 out_json = gr.JSON()
 
+    lvl_1.change(upd_drop,[lvl_1,lvl_2,lvl_3,lvl_4,lvl_5,data_set_url,config_set,split_set],[lvl_2])
+    lvl_2.change(upd_drop,[lvl_1,lvl_2,lvl_3,lvl_4,lvl_5,data_set_url,config_set,split_set],[lvl_3])
+    lvl_3.change(upd_drop,[lvl_1,lvl_2,lvl_3,lvl_4,lvl_5,data_set_url,config_set,split_set],[lvl_4])
+    lvl_4.change(upd_drop,[lvl_1,lvl_2,lvl_3,lvl_4,lvl_5,data_set_url,config_set,split_set],[lvl_5])
 
-    find_btn.click(upd_drop,[lvl_1,lvl_2,lvl_3,lvl_4,lvl_5,data_set_url,config_set,split_set],out_find)
+    
+    find_btn.click(find_items,[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])
     gal_btn.click(img_fn,out_find,out_gal)