Update app.py

app.py

@@ -73,23 +73,35 @@ for tab, city in zip(tabs, unique_cities):
             fig = px.line(forecast_df, x=forecast_df.index, y="Forecast")
             st.plotly_chart(fig)
 
-        # Grid search button
-        if st.button(f'Run Grid Search for {city}'):
-            best_aic = float('inf')
-            best_params = None
-            for param_set in product(range(3), repeat=3):  # Adjust the range and repeat parameters as needed
-                for seasonal_param_set in product(range(3), repeat=4):  # Adjust for seasonal parameters
-                    _, temp_aic = run_sarimax(city_data, param_set, seasonal_param_set+(12,))
-                    if temp_aic and temp_aic < best_aic:
-                        best_aic = temp_aic
-                        best_params = (param_set, seasonal_param_set+(12,))
+            # Grid search button
+            if st.button(f'Run Grid Search for {city}'):
+                best_aic = float('inf')
+                best_params = None
+                # Define the range for each parameter
+                p_range = d_range = q_range = range(3)
+                P_range = D_range = Q_range = range(3)
+                S = 12  # Assuming a fixed seasonal period, adjust as needed
+            
+                # Perform the grid search
+                for params in product(p_range, d_range, q_range, P_range, D_range, Q_range):
+                    order = params[:3]
+                    seasonal_order = params[3:] + (S,)
+                    try:
+                        _, temp_aic = run_sarimax(city_data, order, seasonal_order)
+                        if temp_aic < best_aic:
+                            best_aic = temp_aic
+                            best_params = (order, seasonal_order)
+                    except Exception as e:
+                        st.error(f"An error occurred for parameters {params}: {e}")
+            
+                # Update the session state with the best parameters
+                if best_params is not None:
+                    st.session_state.best_params = {
+                        'order': best_params[0],
+                        'seasonal_order': best_params[1]
+                    }
+                    st.write(f"Best Parameters for {city}: {best_params} with AIC: {best_aic}")
 
-            # Updating session state with the best parameters
-            st.session_state.best_params = {
-                'order': best_params[0],
-                'seasonal_order': best_params[1]
-            }
-            st.write(f"Best Parameters for {city}: {best_params} with AIC: {best_aic}")
 
         # Export to Excel button
         if st.button(f'Export {city} to Excel'):