"""Configuration constants for the Reddit Adverse Event Agent."""

import os

# StemeDB API
STEMEDB_URL = os.getenv("STEMEDB_URL", "http://localhost:18180")

# Reddit scraping
TARGET_SUBREDDITS = ["Ozempic", "Mounjaro", "Semaglutide", "Wegovy"]
ADVERSE_EVENT_KEYWORDS = [
    "stomach",
    "paralysis",
    "gastroparesis",
    "vomit",
    "nausea",
    "emergency",
    "hospital",
    "pain",
    "stopped working",
    "hair loss",
    "side effect",
    "adverse",
    "reaction",
]

# Reddit HTTP headers
REDDIT_HEADERS = {
    "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36",
    "Accept": "application/json",
}

# StemeDB Source Class for Social/Anecdotal (Tier 5)
SOURCE_CLASS_SOCIAL = "Anecdotal"

# Confidence limits for Tier 5 data
# Per plan: 0.3-0.7 max for anecdotal data
MIN_CONFIDENCE = 0.3
MAX_CONFIDENCE = 0.7

# Lifecycle stage - all assertions start as Proposed
DEFAULT_LIFECYCLE = "Proposed"

# Drug name mappings (subreddit -> canonical drug name)
DRUG_MAP = {
    "ozempic": "semaglutide",
    "wegovy": "semaglutide",
    "mounjaro": "tirzepatide",
    "semaglutide": "semaglutide",
}

# Environment variable names
ENV_STEMEDB_AGENT_SEED = "STEMEDB_AGENT_SEED"
ENV_GOOGLE_API_KEY = "GOOGLE_API_KEY"
