Update app.py

app.py

@@ -35,10 +35,13 @@ def query2(fetch_url,config="TEST",split="test",offset=0,length=10):
     API_URL2 = f"https://datasets-server.huggingface.co/rows?dataset={fetch_url}&config={config}&split={split}&offset={offset}&length={length}"
     response = requests.get(API_URL2)
     dictionary=response.json()
-    print (len(dictionary))
+    #print (len(dictionary))
+    lvl_1=[]
+    
     for key in dictionary:
+        lvl_1.append(key)
         print (key)
-    return dictionary
+    return dictionary,gr.update(choices=[l for l in lvl_1])
 
 def find_fn(lvl_1,lvl_2,fetch_url,config="TEST",split="test",offset=0,length=10):
     #print(out_json['rows'])
@@ -64,16 +67,14 @@ def find_fn(lvl_1,lvl_2,fetch_url,config="TEST",split="test",offset=0,length=10)
     return img_list
 
 
-def upd_drop(inp1,fetch_url,config="TEST",split="test",offset=0,length=10):
+def upd_drop(lvl_1,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])
+    lvl_2=[]
+    for ea in out_json[lvl_1]:
+        lvl_2.append(out_json[lvl_1][ea])    
+    return gr.update(choices=[l for l in lvl_2])
 
 with gr.Blocks() as app:
     with gr.Row():
@@ -101,7 +102,7 @@ with gr.Blocks() as app:
 
 
     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])
+    view_data.click(query2,[data_set_url,config_set,split_set],[out_find,lvl_1])