Update app.py

app.py

@@ -51,8 +51,8 @@ if date_column in features:
 
 selected_features = st.multiselect("Select features for analysis", features)
 
-# Specify forecast period
-forecast_period = st.number_input("Enter forecast period in days", min_value=1, value=180)
+# Specify forecast period in months
+forecast_period_months = st.number_input("Enter forecast period in months", min_value=1, value=6)
 
 # Check if the necessary columns exist
 if target_variable and date_column in data.columns:
@@ -63,8 +63,8 @@ if target_variable and date_column in data.columns:
     model = Prophet()
     model.fit(train_data)
 
-    # Make future predictions for the specified number of days
-    future = model.make_future_dataframe(periods=forecast_period)  # Predicting specified days into the future
+    # Make future predictions for the specified number of months
+    future = model.make_future_dataframe(periods=forecast_period_months, freq='M')  # Predicting specified months into the future
     forecast = model.predict(future)
 
     # Display the forecast