tab / app.py
GhoulMac's picture
Update app.py
99a450c
import gradio as gr
import pandas as pd
import Generate
import plotly.express as px
example_df_aircomp=pd.read_csv('data/aircompressordata.csv')
example_df_ener=pd.read_csv('data/energymeter.csv')
example_df_boiler=pd.read_csv('data/Boiler1.csv')
demo=gr.Blocks(title="EcoForecast")
def pred_air(date):
preds=Generate.inference_Aircomp(date,example_df_aircomp)
return preds
def pred_ener(date):
preds=Generate.inference_Energy(date,example_df_ener)
return preds
def pred_boiler(date):
preds=Generate.inference_boiler(date,example_df_boiler)
return preds
def upload_file(files):
file_paths = [file.name for file in files]
return file_paths
with demo:
gr.Markdown("Tool for predicting the next seven days of data in the future using the last 200 points of data incoming")
with gr.Tabs():
with gr.TabItem("Air compressor data"):
with gr.Row():
Air_input=gr.Text(placeholder="Enter date and like the example only",show_label=False)
air_dataframe_input=gr.Dataframe(example_df_aircomp.head(100))
Air_dataframe_output=gr.Dataframe()
with gr.Column():
Aircomp_output_btn=gr.Button("Forecast")
Air_plot_forecast=gr.Button("Plot")
Air_plots=gr.Plot()
with gr.TabItem("Energymeter data"):
with gr.Row():
ener_input=gr.Text(placeholder="Enter the date and time in example format only",show_label=False)
ener_dataframe_input=gr.Dataframe(example_df_ener.head(100))
Ener_dataframe_output=gr.Dataframe()
with gr.Column():
Energy_output_btn=gr.Button("Forecast")
Ener_plot_forecast=gr.Button("Plot")
Ener_plots=gr.Plot()
with gr.TabItem("Boiler data"):
with gr.Row():
boiler_input=gr.Text(placeholder="Enter the date and time in example format only",show_label=False)
boiler_dataframe_input=gr.Dataframe(example_df_boiler.head(100))
boiler_dataframe_output=gr.Dataframe()
with gr.Column():
Boiler_output_btn=gr.Button("Forecast")
boiler_plot_forecast=gr.Button("Plot")
boil_plots=gr.Plot()
with gr.TabItem("Upload data"):
file_output = gr.File()
upload_button = gr.UploadButton("Click to Upload a File", file_types=["csv"], file_count="multiple")
upload_button.upload(upload_file, upload_button, file_output)
def plotg_air(dataframe):
fig=px.line(dataframe,x='timestamp',y=dataframe.columns,
hover_data={"timestamp":"|%B %d, %Y"},
title="Predictions")
fig.update_xaxes(
dtick="D1",
tickformat="%b\n%Y"
)
return fig
def plotg_ener(dataframe):
fig=px.line(dataframe,x='parameter_timestamp',y=dataframe.columns,
hover_data={"parameter_timestamp":"|%B %d, %Y"},
title="Predictions")
fig.update_xaxes(
dtick="D1",
tickformat="%b\n%Y"
)
return fig
def plotg_boiler(dataframe):
fig=px.line(dataframe,x='DateString',y=dataframe.columns,
hover_data={"DateString":"|%B %d, %Y"},
title="Predictions")
fig.update_xaxes(
dtick="D1",
tickformat="%b\n%Y"
)
return fig
Aircomp_output_btn.click(pred_air,inputs=Air_input,outputs=Air_dataframe_output)
Energy_output_btn.click(pred_ener,inputs=ener_input,outputs=Ener_dataframe_output)
Boiler_output_btn.click(pred_boiler,inputs=boiler_input,outputs=boiler_dataframe_output)
Air_plot_forecast.click(plotg_air,inputs=Air_dataframe_output,outputs=Air_plots)
Ener_plot_forecast.click(plotg_ener,inputs=Ener_dataframe_output,outputs=Ener_plots)
boiler_plot_forecast.click(plotg_boiler,inputs=boiler_dataframe_output,outputs=boil_plots)
demo.launch(share=True)