Update app.py

app.py

@@ -50,7 +50,7 @@ def make_tree(url1="",url2="",url3="",url4=""):
         for line in spl:
             spl2 = line.split(">")[0]
             print(spl2)
-            if spl2.endswith('/"') or spl2.endswith('.json"'):
+            if spl2.endswith('/"') or spl2.endswith('.json"') or spl2.endswith('.png"') or spl2.endswith('.gif"') or spl2.endswith('.jpg"'):
                 fin=line.split(">")[0].strip('""')
                 link_box.append(fin)
                 #html_out=html_out+html.replace("PAGE_LINK",fin)
@@ -91,13 +91,14 @@ with gr.Blocks() as app:
             drop3=gr.Dropdown()
             drop4=gr.Dropdown()
         load_btn=gr.Button("Load JSON")
+        html_raw=gr.HTML()
         links=gr.JSON()
 
     ###### Images ##########
     
         
     ####### Raw ############
-    load_btn.click(load_json,[drop1,drop2,drop3,drop4],[html_im,links])
+    load_btn.click(load_json,[drop1,drop2,drop3,drop4],[html_raw,links])
     drop1.change(make_tree,drop1,[drop2])
     drop2.change(make_tree,[drop1,drop2],[drop3])
     drop3.change(make_tree,[drop1,drop2,drop3],[drop4])