Update app.py

app.py

@@ -85,21 +85,25 @@ def upd_drop(lvl_1,lvl_2,lvl_3,lvl_4,lvl_5,fetch_url,config="TEST",split="test",
     out2=None 
     out3=None 
     out4=None 
-    
+    if lvl_1 == "rows":
+        lvl_src= out_json[lvl_1][0]
+    else:
+        lvl_src=out_json[lvl_1]
+        
     if lvl_1 !="" and lvl_1 != None:
-        for ea in (out_json[lvl_1]):
+        for ea in (lvl_src):
             box_1.append(ea)    
         out= gr.update(choices = [m for m in box_1])
     if lvl_2 !="" and lvl_2 != None:
-        for ea in (out_json[lvl_1]):
+        for ea in (lvl_src):
             box_2.append(ea[lvl_2])    
         out= gr.update(choices = [m for m in box_2])
     if lvl_3 !="" and lvl_3 != None:
-        for ea in (out_json[lvl_1]):
+        for ea in (lvl_src):
             box_3.append(ea[lvl_2][lvl_3])    
         out= gr.update(choices = [m for m in box_3])
     if lvl_4 !="" and lvl_4 != None:
-        for ea in (out_json[lvl_1]):
+        for ea in (lvl_src):
             box_4.append(ea[lvl_2][lvl_3][lvl_4])    
         out= gr.update(choices = [m for m in box_4])  
     return out