Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import pandas as pd | |
| from prophet import Prophet | |
| import plotly.graph_objs as go | |
| import numpy as np | |
| # Function to replace null values with average | |
| def replace_null_with_avg_values(df, column_name): | |
| df[column_name] = pd.to_numeric(df[column_name], errors='coerce') | |
| avg_value = round(df[column_name].mean(), 1) | |
| df[column_name].fillna(avg_value, inplace=True) | |
| # Load and process the data | |
| broiler_data = pd.read_csv('Broiler market price.csv') | |
| replace_null_with_avg_values(broiler_data, 'DOC') | |
| replace_null_with_avg_values(broiler_data, 'Farm Rate') | |
| replace_null_with_avg_values(broiler_data, 'Open') | |
| replace_null_with_avg_values(broiler_data, 'Close') | |
| broiler_data['Date'] = pd.to_datetime(broiler_data['Date']) | |
| # Prepare dataframes for Prophet models | |
| farm_rate_df = broiler_data[['Date', 'Farm Rate']].rename(columns={'Date': 'ds', 'Farm Rate': 'y'}) | |
| open_rate_df = broiler_data[['Date', 'Open']].rename(columns={'Date': 'ds', 'Open': 'y'}) | |
| close_rate_df = broiler_data[['Date', 'Close']].rename(columns={'Date': 'ds', 'Close': 'y'}) | |
| doc_rate_df = broiler_data[['Date', 'DOC']].rename(columns={'Date': 'ds', 'DOC': 'y'}) | |
| # Initialize Prophet models | |
| farm_rate_model = Prophet(daily_seasonality=True) | |
| open_rate_model = Prophet(daily_seasonality=True) | |
| close_rate_model = Prophet(daily_seasonality=True) | |
| doc_rate_model = Prophet(daily_seasonality=True) | |
| # Add holidays | |
| farm_rate_model.add_country_holidays(country_name='PAK') | |
| open_rate_model.add_country_holidays(country_name='PAK') | |
| close_rate_model.add_country_holidays(country_name='PAK') | |
| doc_rate_model.add_country_holidays(country_name='PAK') | |
| # Fit the models | |
| farm_rate_model.fit(farm_rate_df) | |
| open_rate_model.fit(open_rate_df) | |
| close_rate_model.fit(close_rate_df) | |
| doc_rate_model.fit(doc_rate_df) | |
| # Function to generate predictions based on selected days | |
| def predict_values(days_option): | |
| # Generate future dataframes | |
| farm_future_rate = farm_rate_model.make_future_dataframe(periods=days_option, freq='D') | |
| open_future_rate = open_rate_model.make_future_dataframe(periods=days_option, freq='D') | |
| close_future_rate = close_rate_model.make_future_dataframe(periods=days_option, freq='D') | |
| doc_future_rate = doc_rate_model.make_future_dataframe(periods=days_option, freq='D') | |
| # Make predictions | |
| farm_forecast_rate = farm_rate_model.predict(farm_future_rate) | |
| open_forecast_rate = open_rate_model.predict(open_future_rate) | |
| close_forecast_rate = close_rate_model.predict(close_future_rate) | |
| doc_forecast_rate = doc_rate_model.predict(doc_future_rate) | |
| # 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() | |
| open_output = open_forecast_rate[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail(days_option).copy() | |
| close_output = close_forecast_rate[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail(days_option).copy() | |
| doc_output = doc_forecast_rate[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail(days_option).copy() | |
| # Round the predictions to 1 decimal place | |
| for output in [farm_output, open_output, close_output, doc_output]: | |
| output['yhat'] = np.round(output['yhat'], 1) | |
| output['yhat_lower'] = np.round(output['yhat_lower'], 1) | |
| output['yhat_upper'] = np.round(output['yhat_upper'], 1) | |
| # Create Plotly graphs | |
| fig_farm = go.Figure() | |
| fig_farm.add_trace(go.Scatter(x=farm_forecast_rate['ds'], y=farm_forecast_rate['yhat'], mode='lines+markers', name='Farm Rate Prediction')) | |
| fig_farm.update_layout(title='Farm Rate Predictions over Time', xaxis_title='Date', yaxis_title='Predicted Farm Rate') | |
| fig_open = go.Figure() | |
| fig_open.add_trace(go.Scatter(x=open_forecast_rate['ds'], y=open_forecast_rate['yhat'], mode='lines+markers', name='Open Rate Prediction')) | |
| fig_open.update_layout(title='Open Rate Predictions over Time', xaxis_title='Date', yaxis_title='Predicted Open Rate') | |
| fig_close = go.Figure() | |
| fig_close.add_trace(go.Scatter(x=close_forecast_rate['ds'], y=close_forecast_rate['yhat'], mode='lines+markers', name='Close Rate Prediction')) | |
| fig_close.update_layout(title='Close Rate Predictions over Time', xaxis_title='Date', yaxis_title='Predicted Close Rate') | |
| fig_doc = go.Figure() | |
| fig_doc.add_trace(go.Scatter(x=doc_forecast_rate['ds'], y=doc_forecast_rate['yhat'], mode='lines+markers', name='DOC Prediction')) | |
| fig_doc.update_layout(title='DOC Predictions over Time', xaxis_title='Date', yaxis_title='Predicted DOC') | |
| return farm_output, open_output, close_output, doc_output, fig_farm, fig_open, fig_close, fig_doc | |
| # Define Gradio interface | |
| def interface(days_option): | |
| # Check if a valid selection is made | |
| if days_option is None: | |
| return "Please select a valid option for days.", None, None, None, None, None, None | |
| 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, results_doc, plot_farm, plot_open, plot_close, plot_doc = predict_values(days_selected) | |
| return results_farm, results_open, results_close, results_doc, plot_farm, plot_open, plot_close, plot_doc | |
| # Create Gradio inputs and outputs | |
| 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") | |
| output_table_open = gr.components.Dataframe(label="Predicted Open Rate Values") | |
| output_table_close = gr.components.Dataframe(label="Predicted Close Rate Values") | |
| output_table_doc = gr.components.Dataframe(label="Predicted DOC Values") | |
| output_plot_farm = gr.components.Plot(label="Farm Rate Predictions") | |
| output_plot_open = gr.components.Plot(label="Open Rate Predictions") | |
| output_plot_close = gr.components.Plot(label="Close Rate Predictions") | |
| output_plot_doc = gr.components.Plot(label="DOC Predictions") | |
| # Set up Gradio interface | |
| gr.Interface( | |
| fn=interface, | |
| inputs=[days_dropdown], | |
| outputs=[output_table_farm, output_table_open, output_table_close, output_table_doc, output_plot_farm, output_plot_open, output_plot_close, output_plot_doc], | |
| title="Farm Rate Prediction Tool", | |
| description="Select the number of days to generate predictions and plot." | |
| ).launch(debug=True) | |