# retrieval.py (Updated)
# Updates: Removed all CAP dataset, FAISS, municipal, and CourtListener API/code (including browse fallback).
# Simplified to focus on Grok's built-in live search suggestions.
# Kept parse_jurisdiction and STATES (used in prompts).
# Removed unnecessary imports and globals.

import os
import logging
import re
from datetime import datetime

# Logging setup
logger = logging.getLogger("retrieval")
logging.basicConfig(level=logging.INFO)

# State dictionary for jurisdiction
STATES = {
    "AL": "Alabama", "AK": "Alaska", "AZ": "Arizona", "AR": "Arkansas", "CA": "California",
    "CO": "Colorado", "CT": "Connecticut", "DE": "Delaware", "FL": "Florida", "GA": "Georgia",
    "HI": "Hawaii", "ID": "Idaho", "IL": "Illinois", "IN": "Indiana", "IA": "Iowa",
    "KS": "Kansas", "KY": "Kentucky", "LA": "Louisiana", "ME": "Maine", "MD": "Maryland",
    "MA": "Massachusetts", "MI": "Michigan", "MN": "Minnesota", "MS": "Mississippi",
    "MO": "Missouri", "MT": "Montana", "NE": "Nebraska", "NV": "Nevada", "NH": "New Hampshire",
    "NJ": "New Jersey", "NM": "New Mexico", "NY": "New York", "NC": "North Carolina",
    "ND": "North Dakota", "OH": "Ohio", "OK": "Oklahoma", "OR": "Oregon", "PA": "Pennsylvania",
    "RI": "Rhode Island", "SC": "South Carolina", "SD": "South Dakota", "TN": "Tennessee",
    "TX": "Texas", "UT": "Utah", "VT": "Vermont", "VA": "Virginia", "WA": "Washington",
    "WV": "West Virginia", "WI": "Wisconsin", "WY": "Wyoming",
    "Federal": "Federal", "All States": "All States", "Other": "Other States"
}
logger.info("✅ States dictionary loaded.")

def parse_jurisdiction(prompt, default="KY"):
    prompt_words = re.findall(r'\b\w+\b', prompt.upper())  # Split into words
    for code, name in STATES.items():
        if code in prompt_words or name.upper() in prompt_words:
            return code
    return default