Update main.py
Browse files
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 |
-
|
|
|
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(
|
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()
|