chore: initialize virtual environment and install project dependencies

src/daily_retrieve.py

@@ -31,7 +31,7 @@ load_dotenv(ROOT / ".env")
 # ---------------------------------------------------------------------------
 DATA_DIR = ROOT / "data"
 HF_DATASET_REPO = "Elfsong/hf_paper_summary"
-HF_TRENDING_REPO = "Elfsong/hf_paper_trending"
+HF_TRENDING_REPO = "Elfsong/hf_paper_daily_trending"
 HF_API_URL = "https://huggingface.co/api/daily_papers"
 
 SSL_CTX = ssl.create_default_context()

src/monthly_retrieve.py

@@ -195,26 +195,33 @@ def _build_paper_prompt_content(papers: list[dict]) -> str:
 # ---------------------------------------------------------------------------
 # Gemini call
 # ---------------------------------------------------------------------------
-def generate_monthly_trending(papers: list[dict]) -> dict:
+def generate_monthly_trending(papers: list[dict], max_retries: int = 3) -> dict:
     from google import genai
     api_key = _get_env("GEMINI_API_KEY")
     if not api_key:
         raise RuntimeError("GEMINI_API_KEY not set")
     content = _build_paper_prompt_content(papers)
     client = genai.Client(api_key=api_key)
-    resp = client.models.generate_content(
-        model="gemini-3.1-pro-preview",
-        contents=content,
-        config=genai.types.GenerateContentConfig(
-            system_instruction=MONTHLY_TRENDING_SYSTEM_PROMPT,
-            temperature=0.3,
-            max_output_tokens=4096 * 6,
-            response_mime_type="application/json",
-        ),
-    )
-    decoder = json.JSONDecoder()
-    result, _ = decoder.raw_decode(resp.text.strip())
-    return result
+    for attempt in range(max_retries):
+        try:
+            resp = client.models.generate_content(
+                model="gemini-3.1-pro-preview",
+                contents=content,
+                config=genai.types.GenerateContentConfig(
+                    system_instruction=MONTHLY_TRENDING_SYSTEM_PROMPT,
+                    temperature=0.3,
+                    max_output_tokens=65536,
+                    response_mime_type="application/json",
+                ),
+            )
+            decoder = json.JSONDecoder()
+            result, _ = decoder.raw_decode(resp.text.strip())
+            return result
+        except Exception as e:
+            if attempt < max_retries - 1:
+                time.sleep((attempt + 1) * 5)
+            else:
+                raise
 
 
 # ---------------------------------------------------------------------------
@@ -461,9 +468,10 @@ def run(month_str: str, no_push: bool = False):
         topics = trending.get("top_topics", [])
         display.complete_step(3, f"{len(topics)} topics, {len(trending.get('keywords', []))} keywords")
     except Exception as e:
-        display.error_step(3, str(e)[:60])
+        display.error_step(3, "failed")
         display.skip_step(4)
         display.finish()
+        print(f"\n  {_YELLOW}{_BOLD}Error:{_RESET} {e}\n")
         return
 
     # --- Step 4: Push ---
@@ -475,7 +483,10 @@ def run(month_str: str, no_push: bool = False):
             push_monthly_trending_to_hf(trending, month_str)
             display.complete_step(4, f"split={target_split}")
         except Exception as e:
-            display.error_step(4, str(e)[:60])
+            display.error_step(4, "failed")
+            display.finish()
+            print(f"\n  {_YELLOW}{_BOLD}Error:{_RESET} {e}\n")
+            return
 
     display.finish()
 

src/streamlit_app.py

@@ -280,7 +280,7 @@ div[data-testid="stHorizontalBlock"] > div[data-testid="stColumn"] > div > div[d
 # ---------------------------------------------------------------------------
 DATA_DIR = Path(__file__).resolve().parent.parent / "data"
 HF_DATASET_REPO = "Elfsong/hf_paper_summary"
-HF_TRENDING_REPO = "Elfsong/hf_paper_trending"
+HF_TRENDING_REPO = "Elfsong/hf_paper_daily_trending"
 HF_MONTHLY_TRENDING_REPO = "Elfsong/hf_paper_monthly_trending"
 MONTH_RANGE = 6
 
@@ -1452,13 +1452,12 @@ def _load_trending_and_render(
 # ---------------------------------------------------------------------------
 yesterday_str = (datetime.now(timezone.utc) - timedelta(days=1)).strftime("%Y-%m-%d")
 
-# --- Tabs: Daily / Monthly ---
-tab_daily, tab_monthly = st.tabs(["Daily", "Monthly"])
-
+# --- Tab selection ---
 today = datetime.now(timezone.utc).date()
+active_tab = st.segmented_control("", ["Daily", "Monthly"], default="Daily", key="active_tab")
 
 # ---- Daily tab ----
-with tab_daily:
+if active_tab == "Daily":
     col_date_d, col_spacer_d, col_lang_d = st.columns([0.15, 0.75, 0.1], vertical_alignment="center")
     with col_date_d:
         available_dates = list_available_dates()
@@ -1514,7 +1513,7 @@ with tab_daily:
     _render_papers_section(papers, lang, selected_date_str, "daily")
 
 # ---- Monthly tab ----
-with tab_monthly:
+elif active_tab == "Monthly":
     # Discover available monthly trending splits on HF
     _monthly_splits_key = "monthly_available_splits"
     if _monthly_splits_key not in st.session_state: