cmcmaster commited on
Commit
e7dd981
·
verified ·
1 Parent(s): acbada0

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -7
main.py CHANGED
@@ -9,8 +9,6 @@ from generate_newsletter import process_new_papers
9
  from apscheduler.schedulers.background import BackgroundScheduler
10
  from apscheduler.triggers.cron import CronTrigger
11
 
12
- from fasthtml_hf import setup_hf_backup
13
-
14
  # Initialize Hugging Face API
15
  HF_TOKEN = os.environ.get("HF_TOKEN")
16
  DATASET_NAME = "cmcmaster/this_week_in_rheumatology"
@@ -18,12 +16,12 @@ api = HfApi(token=HF_TOKEN)
18
 
19
  # Initialize scheduler
20
  scheduler = BackgroundScheduler()
21
-
22
  # Schedule newsletter generation to run every Monday at 1 AM UTC
23
  scheduler.add_job(process_new_papers,
24
- CronTrigger(day_of_week='mon', hour=1),
 
25
  kwargs={
26
- 'end_date': '{{ (execution_date - timedelta(days=1)).strftime("%Y-%m-%d") }}',
27
  'test': False
28
  },
29
  id='generate_newsletter',
@@ -125,7 +123,7 @@ def index():
125
  A(datetime.strptime(n.split('/')[0], '%Y%m%d').strftime('%B %d, %Y'),
126
  href=f"/newsletter/{n.split('/')[0]}")) for n in newsletters
127
  ]
128
- return Titled(f"This Week in Rheumatology - {datetime.today()}", H2("Available Newsletters"),
129
  Ul(*links))
130
 
131
 
@@ -163,5 +161,4 @@ def download_pdf(date: str):
163
  P(f"Unable to load PDF for date: {date}"),
164
  A("Back to Index", href="/"))
165
 
166
- setup_hf_backup(app)
167
  serve()
 
9
  from apscheduler.schedulers.background import BackgroundScheduler
10
  from apscheduler.triggers.cron import CronTrigger
11
 
 
 
12
  # Initialize Hugging Face API
13
  HF_TOKEN = os.environ.get("HF_TOKEN")
14
  DATASET_NAME = "cmcmaster/this_week_in_rheumatology"
 
16
 
17
  # Initialize scheduler
18
  scheduler = BackgroundScheduler()
 
19
  # Schedule newsletter generation to run every Monday at 1 AM UTC
20
  scheduler.add_job(process_new_papers,
21
+ trigger="interval",
22
+ hours=6,
23
  kwargs={
24
+ 'end_date': '{{ (execution_date - timedelta(days=execution_date.weekday() + 1)).strftime("%Y-%m-%d") }}',
25
  'test': False
26
  },
27
  id='generate_newsletter',
 
123
  A(datetime.strptime(n.split('/')[0], '%Y%m%d').strftime('%B %d, %Y'),
124
  href=f"/newsletter/{n.split('/')[0]}")) for n in newsletters
125
  ]
126
+ return Titled("This Week in Rheumatology", H2("Available Newsletters"),
127
  Ul(*links))
128
 
129
 
 
161
  P(f"Unable to load PDF for date: {date}"),
162
  A("Back to Index", href="/"))
163
 
 
164
  serve()