Update app.py

app.py

@@ -45,7 +45,7 @@ def get_prediction_result():
     df.loc[df['forecast_point'].isnull(), 'forecast_point'] = df['target_date'].apply(lambda x:x-relativedelta(months=1))
     df = df.loc[~((df['target_date']<(today+relativedelta(months=1)))&(df['電気代'].isnull()))]
     df = df.rename(columns={'電気代':'電気代_予測'})
-    return df[['forecast_point', 'target_date', '電気代_予測']]
+    return df[['forecast_point', 'target_date', '電気代_予測']], df[['forecast_point', 'target_date', '電気代_予測']]
 
 def plot_prediction_result():
     update = gr.LinePlot.update(
@@ -133,6 +133,8 @@ with gr.Blocks() as electoric_ploting:
                              """
                         )
     with gr.Tab("予測結果"):
+        with gr.Row():
+            reload_btn= gr.Button(value="再読み込み")
         with gr.Row():
             with gr.Column():
                 plot = gr.LinePlot(show_label=False)
@@ -147,7 +149,7 @@ with gr.Blocks() as electoric_ploting:
                 gr.Textbox(df['電気代_予測'].median(),
                                 label='現在までの予測値の中央値')
         with gr.Row():
-            gr.DataFrame(get_prediction_result)
+            output_dataframe = gr.DataFrame()
             
             
     with gr.Tab("モデル情報"):
@@ -241,13 +243,14 @@ with gr.Blocks() as electoric_ploting:
                 )
 
     retrain_btn.click(retrain, inputs=None, outputs = [output_model_type, output_acc, output_plot])
+    retrain_btn.click(retrain, inputs=None, outputs = [output_model_type, output_acc, output_plot])
     
     
-    electoric_ploting.load(lambda: datetime.datetime.now(), 
+    electoric_ploting.load(plot_prediction_result, 
                            None,
-                           # c_time2,
+                           [plot, output_dataframe],
                            every=3600)
-    dep = electoric_ploting.load(plot_prediction_result, None, plot, every=3600)
+    # dep = electoric_ploting.load(plot_prediction_result, None, [plot, output_dataframe], every=3600)
 
 electoric_ploting.launch()