autorestart

app.py

@@ -8,6 +8,7 @@ import altair as alt
 import argilla as rg
 from argilla.feedback import FeedbackDataset
 from argilla.client.feedback.dataset.remote.dataset import RemoteFeedbackDataset
+from huggingface_hub import restart_space
 import gradio as gr
 import pandas as pd
 
@@ -17,6 +18,7 @@ It's designed as a template to recreate the dashboard for the prompt translation
 
 To create a new dashboard, you need several environment variables, that you can easily set in the HuggingFace Space that you are using to host the dashboard:
 
+- HF_TOKEN: Token with write access from your Hugging Face account: https://huggingface.co/settings/tokens
 - SOURCE_DATASET: The dataset id of the source dataset
 - SOURCE_WORKSPACE: The workspace id of the source dataset
 - TARGET_RECORDS: The number of records that you have as a target to annotate. We usually set this to 500.
@@ -25,15 +27,30 @@ To create a new dashboard, you need several environment variables, that you can
 """
 
 # Translation of legends and titles
-ANNOTATED = 'Annotations'
-NUMBER_ANNOTATED = 'Total Annotations'
-PENDING = 'Pending'
+ANNOTATED = "Annotations"
+NUMBER_ANNOTATED = "Total Annotations"
+PENDING = "Pending"
 
 NUMBER_ANNOTATORS = "Number of annotators"
-NAME = 'Username'
-NUMBER_ANNOTATIONS = 'Number of annotations'
+NAME = "Username"
+NUMBER_ANNOTATIONS = "Number of annotations"
+
+CATEGORY = "Category"
+
+
+def restart() -> None:
+    """
+    This function restarts the space where the dashboard is hosted.
+    """
+
+    # Update Space name with your Space information
+    gr.Info("Restarting space at " + str(datetime.datetime.now()))
+    restart_space(
+        "ignacioct/TryingRestartDashboard",
+        token=os.getenv("HF_TOKEN"),
+        # factory_reboot=True,
+    )
 
-CATEGORY = 'Category'
 
 def obtain_source_target_datasets() -> (
     Tuple[
@@ -186,9 +203,7 @@ def kpi_chart_total_annotators() -> alt.Chart:
     total_annotators = len(user_ids_annotations)
 
     # Assuming you have a DataFrame with user data, create a sample DataFrame
-    data = pd.DataFrame(
-        {"Category": [NUMBER_ANNOTATORS], "Value": [total_annotators]}
-    )
+    data = pd.DataFrame({"Category": [NUMBER_ANNOTATORS], "Value": [total_annotators]})
 
     # Create Altair chart
     chart = (
@@ -201,14 +216,14 @@ def kpi_chart_total_annotators() -> alt.Chart:
     return chart
 
 
-def render_hub_user_link(hub_id:str) -> str:
+def render_hub_user_link(hub_id: str) -> str:
     """
     This function returns a link to the user's profile on Hugging Face.
 
     Args:
         hub_id: The user's id on Hugging Face.
 
-    Returns:    
+    Returns:
         A string with the link to the user's profile on Hugging Face.
     """
     link = f"https://huggingface.co/{hub_id}"
@@ -255,7 +270,7 @@ def fetch_data() -> None:
     print(f"Data fetched: {datetime.datetime.now()}")
 
 
-def get_top(N = 50) -> pd.DataFrame:
+def get_top(N=50) -> pd.DataFrame:
     """
     This function returns the top N users with the most annotations.
 
@@ -287,7 +302,7 @@ def main() -> None:
     }
     """
 
-    with gr.Blocks(css=css) as demo:
+    with gr.Blocks(css=css, delete_cache=(300, 300)) as demo:
         gr.Markdown(
             """
             # 🌍 [YOUR LANGUAGE] - Multilingual Prompt Evaluation Project
@@ -343,9 +358,7 @@ def main() -> None:
         with gr.Row():
 
             kpi_hall_plot = gr.Plot(label="Plot")
-            demo.load(
-                kpi_chart_total_annotators, inputs=[], outputs=[kpi_hall_plot]
-            )
+            demo.load(kpi_chart_total_annotators, inputs=[], outputs=[kpi_hall_plot])
 
             top_df_plot = gr.Dataframe(
                 headers=[NAME, NUMBER_ANNOTATIONS],
@@ -359,9 +372,14 @@ def main() -> None:
             )
             demo.load(get_top, None, [top_df_plot])
 
+    # Manage background refresh
+    scheduler = BackgroundScheduler()
+    _ = scheduler.add_job(restart, "interval", minutes=30)
+    scheduler.start()
+
     # Launch the Gradio interface
     demo.launch()
 
 
 if __name__ == "__main__":
-    main()
+    main()