Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import requests
 import json
+from PIL import Image
 main_directory = "https://services.swpc.noaa.gov/"
 
 html = """
@@ -72,8 +73,7 @@ def make_tree(url1="",url2="",url3="",url4=""):
 
 def get_images():
     html_out=f"<style>{css}</style><div class='img_box'>"
-    get_url=f'{main_directory}images/'
-    
+    get_url=f'{main_directory}images/geospace/'
     feed1 = requests.get(get_url)
     spl = feed1.text.split("href=")
     for line in spl:
@@ -83,19 +83,52 @@ def get_images():
             html_out+=f'<div class="img_class"><a href="{get_url}{spl2}" target="_blank">{get_url}{spl2}</a><br><img src={get_url}{spl2}></div>'
         else:
             print(spl2)
+    get_url2=f'{main_directory}images/'
+    feed2 = requests.get(get_url2)
+    spl = feed2.text.split("href=")
+    for line2 in spl:
+        spl2 = line2.split(">")[0].strip('""')
+        if spl2.endswith(".png") or spl2.endswith(".gif") or spl2.endswith(".jpg"):
+            print(spl2)
+            html_out+=f'<div class="img_class"><a href="{get_url2}{spl2}" target="_blank">{get_url2}{spl2}</a><br><img src={get_url2}{spl2}></div>'
+        else:
+            print(spl2)            
     html_out+="</div>"
     return html_out
-    
+
+def make_animation():
+    html_out=f"<style>{css}</style><div class='img_box'>"
+    get_url=f'{main_directory}images/animations/'
+    feed1 = requests.get(get_url)
+    spl = feed1.text.split("href=")
+    for line in spl:
+        spl2 = line.split(">")[0].strip('""')
+        if spl2.endswith("/"):
+            feed2 = requests.get(get_url)
+            spl3 = feed2.text.split("href=")
+            for line2 in spl3:
+                spl3 = line2.split(">")[0].strip('""')
+                if spl3.endswith(".png") or spl2.endswith(".gif") or spl2.endswith(".jpg"):
+                    gif_box.append(f'{get_url}{spl2}{spl3}')
+    '''                    
+    frames = [Image.open(image) for image in glob.glob(f"{frame_folder}/*.JPG")]
+    frame_one = frames[0]
+    frame_one.save("my_awesome.gif", format="GIF", append_images=frames,
+               save_all=True, duration=100, loop=0)
+    '''
+    return gif_box
+
 def run():
     out=make_tree()
     im_html=get_images()
     return out, im_html
 
-
-    
 with gr.Blocks() as app:
     with gr.Tab("Images"):
         html_im=gr.HTML()
+    with gr.Tab("Animations"):
+        anim_btn=gr.Button()
+        anim_out=gr.JSON()        
     with gr.Tab("Raw"):
         with gr.Row():
             drop1=gr.Dropdown()
@@ -108,9 +141,10 @@ with gr.Blocks() as app:
 
     ###### Images ##########
     
+    load_btn.click(load_json,[drop1,drop2,drop3,drop4],[html_raw,links])
+    anim_btn.click(make_animatin,None,[anim_out])
         
     ####### Raw ############
-    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])