Upload 2 files

app.py

@@ -0,0 +1,65 @@
+# -*- coding: utf-8 -*-
+"""
+Created on Fri Jun  9 09:12:15 2023
+
+@author: mritchey
+"""
+import gradio as gr
+from joblib import Parallel, delayed
+import pandas as pd
+import rioxarray
+import xarray as xr
+import numpy as np
+from tqdm import tqdm
+
+def get_rtma_data(date_time,var='GUST'):
+    dt=date_time.strftime('%Y%m%d%H%M')
+    date=dt[:8]
+    year,month,day,time=dt[:4],dt[4:6],dt[6:8],dt[8:]
+    url=f'https://mtarchive.geol.iastate.edu/{year}/{month}/{day}/grib2/ncep/RTMA/{date}{time}_{var}.grib2'
+    try:
+        ds_values = xr.open_rasterio(url).values
+    except:
+        ds_values=url
+    return ds_values
+
+
+def get_rtma_values(date,time,days_head,variable):
+    start_date=pd.to_datetime(date+time)
+    end_date = start_date + pd.DateOffset(days=int(days_head))
+    hourly_range = pd.date_range(start=start_date, end=end_date, freq='H')
+
+    mat=Parallel(n_jobs=16)(delayed(get_rtma_data)(date_time,variable) for date_time in tqdm(hourly_range[:2]))
+    mat = [x for x in mat if isinstance(x, np.ndarray)]
+
+    mat_max=np.max(mat,axis=0)
+    mat_min=np.min(mat,axis=0)
+    mat_mean=np.mean(mat,axis=0)
+    mat_std=np.std(mat,axis=0)
+    
+    return [mat_max,mat_min,mat_mean,mat_std]
+    # return [mat_max,mat_min]
+
+    
+
+with gr.Blocks(show_api=True) as demo:
+    date = gr.Textbox(label="Date",placeholder='20220927')
+    time = gr.Textbox(label="Time",placeholder='1200')
+    days_head = gr.Textbox(label="Days Ahead" ,placeholder=1)
+    variable = gr.Textbox(label="Variable",placeholder='GUST')
+    greet_btn = gr.Button("Get Matrix")
+    output = gr.Numpy(label="Max")
+    greet_btn.click(fn=get_rtma_values, 
+                    inputs=[date, time, days_head, variable], 
+                    outputs=output,
+                    api_name="rtma range")
+
+
+
+demo.launch()
+
+# date='20220927'
+# time='1200'
+# days_head=1
+# variable='GUST'
+# get_rtma_values(date,time,days_head,variable)

requirements.txt

@@ -0,0 +1,7 @@
+gradio==3.34.0
+joblib==1.1.0
+numpy==1.19.5
+pandas==1.5.2
+rioxarray==0.10.3
+tqdm==4.65.0
+xarray==2022.11.0