Update app.py

app.py

@@ -95,7 +95,12 @@ def upd_drop(lvl_1,lvl_2,lvl_3,lvl_4,lvl_5,fetch_url,config="TEST",split="test",
             box_5.append(ea[lvl_2][lvl_3][lvl_4][lvl_5])    
         out = box_5          
     return out
-
+def img_fn(inp):
+    img_box=[]
+    for img in inp:
+        out = Image.open(inp)
+        img_box.append(out)
+    return img_box
 with gr.Blocks() as app:
     with gr.Box():
         with gr.Row():
@@ -109,15 +114,18 @@ 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.Textbox(label="lvl_1")
+                    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")
                     
                     find_btn=gr.Button("Search")
-                out_find = gr.JSON()
-                out_gal = gr.Gallery()
+                with gr.Tab("JSON"):
+                    out_find = gr.JSON()
+                with gr.Tab("Gallery"):
+                    gal_btn=gr.Button("Load Images")
+                    out_gal = gr.Gallery()
 
             with gr.Column(scale=1):
                 out_json = gr.JSON()
@@ -125,7 +133,7 @@ with gr.Blocks() as app:
 
     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])
-    
+    gal_btn.click(img_fn,out_find,out_gal)
     
     
     #find_btn.click(find_fn,[lvl_1,lvl_2,data_set_url,config_set,split_set],[out_find])