Update app.py

app.py

@@ -8,14 +8,37 @@ html = """
 PAGE_LINK
 </div>
 """
+css="""
+.img_box{
+    display: flex;
+    flex-direction: column;
+    flex-wrap: wrap;
+    height: 20000px;
+}
+.img_class{
+background: #ffffff;
+    max-width: 48%;
+    font-family: monospace;
+    border-top: #9300ff;
+    border-style: inset;
+    margin-top: 5px;
+}
+"""
 def load_json(url1="",url2="",url3="",url4=""):
     get_url=f'{main_directory}{url1}{url2}{url3}{url4}'
     print(f'{get_url}')
-    get_url=get_url.split(".json")[0]+".json"
+    get_url=get_url.split("None")[0]
+    
     if get_url.endswith('.json'):
         feed1 = requests.get(get_url)
-        return feed1.text
-    return None
+        return None, feed1.text
+    elif get_url.endswith(".png") or get_url.endswith(".gif") or get_url.endswith(".jpg"):
+        html_out=f"<style>{css}</style><div>"
+        html_out+=f'<div class="img_class"><a href="{get_url}" target="_blank">{get_url}</a><br><img src={get_url}></div></div>'
+        
+        return html_out, None
+    return None,None
+    
 def make_tree(url1="",url2="",url3="",url4=""):
     link_box=[]
     html_out=""
@@ -34,22 +57,7 @@ def make_tree(url1="",url2="",url3="",url4=""):
         return gr.update(choices=[l for l in link_box],interactive=True)
     else:
         return None
-css="""
-.img_box{
-    display: flex;
-    flex-direction: column;
-    flex-wrap: wrap;
-    height: 20000px;
-}
-.img_class{
-background: #ffffff;
-    max-width: 48%;
-    font-family: monospace;
-    border-top: #9300ff;
-    border-style: inset;
-    margin-top: 5px;
-}
-"""
+
 def get_images():
     html_out=f"<style>{css}</style><div class='img_box'>"
     get_url=f'{main_directory}images/'
@@ -89,7 +97,7 @@ with gr.Blocks() as app:
     
         
     ####### Raw ############
-    load_btn.click(load_json,[drop1,drop2,drop3,drop4],links)
+    load_btn.click(load_json,[drop1,drop2,drop3,drop4],[html_im,links])
     drop1.change(make_tree,drop1,[drop2])
     drop2.change(make_tree,[drop1,drop2],[drop3])
     drop3.change(make_tree,[drop1,drop2,drop3],[drop4])