#!/usr/bin/env python3
"""
Chilean Heritage Institutions - Batch 19 Enrichment
Address gap types: RESEARCH_CENTER and MIXED institutions with 0% coverage

Target: 68/90 → 71/90 (78.9% coverage)

Enrichment Strategy:
- Fundación Iglesias Patrimoniales → Q86283277 (direct match)
- Instituto Alemán Puerto Montt → Q36214 (parent_organization: Puerto Montt city)
- Centro Cultural Sofia Hott → Q51059 (parent_organization: Osorno city)

Coverage Impact:
- RESEARCH_CENTER: 0/2 → 1/2 (50%)
- MIXED: 0/3 → 2/3 (66.7%)
- Overall: 75.6% → 78.9% (+3.3 percentage points)

Date: 2025-11-09
"""

import yaml
from pathlib import Path

# Input/output paths
INPUT_FILE = Path(__file__).parent.parent / "data/instances/chile/chilean_institutions_batch18_enriched.yaml"
OUTPUT_FILE = Path(__file__).parent.parent / "data/instances/chile/chilean_institutions_batch19_enriched.yaml"

# Enrichment mappings: institution ID → (Q-number, enrichment type, reason)
ENRICHMENTS = {
    # RESEARCH_CENTER - Direct match
    "https://w3id.org/heritage/custodian/cl/r-fundaci-n-iglesias-patrimonial-0075": {
        "q_number": "Q86283277",
        "type": "direct",
        "reason": "Direct match - Fundación Amigos de las iglesias de Chiloé (museum in Chile)"
    },
    
    # MIXED institutions - Parent organizations (cities)
    "https://w3id.org/heritage/custodian/cl/m-instituto-alem-n-puerto-montt-0074": {
        "q_number": "Q36214",
        "type": "parent_organization",
        "reason": "Parent organization - German school with library/archive in Puerto Montt"
    },
    "https://w3id.org/heritage/custodian/cl/m-centro-cultural-sofia-hott-0079": {
        "q_number": "Q51059",
        "type": "parent_organization",
        "reason": "Parent organization - Cultural center with collections in Osorno"
    }
}

def enrich_institution(inst: dict) -> bool:
    """
    Enrich institution with Wikidata Q-number if applicable.
    Returns True if enriched, False otherwise.
    """
    inst_id = inst.get('id')
    
    if inst_id not in ENRICHMENTS:
        return False
    
    enrichment = ENRICHMENTS[inst_id]
    q_number = enrichment['q_number']
    enrich_type = enrichment['type']
    reason = enrichment['reason']
    
    # Check if already has this Wikidata identifier
    if inst.get('identifiers'):
        for ident in inst['identifiers']:
            if ident.get('identifier_scheme') == 'Wikidata' and ident.get('identifier_value') == q_number:
                print(f"⚠ Already enriched: {inst.get('name')} ({q_number})")
                return False
    
    # Add Wikidata identifier
    if not inst.get('identifiers'):
        inst['identifiers'] = []
    
    inst['identifiers'].append({
        'identifier_scheme': 'Wikidata',
        'identifier_value': q_number,
        'identifier_url': f'https://www.wikidata.org/wiki/{q_number}'
    })
    
    # Update provenance notes
    if not inst.get('provenance'):
        inst['provenance'] = {}
    
    existing_notes = inst['provenance'].get('notes', '')
    enrichment_note = f"Wikidata enrichment (Batch 19 - Gap types): {enrich_type} - {reason}"
    
    if existing_notes:
        inst['provenance']['notes'] = f"{existing_notes} | {enrichment_note}"
    else:
        inst['provenance']['notes'] = enrichment_note
    
    print(f"✓ Enriched: {inst.get('name')}")
    print(f"  Institution type: {inst.get('institution_type')}")
    print(f"  Q-number: {q_number} ({enrich_type})")
    print(f"  Reason: {reason}")
    print()
    
    return True

def main():
    # Load institutions
    with open(INPUT_FILE, 'r', encoding='utf-8') as f:
        institutions = yaml.safe_load(f)
    
    # Enrich institutions
    enriched_count = 0
    enriched_by_type = {}
    
    for inst in institutions:
        if enrich_institution(inst):
            enriched_count += 1
            inst_type = inst.get('institution_type', 'UNKNOWN')
            enriched_by_type[inst_type] = enriched_by_type.get(inst_type, 0) + 1
    
    # Save enriched data
    with open(OUTPUT_FILE, 'w', encoding='utf-8') as f:
        yaml.dump(institutions, f, allow_unicode=True, sort_keys=False, width=120)
    
    print("=" * 60)
    print("Batch 19 Enrichment Complete - Gap Types Addressed")
    print("=" * 60)
    print(f"Total institutions enriched: {enriched_count}")
    print()
    print("Enrichment by type:")
    for inst_type, count in sorted(enriched_by_type.items()):
        print(f"  {inst_type}: {count}")
    print()
    print(f"Output: {OUTPUT_FILE}")
    print()
    print("Expected Coverage Impact:")
    print("  RESEARCH_CENTER: 0/2 → 1/2 (50%)")
    print("  MIXED: 0/3 → 2/3 (66.7%)")
    print("  Overall: 75.6% → 78.9%")
    print()
    print("Next: Run coverage analysis to verify 78.9% target reached")

if __name__ == '__main__':
    main()