Country Research Assistants¶

import json
from datetime import datetime
from IPython.display import HTML, display
import arcticdb as adb

# Connect to Arctic
arctic_uri = "s3://s3.us-east-2.amazonaws.com:kate-cap-arctic?aws_auth=true"
arctic = adb.Arctic(arctic_uri)
lib = arctic.get_library(
    "external_research",
    create_if_missing=True,
    library_options=adb.LibraryOptions(dynamic_schema=True, dedup=True),
)

# Load topics and questions from Arctic
try:
    topics_df = lib.read("notebook_topics").data
except Exception:
    topics_df = None

try:
    questions_df = lib.read("notebook_questions").data
except Exception:
    questions_df = None

# Notebook configuration
NOTEBOOKS = {
    "usa_economic": {
        "title": "USA Economic Research",
        "country": "USA",
        "url": "https://notebooklm.cloud.google.com/us/notebook/65d9153d-c23f-48fe-a72f-d538ccf67892?project=294334118581",
        "description": "CPI, GDP, unemployment, Fed policy, and bond markets",
    },
    "uk_economic": {
        "title": "UK Economic Research",
        "country": "UK",
        "url": "https://notebooklm.cloud.google.com/us/notebook/71e46ba2-174e-4681-b705-6e65d9a3e21d?project=294334118581",
        "description": "Inflation, growth, Bank of England policy, and gilt markets",
    },
    "japan_economic": {
        "title": "Japan Economic Research",
        "country": "Japan",
        "url": "https://notebooklm.cloud.google.com/us/notebook/c3b43b06-9bb4-4e7a-8db9-bde46d61523e?project=294334118581",
        "description": "Inflation dynamics, BOJ policy normalization, and JGB markets",
    },
    "eurozone_economic": {
        "title": "Eurozone Economic Research",
        "country": "Eurozone",
        "url": "https://notebooklm.cloud.google.com/us/notebook/5465b34e-819d-431b-8e61-2776d0400202?project=294334118581",
        "description": "Inflation, growth, ECB policy, and peripheral spreads",
    },
    "canada_economic": {
        "title": "Canada Economic Research",
        "country": "Canada",
        "url": "https://notebooklm.cloud.google.com/us/notebook/c4806561-a4be-4df0-82fc-279c1372c031?project=294334118581",
        "description": "Inflation, housing markets, Bank of Canada policy, and government bonds",
    },
    "china_economic": {
        "title": "China Economic Research",
        "country": "China",
        "url": "https://notebooklm.cloud.google.com/us/notebook/4d296776-d014-42b0-964a-e74d393aea78?project=294334118581",
        "description": "Growth, property sector, PBOC policy, and government bonds",
    },
    "germany_economic": {
        "title": "Germany Economic Research",
        "country": "Germany",
        "url": "https://notebooklm.cloud.google.com/us/notebook/2dd0d74c-8672-4d55-b9cd-9a67c24e17ef?project=294334118581",
        "description": "Inflation, manufacturing, Bundesbank policy, and Bund markets",
    },
    "brazil_economic": {
        "title": "Brazil Economic Research",
        "country": "Brazil",
        "url": "https://notebooklm.cloud.google.com/us/notebook/8c0b86c0-e846-4170-a6bf-0391462b604d?project=294334118581",
    },
}


def get_topics(notebook_id, limit=10):
    """Get topics for a notebook."""
    if topics_df is None:
        return []
    df = (
        topics_df[topics_df["notebook_id"] == notebook_id]
        .sort_values("rank")
        .head(limit)
    )
    return df["topic"].tolist()


def get_questions(notebook_id):
    """Get questions for a notebook."""
    if questions_df is None:
        return []
    df = questions_df[questions_df["notebook_id"] == notebook_id].sort_values(
        "question_index"
    )
    return df["question"].tolist()


def render_notebook_card(notebook_id, config):
    """Render a notebook card with topics and questions."""
    topics = get_topics(notebook_id)
    questions = get_questions(notebook_id)

    topics_html = ""
    if topics:
        topic_items = "".join(
            [
                f'<span style="background:#e3f2fd;padding:4px 12px;border-radius:16px;margin:3px;display:inline-block;font-size:1em;">{t}</span>'
                for t in topics[:8]
            ]
        )
        topics_html = f'<div style="margin:15px 0;"><strong style="font-size:1.1em;">Topics:</strong><br><div style="margin-top:8px;">{topic_items}</div></div>'

    questions_html = ""
    if questions:
        q_items = "".join(
            [
                f'<li style="margin:8px 0;font-size:1em;line-height:1.4;">{q}</li>'
                for q in questions
            ]
        )
        questions_html = f'<div style="margin:15px 0;"><strong style="font-size:1.1em;">Sample Questions:</strong><ol style="margin:10px 0;padding-left:25px;">{q_items}</ol></div>'

    return f'''
    <div style="border:2px solid #ddd;border-radius:12px;padding:25px;margin:20px 0;background:#fafafa;box-shadow:0 2px 8px rgba(0,0,0,0.1);">
        <h2 style="margin:0 0 10px 0;font-size:1.6em;">
            <a href="{config["url"]}" target="_blank" style="color:#1a73e8;text-decoration:none;">
                {config["country"]} - {config["title"]}
            </a>
        </h2>
        <p style="color:#555;margin:10px 0;font-style:italic;font-size:1.1em;">{config["description"]}</p>
        {topics_html}
        {questions_html}
        <a href="{config["url"]}" target="_blank" 
           style="display:inline-block;background:#1a73e8;color:white;padding:12px 24px;border-radius:6px;text-decoration:none;margin-top:15px;font-size:1.1em;font-weight:bold;">
            Open in NotebookLM
        </a>
    </div>
    '''


# Render all notebook cards
html_output = '<div style="max-width:1200px;">'
for notebook_id, config in NOTEBOOKS.items():
    html_output += render_notebook_card(notebook_id, config)

# Add timestamp
html_output += f'<p style="color:#888;font-size:1em;margin-top:30px;">Last updated: {datetime.now().strftime("%B %d, %Y at %I:%M %p")}</p>'
html_output += "</div>"

display(HTML(html_output))