add `refresh_scheduler()`

app.py

@@ -11,6 +11,9 @@ from argilla.client.feedback.dataset.remote.dataset import RemoteFeedbackDataset
 import gradio as gr
 import pandas as pd
 
+import time
+from huggingface_hub import HfApi
+
 """
 This is the main file for the dashboard application. It contains the main function and the functions to obtain the data and create the charts.
 It's designed as a template to recreate the dashboard for the prompt translation project of any language. 
@@ -273,12 +276,29 @@ def get_top(N = 10) -> pd.DataFrame:
 
     return obtain_top_users(user_ids_annotations, N=N)
 
+def refresh_scheduler():
+    
+    # Configure your HF Token as an env variable
+    token = os.getenv("HF_TOKEN")
+    
+    # Unique identifier for your space
+    space_id = "DIBT-Russian/MPEP_Dashboard"
 
-def main() -> None:
+    hf_api = HfApi()
+
+    while True:
+        try:
+            # Restart the space
+            hf_api.restart_space(space_id=space_id, token=token)
+            print(f"Space {space_id} has been restarted.")
+        except Exception as e:
+            print(f"Failed to restart Space {space_id}: {e}")
+        
+        # Wait for 10 minutes before restarting again
+        time.sleep(60)  # 600 seconds = 10 minutes
 
-    # Set the update interval
-    update_interval = 300  # seconds
-    update_interval_charts = 30  # seconds
+
+def main() -> None:
 
     # Connect to the space with rg.init()
     rg.init(
@@ -289,13 +309,8 @@ def main() -> None:
     # Fetch the data initially
     fetch_data()
 
-    
-    # Fetch the data on a schedule
-    scheduler = BackgroundScheduler()
-    scheduler.add_job(
-        func=fetch_data, trigger="interval", seconds=update_interval, max_instances=1
-    )
-    scheduler.start()
+    # Fetch data on a schedule
+    refresh_scheduler()
 
     # To avoid the orange border for the Gradio elements that are in constant loading
     css = """
@@ -336,7 +351,6 @@ def main() -> None:
                 kpi_chart_submitted,
                 inputs=[],
                 outputs=[kpi_submitted_plot],
-                every=update_interval_charts,
             )
 
             kpi_remaining_plot = gr.Plot(label="Plot")
@@ -344,7 +358,6 @@ def main() -> None:
                 kpi_chart_remaining,
                 inputs=[],
                 outputs=[kpi_remaining_plot],
-                every=update_interval_charts,
             )
 
             donut_total_plot = gr.Plot(label="Plot")
@@ -352,7 +365,6 @@ def main() -> None:
                 donut_chart_total,
                 inputs=[],
                 outputs=[donut_total_plot],
-                every=update_interval_charts,
             )
 
         gr.Markdown(
@@ -366,7 +378,7 @@ def main() -> None:
 
             kpi_hall_plot = gr.Plot(label="Plot")
             demo.load(
-                kpi_chart_total_annotators, inputs=[], outputs=[kpi_hall_plot], every=update_interval_charts
+                kpi_chart_total_annotators, inputs=[], outputs=[kpi_hall_plot]
             )
 
             top_df_plot = gr.Dataframe(
@@ -378,9 +390,8 @@ def main() -> None:
                 row_count=10,
                 col_count=(2, "fixed"),
                 interactive=False,
-                every=update_interval,
             )
-            demo.load(get_top, None, [top_df_plot], every=update_interval_charts)
+            demo.load(get_top, None, [top_df_plot])
 
     # Launch the Gradio interface
     demo.launch()