Update app.py

app.py

@@ -36,16 +36,18 @@ def make_tree(url1="",url2="",url3="",url4=""):
         return None
 
 def get_images():
-    html_out="<div style='column_count:2;'>"
+    html_out="<div style='column-count:2;'>"
     get_url=f'{main_directory}images/'
     
     feed1 = requests.get(get_url)
     spl = feed1.text.split("href=")
     for line in spl:
         spl2 = line.split(">")[0].strip('""')
-        if not spl2.endswith("/"):
+        if spl2.endswith(".png|.gif|.jpg"):
             print(spl2)
             html_out+=f'<div><a href="{get_url}{spl2}">{get_url}{spl2}</a><br><img src={get_url}{spl2}></div>'
+        else:
+            print(spl2)
     html_out+="</div>"
     return html_out