Update app.py

app.py

@@ -13,19 +13,23 @@ PAGE_LINK
 def make_tree(url1="",url2="",url3="",url4=""):
     link_box=[]
     html_out=""
-    feed1 = requests.get(f'{main_directory}{url1}{url2}{url3}{url4}')
-    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)
+    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 out_json, gr.update(choices=[l for l in link_box],interactive=True)
 
 def run():
-    out=make_tree()
+    _,out=make_tree()
     return out
     
 with gr.Blocks() as app:
@@ -36,8 +40,8 @@ with gr.Blocks() as app:
         drop4=gr.Dropdown()
     links=gr.JSON()
     html_=gr.HTML()
-    drop1.change(make_tree,drop1,drop2)
-    drop2.change(make_tree,[drop1,drop2],drop3)
-    drop3.change(make_tree,[drop1,drop2,drop3],drop4)
+    drop1.change(make_tree,drop1,[links,drop2])
+    drop2.change(make_tree,[drop1,drop2],[links,drop3])
+    drop3.change(make_tree,[drop1,drop2,drop3],[links,drop4])
     app.load(run,None,[drop1])
 app.launch()