Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| import requests | |
| import json | |
| import os | |
| from PIL import Image | |
| import moviepy.video.io.ImageSequenceClip as ic | |
| from pathlib import Path | |
| import bs4 | |
| import datetime | |
| import urllib.request | |
| import uuid | |
| main_directory = "https://services.swpc.noaa.gov/" | |
| sdo_source = "https://sdo.gsfc.nasa.gov/assets/img/browse/" | |
| sdo_source_format = "https://sdo.gsfc.nasa.gov/assets/img/browse/YEAR/MONTH/DAY/DATE_IDENT_SIZE_TOOL.jpg" | |
| sdo_aia_latest=[ | |
| {"name":"Fe18 94A", "source":"https://umbra.nascom.nasa.gov/images/latest_aia_94.gif"}, | |
| {"name":"Fe20 131A", "source":"https://umbra.nascom.nasa.gov/images/latest_aia_131.gif"}, | |
| {"name":"Fe9/10 171A", "source":"https://umbra.nascom.nasa.gov/images/latest_aia_171.gif"}, | |
| {"name":"Fe12 193A", "source":"https://umbra.nascom.nasa.gov/images/latest_aia_193.gif"}, | |
| {"name":"Fe14 211A", "source":"https://umbra.nascom.nasa.gov/images/latest_aia_211.gif"}, | |
| {"name":"He2 304A", "source":"https://umbra.nascom.nasa.gov/images/latest_aia_304.gif"}, | |
| {"name":"Fe16 335A", "source":"https://umbra.nascom.nasa.gov/images/latest_aia_335.gif"}, | |
| {"name":"cont+C4 1600A", "source":"https://umbra.nascom.nasa.gov/images/latest_aia_1600.gif"}, | |
| {"name":"continuum 1700A", "source":"https://umbra.nascom.nasa.gov/images/latest_aia_1700.gif"}, | |
| {"name":"continuum 4500A", "source":"https://umbra.nascom.nasa.gov/images/latest_aia_4500.gif"}, | |
| ] | |
| nasa_images=[ | |
| {"name":"EIT 171A", "source":"https://soho.nascom.nasa.gov/data/realtime/eit_171/512/latest.jpg"}, | |
| {"name":"EIT 195A", "source":"https://soho.nascom.nasa.gov/data/realtime/eit_195/512/latest.jpg"}, | |
| {"name":"EIT 284A", "source":"https://soho.nascom.nasa.gov/data/realtime/eit_284/512/latest.jpg"}, | |
| {"name":"EIT 304A", "source":"https://soho.nascom.nasa.gov/data/realtime/eit_304/512/latest.jpg"}, | |
| ] | |
| nasa_soho_gifs=[ | |
| {"name":"EIT 304A", "source":"https://soho.nascom.nasa.gov/data/LATEST/current_eit_304small.gif"}, | |
| {"name":"EIT 195A", "source":"https://soho.nascom.nasa.gov/data/LATEST/current_eit_195small.gif"}, | |
| {"name":"EIT 171A", "source":"https://soho.nascom.nasa.gov/data/LATEST/current_eit_171small.gif"}, | |
| {"name":"EIT 284A", "source":"https://soho.nascom.nasa.gov/data/LATEST/current_eit_284small.gif"}, | |
| ] | |
| nasa_soho_mp4=[ | |
| {"name":"EIT 171A", "source":"https://soho.nascom.nasa.gov/data/LATEST/current_eit_171small.mp4"}, | |
| {"name":"EIT 195A", "source":"https://soho.nascom.nasa.gov/data/LATEST/current_eit_195small.mp4"}, | |
| {"name":"EIT 284A", "source":"https://soho.nascom.nasa.gov/data/LATEST/current_eit_284small.mp4"}, | |
| {"name":"EIT 304A", "source":"https://soho.nascom.nasa.gov/data/LATEST/current_eit_304small.mp4"}, | |
| {"name":"LASCO C2", "source":"https://soho.nascom.nasa.gov/data/LATEST/current_c2small.mp4"}, | |
| {"name":"LASCO C3", "source":"https://soho.nascom.nasa.gov/data/LATEST/current_c3small.mp4"}, | |
| {"name":"SDO/HMI Continuum", "source":"https://soho.nascom.nasa.gov/data/LATEST/current_hmi_igr-512.mp4"}, | |
| {"name":"SDO/HMI Magnetogram", "source":"https://soho.nascom.nasa.gov/data/LATEST/current_hmi_mag-512.mp4"}, | |
| ] | |
| sdo_obj = ["HMIB","HMIBC","HMIIC","HMIIF","HMID","HMII","HMI171", | |
| "0094","0131","0171","0193","0211", | |
| "0304","0335","1600","1700","4500", | |
| "211193171","211193171n","211193171rg", | |
| "094335193","304211171"] | |
| sdo_size= [256,512,1024,2048,4096] | |
| html = """ | |
| <div> | |
| PAGE_LINK | |
| </div> | |
| """ | |
| css=""" | |
| .img_box{ | |
| display: flex; | |
| flex-direction: column; | |
| flex-flow: unset; | |
| flex-wrap: wrap; | |
| justify-content: space-around; | |
| } | |
| .img_class{ | |
| background: #ffffff; | |
| max-width: 48%; | |
| font-family: monospace; | |
| border-top: #9300ff; | |
| border-style: inset; | |
| margin-top: 5px; | |
| } | |
| .img_class_raw{ | |
| background: #ffffff; | |
| width: 100%; | |
| font-family: monospace; | |
| border-top: #9300ff; | |
| border-style: inset; | |
| margin-top: 5px; | |
| display:flex; | |
| flex-direction:column; | |
| } | |
| .img_box_soho{ | |
| display: flex; | |
| flex-direction: row; | |
| flex-wrap: wrap; | |
| justify-content: space-between; | |
| } | |
| .img_class_soho{ | |
| background: #ffffff; | |
| font-family: monospace; | |
| border-top: #9300ff; | |
| border-style: inset; | |
| margin-top: 5px; | |
| max-width: 48%; | |
| } | |
| .img_class_sdo{ | |
| background: #ffffff; | |
| font-family: monospace; | |
| border-top: #9300ff; | |
| border-style: inset; | |
| margin-top: 5px; | |
| max-width: 25%; | |
| font-size: small; | |
| } | |
| """ | |
| 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("None")[0] | |
| print(f'{get_url}') | |
| get_url=get_url.split("[]")[0] | |
| print(f'{get_url}') | |
| if get_url.endswith('.json'): | |
| feed1 = requests.get(get_url) | |
| 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_raw"><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="" | |
| get_url=f'{main_directory}{url1}{url2}{url3}{url4}' | |
| print(f'######### :: {get_url}') | |
| if not get_url.endswith('.json'): | |
| feed1 = requests.get(get_url) | |
| spl = feed1.text.split("href=") | |
| for line in spl: | |
| spl2 = line.split(">")[0] | |
| print(spl2) | |
| 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) | |
| return gr.update(choices=[l for l in link_box],interactive=True) | |
| else: | |
| return None | |
| def get_images(): | |
| html_out=f"<style>{css}</style><div class='img_box'>" | |
| get_url=f'{main_directory}images/geospace/' | |
| feed1 = requests.get(get_url) | |
| spl = feed1.text.split("href=") | |
| for line in spl: | |
| spl2 = line.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_url}{spl2}" target="_blank">{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">{spl2}</a><br><img src={get_url2}{spl2}></div>' | |
| else: | |
| print(spl2) | |
| html_out+="</div>" | |
| return html_out | |
| def make_animation(): | |
| get_url=f'{main_directory}images/animations/' | |
| feed1 = requests.get(get_url) | |
| spl = feed1.text.split("href=") | |
| files_out=[] | |
| for line in spl: | |
| gif_box=[] | |
| spl2 = line.split(">")[0].strip('""') | |
| if spl2.endswith("/"): | |
| feed2 = requests.get(f'{get_url}{spl2}') | |
| spl3 = feed2.text.split("href=") | |
| for line2 in spl3: | |
| spl3 = line2.split(">")[0].strip('""') | |
| print("$$$$$$$$$$$$$$") | |
| print(spl3) | |
| if spl3.endswith(".png") or spl3.endswith(".gif") or spl3.endswith(".jpg"): | |
| gif_box.append(f'{get_url}{spl2}{spl3}') | |
| if gif_box: | |
| try: | |
| #frames = [Image.open(image) for image in glob.glob(f"{frame_folder}/*.JPG")] | |
| frames = [] | |
| for i,ea in enumerate(gif_box): | |
| print("############") | |
| print(ea) | |
| urllib.request.urlretrieve(ea,f'tmp{i}.png') | |
| frames.append(f'tmp{i}.png') | |
| fps=60 | |
| uid=uuid.uuid4() | |
| clip = ic.ImageSequenceClip(frames, fps = fps) | |
| clip.to_gif(f"{uid}.gif",fps=fps) | |
| files_out.append(f'{uid}.gif') | |
| except Exception as e: | |
| print(e) | |
| return files_out | |
| def make_html(inp_files): | |
| html_out=f"<style>{css}</style><div class='img_box'>" | |
| for ea in inp_files: | |
| print(ea) | |
| html_out+=f'<div class="img_class"><img src="file={ea}"></div>' | |
| html_out+='</div>' | |
| return html_out | |
| def make_nasa_soho_videos(): | |
| html_out=f"<style>{css}</style><div class='img_box_soho'>" | |
| for ea in nasa_soho_mp4: | |
| file_name=ea['source'] | |
| html_out+=f''' | |
| <video width="50%" height="320" controls> | |
| <source src="{file_name}" type="video/mp4"> | |
| Your browser does not support the video tag. | |
| </video>''' | |
| #html_out+=f'<div class="img_class_soho"><img src="file={file_name}"></div>' | |
| html_out+='</div>' | |
| return html_out | |
| def make_nasa_soho_images(inp_src: list): | |
| html_out="" | |
| #html_out=f"<style>{css}</style><div class='img_box_soho'>" | |
| for ea in inp_src: | |
| file_name=ea['source'] | |
| html_out+=f'<div class="img_class_soho"><img src="{file_name}"></div>' | |
| #html_out+='</div>' | |
| #html_out+=html_in | |
| return html_out | |
| def make_nasa_soho_image_trigger(): | |
| html_in=f"<style>{css}</style><div class='img_box_soho'>" | |
| html_in+= make_nasa_soho_images(nasa_images) | |
| html_in+= make_nasa_soho_images(sdo_aia_latest) | |
| #html_in+= make_nasa_soho_videos() | |
| html_in+="</div>" | |
| return html_in | |
| def nasa_sdo_images(obj,size,date1,date2): | |
| html_in=f"<style>{css}</style><div class='img_box_soho'>" | |
| if int(date2[6:8])-int(date1[6:8]) >0: | |
| return ("<div style='font-size:xxx-large'>Reduce Timespan to 24 hours</div>") | |
| in_year=f"{date2[0:4]}" | |
| in_month=f"{date2[4:6]}" | |
| in_day=f"{date2[6:8]}" | |
| file_name="DATE_TIME_SIZE_TOOL.jpg" | |
| sdo_source_format = f"https://sdo.gsfc.nasa.gov/assets/img/browse/{in_year}/{in_month}/{in_day}/" | |
| get_url=sdo_source_format | |
| print(get_url) | |
| feed1 = requests.get(get_url) | |
| if "All" in size: | |
| size = sdo_size | |
| if "All" in obj: | |
| obj = sdo_obj | |
| link_box=[] | |
| # Parse the HTML content using BeautifulSoup | |
| soup = bs4.BeautifulSoup(feed1.content, 'html.parser') | |
| #print(soup) | |
| anchor_elements = soup.find_all('a') | |
| for element in anchor_elements: | |
| href=element.get('href') | |
| if href.endswith('.jpg'): | |
| for o in obj: | |
| for s in size: | |
| ls=href.split("_") | |
| print(ls[1][0:4]) | |
| print(date1[8:13]) | |
| if ls[1][0:4]>=date1[8:13]: | |
| src_obj=ls[3].split('.jpg')[0] | |
| if src_obj == o: | |
| if int(ls[2]) == int(s): | |
| html_in+=f'<div class="img_class_sdo"><a href="{sdo_source_format}{href}" target="_blank">{href}</a><br><img src={sdo_source_format}{href}></div>' | |
| html_in+="</div>" | |
| return html_in | |
| def nasa_sdo_composite(obj,size,date1,date2): | |
| html_in=f"<style>{css}</style><div class='img_box_soho'>" | |
| if int(date2[6:8])-int(date1[6:8]) >0: | |
| return ("<div style='font-size:xxx-large'>Reduce Timespan to 24 hours</div>") | |
| in_year=f"{date2[0:4]}" | |
| in_month=f"{date2[4:6]}" | |
| in_day=f"{date2[6:8]}" | |
| file_name="DATE_TIME_SIZE_TOOL.jpg" | |
| sdo_source_format = f"https://sdo.gsfc.nasa.gov/assets/img/browse/{in_year}/{in_month}/{in_day}/" | |
| get_url=sdo_source_format | |
| print(get_url) | |
| feed1 = requests.get(get_url) | |
| if "All" in size: | |
| size = sdo_size | |
| if "All" in obj: | |
| obj = sdo_obj | |
| link_box=[] | |
| soup = bs4.BeautifulSoup(feed1.content, 'html.parser') | |
| anchor_elements = soup.find_all('a') | |
| anchor_elements.sort() | |
| cnt=0 | |
| max_cnt=4 | |
| for element in anchor_elements: | |
| if cnt<=max_cnt: | |
| href=element.get('href') | |
| if href.endswith('.jpg'): | |
| for o in obj: | |
| for s in size: | |
| ls=href.split("_") | |
| print(ls[1][0:4]) | |
| print(date1[8:13]) | |
| if ls[1][0:4]>=date1[8:13]: | |
| src_obj=ls[3].split('.jpg')[0] | |
| if src_obj == o: | |
| if int(ls[2]) == int(s): | |
| html_in+=f'<div class="img_class_sdo"><a href="{sdo_source_format}{href}" target="_blank">{href}</a><br><img src={sdo_source_format}{href}></div>' | |
| html_in+="</div>" | |
| return html_in | |
| def run(): | |
| out=make_tree() | |
| im_html=get_images() | |
| return out, im_html | |
| def get_date(year1,month1,day1,hour1,minute1,year2,month2,day2,hour2,minute2): | |
| if len(str(month1))<2: m1=0 | |
| else: m1="" | |
| if len(str(day1))<2: d1=0 | |
| else: d1="" | |
| if len(str(hour1))<2: h1=0 | |
| else: h1="" | |
| if len(str(minute1))<2: mi1=0 | |
| else: mi1="" | |
| date1=f'{year1}{m1}{month1}{d1}{day1}{h1}{hour1}{mi1}{minute1}' | |
| if len(str(month2))<2: m2=0 | |
| else: m2="" | |
| if len(str(day2))<2: d2=0 | |
| else: d2="" | |
| if len(str(hour2))<2: h2=0 | |
| else: h2="" | |
| if len(str(minute2))<2: mi2=0 | |
| else: mi2="" | |
| date2=f'{year2}{m2}{month2}{d2}{day2}{h2}{hour2}{mi2}{minute2}' | |
| print(date1) | |
| print(date2) | |
| return date1,date2 | |
| with gr.Blocks() as app: | |
| datetimestamp=str(datetime.datetime.now()).split(".")[0] | |
| datestamp=datetimestamp.replace(" ","-").split("-") | |
| timestamp=datestamp[3].split(":") | |
| gr.HTML(f"""<div style='font-size:xxx-large;'>Current: {datetimestamp} UTC</div>""") | |
| with gr.Tab("NOAA"): | |
| with gr.Tab("Images"): | |
| html_im=gr.HTML() | |
| with gr.Tab("Animations"): | |
| anim_btn=gr.Button() | |
| anim_out=gr.HTML() | |
| file_out=gr.Files() | |
| with gr.Tab("Raw"): | |
| with gr.Row(): | |
| drop1=gr.Dropdown() | |
| drop2=gr.Dropdown() | |
| drop3=gr.Dropdown() | |
| drop4=gr.Dropdown() | |
| load_btn=gr.Button("Load") | |
| html_raw=gr.HTML() | |
| links=gr.JSON() | |
| with gr.Tab("NASA"): | |
| with gr.Tab("SDO Images"): | |
| with gr.Row(): | |
| gr.Markdown("<h5 style='font-size:xx-large;'>From:") | |
| year1=gr.Number(label="Year",minimum=2000,maximum=2024,precision=0,value=int(datestamp[0]),min_width=10) | |
| month1=gr.Number(label="Month",minimum=1,maximum=12,precision=0,value=int(datestamp[1]),min_width=10) | |
| day1=gr.Number(label="Day",minimum=1,maximum=31,precision=0,value=int(datestamp[2]),min_width=10) | |
| hour1=gr.Number(label="Hour",minimum=0,maximum=23,precision=0,value=int(timestamp[0]),min_width=10) | |
| minute1=gr.Number(label="Minute",minimum=0,maximum=59,precision=0,value=int(timestamp[1]),min_width=10) | |
| gr.Markdown("<h5 style='font-size:xx-large;'>To:") | |
| year2=gr.Number(label="Year",minimum=2000,maximum=2024,precision=0,value=int(datestamp[0]),min_width=10) | |
| month2=gr.Number(label="Month",minimum=1,maximum=12,precision=0,value=int(datestamp[1]),min_width=10) | |
| day2=gr.Number(label="Day",minimum=1,maximum=31,precision=0,value=int(datestamp[2]),min_width=10) | |
| hour2=gr.Number(label="Hour",minimum=0,maximum=23,precision=0,value=int(timestamp[0]),min_width=10) | |
| minute2=gr.Number(label="Minute",minimum=0,maximum=59,precision=0,value=int(timestamp[1]),min_width=10) | |
| with gr.Row(): | |
| sdo_drop1=gr.Dropdown(label="Instrument", multiselect=True, choices=["All"]+sdo_obj, value="0094") | |
| sdo_drop2=gr.Dropdown(label="Size", multiselect=True, choices=["All"]+sdo_size,value=512) | |
| with gr.Tab("Composite"): | |
| sdo_comp_btn=gr.Button("Load") | |
| sdo_comp_html=gr.HTML() | |
| with gr.Tab("Images"): | |
| sdo_btn=gr.Button("Load") | |
| sdo_html=gr.HTML() | |
| date1=gr.Textbox(visible=False) | |
| date2=gr.Textbox(visible=False) | |
| #sdo_json=gr.JSON() | |
| with gr.Tab("SOHO Images"): | |
| soho_btn=gr.Button("Load") | |
| html_nasa=gr.HTML() | |
| with gr.Tab("SOHO Video"): | |
| soho_vid_btn=gr.Button("Load") | |
| html_vid_nasa=gr.HTML() | |
| ###### NASA Images ########## | |
| sdo_comp_btn.click(get_date,[year1,month1,day1,hour1,minute1,year2,month2,day2,hour2,minute2],[date1,date2]).then(nasa_sdo_composite,[sdo_drop1,sdo_drop2,date1,date2],sdo_comp_html) | |
| sdo_btn.click(get_date,[year1,month1,day1,hour1,minute1,year2,month2,day2,hour2,minute2],[date1,date2]).then(nasa_sdo_images,[sdo_drop1,sdo_drop2,date1,date2],sdo_html) | |
| soho_btn.click(make_nasa_soho_image_trigger,None,html_nasa) | |
| soho_vid_btn.click(make_nasa_soho_videos,None,html_vid_nasa) | |
| ###### NOAA Images ########## | |
| load_btn.click(load_json,[drop1,drop2,drop3,drop4],[html_raw,links]) | |
| anim_btn.click(make_animation,None,[file_out]).then(make_html,file_out,anim_out) | |
| ####### Raw ############ | |
| drop1.change(make_tree,drop1,[drop2]) | |
| drop2.change(make_tree,[drop1,drop2],[drop3]) | |
| drop3.change(make_tree,[drop1,drop2,drop3],[drop4]) | |
| ###################### | |
| app.load(run,None,[drop1,html_im]) | |
| app.queue(default_concurrency_limit=10).launch() |