Update app.py

app.py

@@ -50,8 +50,8 @@ def predict_values(start_doc, end_doc, days_option):
     close_future_rate = close_rate_model.make_future_dataframe(periods=days_option, freq='D')
 
     # Generate DOC values within the specified range
-    doc_values = np.linspace(start_doc, end_doc, num=len(farm_future_rate))  # Adjusted line
-    doc_values = np.round(doc_values, 1)  # Round DOC values to 1 decimal place
+    doc_values = np.linspace(start_doc, end_doc, num=len(farm_future_rate))
+    doc_values = np.round(doc_values, 1)
 
     # Assign the generated DOC values to the future dataframes
     farm_future_rate['DOC'] = doc_values
@@ -65,13 +65,24 @@ def predict_values(start_doc, end_doc, days_option):
 
     # Filter to get only the future predictions (last `days_option` days)
     farm_output = farm_forecast_rate[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail(days_option).copy()
-    farm_output['DOC'] = np.round(farm_future_rate['DOC'].tail(days_option), 1)  # Round DOC values
+    farm_output['DOC'] = np.round(farm_future_rate['DOC'].tail(days_option), 1)
+
+    # Round the predictions to 1 decimal place
+    farm_output['yhat'] = np.round(farm_output['yhat'], 1)
+    farm_output['yhat_lower'] = np.round(farm_output['yhat_lower'], 1)
+    farm_output['yhat_upper'] = np.round(farm_output['yhat_upper'], 1)
 
     open_output = open_forecast_rate[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail(days_option).copy()
-    open_output['DOC'] = np.round(open_future_rate['DOC'].tail(days_option), 1)  # Round DOC values
+    open_output['DOC'] = np.round(open_future_rate['DOC'].tail(days_option), 1)
+    open_output['yhat'] = np.round(open_output['yhat'], 1)
+    open_output['yhat_lower'] = np.round(open_output['yhat_lower'], 1)
+    open_output['yhat_upper'] = np.round(open_output['yhat_upper'], 1)
 
     close_output = close_forecast_rate[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail(days_option).copy()
-    close_output['DOC'] = np.round(close_future_rate['DOC'].tail(days_option), 1)  # Round DOC values
+    close_output['DOC'] = np.round(close_future_rate['DOC'].tail(days_option), 1)
+    close_output['yhat'] = np.round(close_output['yhat'], 1)
+    close_output['yhat_lower'] = np.round(close_output['yhat_lower'], 1)
+    close_output['yhat_upper'] = np.round(close_output['yhat_upper'], 1)
 
     # Create Plotly graphs
     fig_farm = go.Figure()