app

app.py

@@ -27,12 +27,13 @@ app, rt = fast_app(html_style=(style,))
 login(token=os.environ.get("HF_TOKEN"))
 
 hf_user = whoami(os.environ.get("HF_TOKEN"))["name"]
-HF_REPO_ID = f"{hf_user}/zotero-articles"
+HF_REPO_ID_TXT = f"{hf_user}/zotero-answer-ai-article-texts"
+HF_REPO_ID_IMG = f"{hf_user}/zotero-answer-ai-article-images"
 
-abstract_ds = load_dataset(HF_REPO_ID, "abstracts", split="train")
-article_ds = load_dataset(HF_REPO_ID, "articles", split="train")
+abstract_ds = load_dataset(HF_REPO_ID_TXT, "abstracts", split="train")
+article_ds = load_dataset(HF_REPO_ID_TXT, "articles", split="train")
 
-image_ds = load_dataset(HF_REPO_ID, "images", split="train")
+image_ds = load_dataset(HF_REPO_ID_IMG, "images", split="train")
 image_ds = image_ds.filter(lambda x: x["page_number"] == 1)
 
 
@@ -89,11 +90,7 @@ def generate_week_content(current_week):
     article_cards = []
     for arxiv_id in articles:
         article, abstract, image = get_article_details(arxiv_id)
-        article_title = (
-            article["contents"][0].get("paper_title", "article")
-            if article["contents"]
-            else "article"
-        )
+        article_title = article["contents"][0].get("paper_title", "article") if article["contents"] else "article"
 
         card_content = [
             H5(
@@ -130,9 +127,7 @@ def generate_week_content(current_week):
     week_end = current_week + timedelta(days=6)
     return Div(
         nav_buttons,
-        H3(
-            f"Week of {current_week.strftime('%B %d')} - {week_end.strftime('%B %d, %Y')} ({len(articles)} articles)"
-        ),
+        H3(f"Week of {current_week.strftime('%B %d')} - {week_end.strftime('%B %d, %Y')} ({len(articles)} articles)"),
         grid,
         nav_buttons,
         id="content",