Update app.py

app.py

@@ -37,7 +37,7 @@ def query2(fetch_url,config="TEST",split="test",offset=0,length=10):
     dictionary=response.json()
     return dictionary
 
-def find_fn(inp,fetch_url,config="TEST",split="test",offset=0,length=10):
+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=[]
@@ -45,10 +45,12 @@ def find_fn(inp,fetch_url,config="TEST",split="test",offset=0,length=10):
     
     inp=inp.strip("[]")
     print(inp)
-    for ea in out_json:
-        #img_ea = ea['row']['image']['src']
-        img_list.append(ea[f"{inp}"])
-    
+    if lvl_2 != "":
+        for ea in out_json[f"{lvl_1}"]:
+            #img_ea = ea['row']['image']['src']
+            img_list.append(ea[f"{lvl_2}"])
+    if lvl_2 == "":
+        img_list.append(out_json[lvl_1])
     return img_list
 with gr.Blocks() as app:
     with gr.Row():
@@ -60,7 +62,9 @@ with gr.Blocks() as app:
         #config_drop = gr.Dropdown(label="Config/Split", choices=[])
         view_data=gr.Button(label="View")
     with gr.Row():
-        find_string=gr.Textbox()
+        lvl_1=gr.Textbox("in:")
+        lvl_2=gr.Textbox("find every:")
+        
         find_btn=gr.Button("Search")
     with gr.Row():
         with gr.Column(scale=1):
@@ -70,6 +74,6 @@ with gr.Blocks() as app:
             out_gal = gr.Gallery()
 
     view_data.click(query2,[data_set_url,config_set,split_set],[out_find])
-    find_btn.click(find_fn,[find_string,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()