Update app.py

app.py

@@ -525,6 +525,31 @@ def timegpt_fcst():
             time_taken = end_time - start_time
             st.success(f"Time taken for TimeGPT forecast: {time_taken:.2f} seconds")
 
+            if 'forecast_df' in st.session_state:
+                forecast_df = st.session_state.forecast_df
+        
+                st.markdown('You can download Input and Forecast Data below')
+                tab_insample, tab_forecast  = st.tabs(
+                                ["Input data", "Forecast"]
+                            )
+                    
+                with tab_insample:
+                    df_grid = df.drop(columns="unique_id")
+                    st.write(df_grid)
+                    # grid_table = AgGrid(
+                    #                 df_grid,
+                    #                 theme="alpine",
+                    #             )
+            
+                with tab_forecast:
+                    if model_choice in forecast_results:
+                        df_grid = forecast_df
+                        st.write(df_grid)
+                        # grid_table = AgGrid(
+                        #                 df_grid,
+                        #                 theme="alpine",
+                        #             )
+
 
 
 def timegpt_anom():
@@ -602,7 +627,31 @@ def timegpt_anom():
             end_time = time.time()  # End timing
             time_taken = end_time - start_time
             st.success(f"Time taken for TimeGPT forecast: {time_taken:.2f} seconds")
+
+            if 'forecast_df' in st.session_state:
+                forecast_df = st.session_state.forecast_df
         
+                st.markdown('You can download Input and Forecast Data below')
+                tab_insample, tab_forecast  = st.tabs(
+                                ["Input data", "Forecast"]
+                            )
+                    
+                with tab_insample:
+                    df_grid = df.drop(columns="unique_id")
+                    st.write(df_grid)
+                    # grid_table = AgGrid(
+                    #                 df_grid,
+                    #                 theme="alpine",
+                    #             )
+            
+                with tab_forecast:
+                    if model_choice in forecast_results:
+                        df_grid = forecast_df
+                        st.write(df_grid)
+                        # grid_table = AgGrid(
+                        #                 df_grid,
+                        #                 theme="alpine",
+                        #             )