Update app.py

app.py

@@ -14,10 +14,17 @@ def query1(fetch_url):
     API_URL1 = f"https://datasets-server.huggingface.co/splits?dataset={fetch_url}"
     response = requests.get(API_URL1)
     json_obj=response.json()
+
+    drop_box = []
     for obj in json_obj:
+        out_config = obj["splits"]["config"]
+        out_split = obj["splits"]["split"]
         print (obj)
-    df = pd.DataFrame.from_dict(json_obj)        
-    return df
+        print (out_config)
+        print (out_split)
+        drop_box.append(obj["splits"])
+    #df = pd.DataFrame.from_dict(json_obj)        
+    return gr.update(choices=[d for d in drop_box])
 
 
 def query2(fetch_url,config="TEST",split="test",offset=0,length=10):
@@ -45,8 +52,9 @@ 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.Textbox(label="Config")
+        #split_set=gr.Textbox(label="Split")
+        config_drop = gr.Dropdown(label="Config/Split", choices=[])
         view_data=gr.Button(label="View")
     with gr.Row():
         find_string=gr.Textbox()
@@ -57,7 +65,7 @@ with gr.Blocks() as app:
             out_find = gr.JSON()
             out_gal = gr.Gallery()
 
-    view_data.click(query2,[data_set_url,config_set,split_set],[out_find])
+    view_data.click(query2,[data_set_url,config],[out_find])
     find_btn.click(find_fn,[find_string,out_json],[out_find,out_gal])
     fetch_btn.click(query1,data_set_url,out_json)
 app.launch()