Update app.py

app.py

@@ -8,25 +8,26 @@ html = """
 PAGE_LINK
 </div>
 """
-
+def load_json(url1="",url2="",url3="",url4=""):
+    get_url=f'{main_directory}{url1}{url2}{url3}{url4}'
+    feed1 = requests.get(get_url)
+    if get_url.endswith(".json"):
+        return feed1.text
 
 def make_tree(url1="",url2="",url3="",url4=""):
     link_box=[]
     html_out=""
     get_url=f'{main_directory}{url1}{url2}{url3}{url4}'
     feed1 = requests.get(get_url)
-    if get_url.endswith(".json"):
-        return feed1.text, None
-    else:
-        spl = feed1.text.split("href=")
-        for line in spl:
-            spl2 = line.split(">")[0]
-            print(spl2)
-            if spl2.endswith('/"') or spl2.endswith('.json"'):
-                fin=line.split(">")[0].strip('""')
-                link_box.append(fin)
-                #html_out=html_out+html.replace("PAGE_LINK",fin)
-        return None, gr.update(choices=[l for l in link_box],interactive=True)
+    spl = feed1.text.split("href=")
+    for line in spl:
+        spl2 = line.split(">")[0]
+        print(spl2)
+        if spl2.endswith('/"') or spl2.endswith('.json"'):
+            fin=line.split(">")[0].strip('""')
+            link_box.append(fin)
+            #html_out=html_out+html.replace("PAGE_LINK",fin)
+    return gr.update(choices=[l for l in link_box],interactive=True)
 
 def run():
     _,out=make_tree()
@@ -38,10 +39,12 @@ with gr.Blocks() as app:
         drop2=gr.Dropdown()
         drop3=gr.Dropdown()
         drop4=gr.Dropdown()
+    load_btn=gr.Button("Load JSON")
     links=gr.JSON()
     html_=gr.HTML()
-    drop1.change(make_tree,drop1,[links,drop2])
-    drop2.change(make_tree,[drop1,drop2],[links,drop3])
-    drop3.change(make_tree,[drop1,drop2,drop3],[links,drop4])
+    load_btn.click(load_json,[drop1,drop2,drop3,drop4],links)
+    drop1.change(make_tree,drop1,[drop2])
+    drop2.change(make_tree,[drop1,drop2],[drop3])
+    drop3.change(make_tree,[drop1,drop2,drop3],[drop4])
     app.load(run,None,[drop1])
 app.launch()