Update app.py

app.py

@@ -15,16 +15,18 @@ def query1(fetch_url):
     response = requests.get(API_URL1)
     json_obj=response.json()
 
-    drop_box = []
+    config_box = []
+    split_box = []
     for obj in json_obj:
         print (obj)
-        out_config = obj[0]
-        out_split = obj[1]
+        out_config = obj['config']
+        out_split = obj['split']
         print (out_config)
         print (out_split)
-        drop_box.append(obj[0])
+        config_box.append(obj['config'])
+        split_box.append(obj['split'])
     #df = pd.DataFrame.from_dict(json_obj)        
-    return json_obj
+    return json_obj,gr.update(choices=[c for c in config_box],gr.update(choices=[s for s in split_box]))
 
 
 def query2(fetch_url,config="TEST",split="test",offset=0,length=10):
@@ -109,8 +111,8 @@ with gr.Blocks() as app:
                     data_set_url=gr.Textbox(label="Dataset (repo/name)")
                     fetch_btn=gr.Button()
                 with gr.Row():
-                    config_set=gr.Textbox(label="Config")
-                    split_set=gr.Textbox(label="Split")
+                    config_set=gr.Dropdown(label="Config")
+                    split_set=gr.Dropdown(label="Split")
                     #config_drop = gr.Dropdown(label="Config/Split", choices=[])
                     view_data=gr.Button(label="View")
                 with gr.Row():
@@ -137,5 +139,5 @@ with gr.Blocks() as app:
     
     
     #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)
+    fetch_btn.click(query1,data_set_url,[out_json,config_set,split_set])
 app.launch()