Update app.py

app.py

@@ -51,6 +51,7 @@ def predict_values(start_doc, end_doc, days_option):
 
     # 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
 
     # Assign the generated DOC values to the future dataframes
     farm_future_rate['DOC'] = doc_values
@@ -64,18 +65,13 @@ 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'] = farm_future_rate['DOC'].tail(days_option)
+    farm_output['DOC'] = np.round(farm_future_rate['DOC'].tail(days_option), 1)  # Round DOC values
 
     open_output = open_forecast_rate[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail(days_option).copy()
-    open_output['DOC'] = open_future_rate['DOC'].tail(days_option)
+    open_output['DOC'] = np.round(open_future_rate['DOC'].tail(days_option), 1)  # Round DOC values
 
     close_output = close_forecast_rate[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail(days_option).copy()
-    close_output['DOC'] = close_future_rate['DOC'].tail(days_option)
-
-    # Round float values to two decimal places
-    farm_output[['yhat', 'yhat_lower', 'yhat_upper']] = farm_output[['yhat', 'yhat_lower', 'yhat_upper']].round(2)
-    open_output[['yhat', 'yhat_lower', 'yhat_upper']] = open_output[['yhat', 'yhat_lower', 'yhat_upper']].round(2)
-    close_output[['yhat', 'yhat_lower', 'yhat_upper']] = close_output[['yhat', 'yhat_lower', 'yhat_upper']].round(2)
+    close_output['DOC'] = np.round(close_future_rate['DOC'].tail(days_option), 1)  # Round DOC values
 
     # Create Plotly graphs
     fig_farm = go.Figure()
@@ -98,7 +94,7 @@ def interface(start_doc, end_doc, days_option):
     if days_option is None:
         return "Please select a valid option for days.", None, None, None, None, None
     
-    days_map = {'7 days': 7, '10 days': 10, '15 days': 15,'40 days':40}
+    days_map = {'7 days': 7, '10 days': 10, '15 days': 15, '40 days': 40}
     days_selected = days_map.get(days_option, 7)  # Default to 7 days if no valid option is provided
     results_farm, results_open, results_close, plot_farm, plot_open, plot_close = predict_values(start_doc, end_doc, days_selected)
     return results_farm, results_open, results_close, plot_farm, plot_open, plot_close
@@ -106,7 +102,7 @@ def interface(start_doc, end_doc, days_option):
 # Create Gradio inputs and outputs
 start_doc_input = gr.components.Number(label="Start DOC Value")
 end_doc_input = gr.components.Number(label="End DOC Value")
-days_dropdown = gr.components.Dropdown(choices=['7 days', '10 days', '15 days','40 days'], label="Select Number of Days",value='7 days')
+days_dropdown = gr.components.Dropdown(choices=['7 days', '10 days', '15 days', '40 days'], label="Select Number of Days", value='7 days')
 
 # Define output components
 output_table_farm = gr.components.Dataframe(label="Predicted Farm Rate Values")