Update app.py

app.py

@@ -79,10 +79,10 @@ for tab, city in zip(tabs, unique_cities):
             st.write(f"Non-Seasonal Order: {(p, d, q)}, Seasonal Order: {(P, D, Q, S)}")
             forecast_index = pd.date_range(start=city_data.index[-1], periods=7, freq='M')[1:]
             forecast_index = forecast_index.to_period('M')  # Convert to period index with monthly frequency
-            forecast_df = pd.DataFrame(forecast, columns=['Forecast'])
+            forecast_df = pd.DataFrame(forecast, columns=['predicted_mean'])
             forecast_df = forecast_df.round(0)
             st.table(forecast_df)
-            fig = px.line(forecast_df, x=forecast_df.index, y="Forecast")
+            fig = px.line(forecast_df, x=forecast_df.index, y="predicted_mean")
             st.plotly_chart(fig)
 
             # Grid search button
@@ -119,6 +119,4 @@ for tab, city in zip(tabs, unique_cities):
         if st.button(f'Export {city} to Excel'):
             df_to_export = forecast_df
             excel_data = to_excel(df_to_export)
-            st.download_button(label='📥 Download Excel', data=excel_data, file_name=f'{city}_forecast.xlsx', mime='application/vnd.ms-excel')
-
-# Rest of your code
+            st.download_button(label='📥 Download Excel', data=excel_data, file_name=f'{city}_forecast.xlsx', mime='application/vnd.ms-excel')