Update models & remove unused ones

Author: rbusdon

fetch_data.py

@@ -68,11 +68,13 @@ def clear_directory(directory: Path) -> None:
             maximum_longitude=18,
             minimum_latitude=42,
             maximum_latitude=47,
+            minimum_depth=0,
+            maximum_depth=120,
             start_datetime=formatted_date,
             end_datetime=formatted_date,
             output_filename=str(depths_nc_path),
             force_download=True
-            )
+        )
 
 d_dataset = nc.Dataset(str(depths_nc_path)).variables["depth"][:]
 

rom.py

@@ -1,7 +1,9 @@
+import sys
 import numpy as np
 import streamlit as st
 import cloudpickle
 import matplotlib.pyplot as plt
+from matplotlib.colors import LogNorm
 from scipy.ndimage import gaussian_filter
 from mpl_toolkits.axes_grid1 import make_axes_locatable
 from pathlib import Path
@@ -15,23 +17,26 @@
     page_icon="ogs_logo.png"
 )
 
-Po=np.linspace(200,8000,10000)
-Isonzo=np.linspace(0,1500,10000)
-Timavo=np.linspace(8,150,10000)
+Po=np.linspace(200,4000,10000)
+Isonzo=np.linspace(0,500,10000)
+Timavo=np.linspace(8,80,10000)
 variables=["N1p","N3n","S","T","Chla"]
 depth={"O m":0, "10 m":1, "30 m":2, "60 m":3, "100 m":4}
 
+vmins={"N1p":0,"N3n": 0, "Chla":0, "S":30, "T":13}
+vmaxs={"N1p":0.2,"N3n": 10, "Chla":4, "S":39.5, "T":30}
+
 # --- CONFIGURE YOUR CHOICES ---
-depth_list = [0,10,30,60,100]
+depth_list = [0,10]
 variables_list=["N1p","N3n","S","T","Chla"]
-days_list=[0,1,2,3,4]
+days_list=[0,1,2,3,4,5]
 
 depth_dict={
     0:0,
     10:1,
-    30:2,
-    60:3,
-    100:4
+    # 30:2,
+    # 60:3,
+    # 100:4
 
 }
 
@@ -63,7 +68,8 @@ def model_predict(days, variable, depth, po, isonzo, timavo):
 fig, ax = plt.subplots()
 divider = make_axes_locatable(ax)
 cax = divider.append_axes('right', size='5%', pad=0.05)
-im=ax.imshow(Z, origin='lower', aspect='auto', cmap="Pastel1")
+# im=ax.imshow(Z, origin='lower', aspect='auto', cmap="Pastel1")
+im=ax.imshow(Z, origin='lower', aspect='auto', cmap="jet",vmin=vmins[variables_ctx],vmax=vmaxs[variables_ctx])
 #ax.set_title(f"{days_ctx} | ctx1={variables_ctx}, ctx2={depth_ctx} | a={po}, b={isonzo}, c={timavo}")
 fig.colorbar(im, cax=cax, orientation='vertical')