Update app.py

app.py

@@ -11,7 +11,8 @@ from st_aggrid import AgGrid
 from nixtla import NixtlaClient
 import os
 
-
+st.set_page_config(layout='wide')
+    
 @st.cache_resource
 def load_model(path, freq):
     nf = NeuralForecast.load(path=path)
@@ -358,6 +359,7 @@ def timegpt_fcst():
             columns.pop(columns.index('unique_id'))
         opt = columns
         y_col = st.selectbox("Select Target column", options=opt, index=0)
+        h = st.number_input("Forecast horizon", value=14)
 
         df = df.rename(columns={ds_col: 'ds', y_col: 'y'})
 
@@ -377,17 +379,15 @@ def timegpt_fcst():
         if st.sidebar.button("Submit"):
             forecast_df = nixtla_client.forecast(
                 df=df,
-                h=7,
+                h=h,
                 freq=freq,
                 level=[90]
             )
+            st.session_state.forecast_df = forecast_df
 
-    st.pyplot(nixtla_client.plot(
-        df,
-        forecast_df,
-        level=[90],
-        max_insample_length=365
-    ))
+    if 'forecast_df' in st.session_state:
+        forecast_df = st.session_state.forecast_df
+        st.pyplot(nixtla_client.plot(df, forecast_df, level=[90]))
 
 
 
@@ -435,11 +435,11 @@ def timegpt_anom():
                 freq=freq,
                 level=90
             )
+            st.session_state.anom_df = anom_df
 
-    st.pyplot(nixtla_client.plot(
-        df,
-        anom_df
-    ))
+    if 'anom_df' in st.session_state:
+        anom_df = st.session_state.anom_df
+        st.pyplot(nixtla_client.plot(df, anom_df))