"""Render only validated structured facts; never infer facts or read gold."""
from copy import deepcopy

from tools.graphrag.graphrag import WORK_TYPE_LABELS


def render_graph_answer(answer: dict, context: dict) -> dict:
    out = deepcopy(answer)
    lines = []
    if out['status'] == 'ok':
        labels = {'work_label': 'záznam', 'work_type': 'typ práce',
                  'title': 'názov práce', 'year': 'rok práce', 'start_year': 'začiatok štúdia',
                  'keyword': 'kľúčové slovo', 'category': 'kategória', 'document_author': 'autor dokumentu'}
        for claim in out['claims']:
            values = [f'{label}: {WORK_TYPE_LABELS.get(claim[key], claim[key]) if key == "work_type" else claim[key]}'
                      for key, label in labels.items() if claim.get(key) is not None]
            if claim.get('record_id') and not claim.get('title'):
                values.append('názov v zdroji nie je uvedený')
            if claim.get('record_id') and claim.get('year') is None:
                values.append('rok v zdroji nie je uvedený')
            lines.append(claim['person'] + (' — ' + '; '.join(values) if values else '') + '.')
        if out.get('count') is not None:
            lines.append(f'Počet vyhovujúcich osôb: {out["count"]}.')
        if context.get('complete') is False or context.get('truncated'):
            lines.append('Výpis je neúplný.')
        elif context.get('complete') is None:
            lines.append('Úplnosť vyhľadania nemožno zaručiť.')
    elif out['status'] == 'ambiguous':
        lines.append('Požiadavka je nejednoznačná; údaje nemožno spoľahlivo určiť.')
    elif out['status'] == 'error':
        lines.append('Vyhľadanie zlyhalo; odpoveď nemožno pripraviť.')
    else:
        lines.append('Požadované údaje sa v dostupných podkladoch nenašli.')
    if out['source_urls']:
        lines.append('Zdroje: ' + ', '.join(out['source_urls']))
    out['answer'] = '\n'.join(lines)
    return out