Update main.py

main.py

@@ -1,10 +1,14 @@
 from fasthtml.common import *
 from datasets import load_dataset
 import datetime
-from apscheduler.schedulers.background import BackgroundScheduler
 from pbs_data import PBSPublicDataAPIClient
 import os
 from fasthtml_hf import setup_hf_backup
+from fasthtml import FastHTML
+from apscheduler.schedulers.background import BackgroundScheduler
+from apscheduler.triggers.interval import IntervalTrigger
+import atexit
+
 
 custom_css = Style("""
     body {
@@ -313,10 +317,22 @@ def update_data():
     except Exception as e:
         print(f"An error occurred while updating data: {str(e)}")
 
+# Set up the scheduler
+
 # Set up the scheduler
 scheduler = BackgroundScheduler()
-scheduler.add_job(func=update_data, trigger='interval', days=1)
+scheduler.add_job(
+    func=update_data,
+    trigger=IntervalTrigger(hours=24),
+    id='update_data',
+    name='Update Data',
+    replace_existing=True
+)
 scheduler.start()
 
+# Make sure to shut down the scheduler when the app is terminated
+
+atexit.register(lambda: scheduler.shutdown())
+
 setup_hf_backup(app)
 serve()