background scheduler

app.py

@@ -12,7 +12,7 @@ import gradio as gr
 import pandas as pd
 
 import time
-from huggingface_hub import HfApi
+from huggingface_hub import restart_space
 import threading
 
 """
@@ -28,6 +28,9 @@ To create a new dashboard, you need several environment variables, that you can
 - ARGILLA_API_KEY: The API key to access the Huggingface Space. Please, write this as a secret in the Huggingface Space configuration.
 """
 
+# Get env variable
+HF_TOKEN = os.getenv("HF_TOKEN")
+
 # Translation of legends and titles
 ANNOTATED = 'Аннотации'
 NUMBER_ANNOTATED = 'Всего аннотаций'
@@ -39,6 +42,9 @@ NUMBER_ANNOTATIONS = 'Количество аннотаций'
 
 CATEGORY = 'Категория'
 
+# Start scheduler timer
+time_start = datetime.now()
+
 def obtain_source_target_datasets() -> (
     Tuple[
         FeedbackDataset | RemoteFeedbackDataset, FeedbackDataset | RemoteFeedbackDataset
@@ -277,6 +283,14 @@ def get_top(N = 10) -> pd.DataFrame:
 
     return obtain_top_users(user_ids_annotations, N=N)
 
+def restart():
+    gr.Info("Restarting space...")
+    restart_space("DIBT-Russian/MPEP_Dashboard", token=HF_TOKEN)
+
+
+with gr.Blocks() as demo:
+    gr.Markdown(f"Time started: {time_start}")
+
 
 def main() -> None:
 
@@ -370,6 +384,11 @@ def main() -> None:
             )
             demo.load(get_top, None, [top_df_plot])
 
+    # Manage background refresh
+    scheduler = BackgroundScheduler()
+    job = scheduler.add_job(restart, "interval", minutes=30)
+    scheduler.start()
+            
     # Launch the Gradio interface
     demo.launch()