Update app.py

app.py

@@ -100,7 +100,7 @@ TAM_HEADER = Pt(9)
 TAM_TINY = Pt(8)
 
 # Página A4
-MARGEM_SUP = Cm(5.0)   # v4.1: aumentado para 5.0cm — mais respiro header↔corpo
+MARGEM_SUP = Cm(6.0)   # 6cm — espaço uniforme header↔corpo em todas as páginas
 MARGEM_INF = Cm(3.0)
 MARGEM_ESQ = Cm(2.5)
 MARGEM_DIR = Cm(2.0)
@@ -153,6 +153,35 @@ def fmt(p, text, bold=False, italic=False, size=None, color=None, font=FONTE, ca
     if caps: run.font.all_caps = True
     return run
 
+
+def _add_highlight_to_run(run, highlight_color="cyan"):
+    """Aplica highlight (fundo colorido) a um run via XML."""
+    rPr = run._r.get_or_add_rPr()
+    rPr.append(parse_xml(f'<w:highlight {nsdecls("w")} w:val="{highlight_color}"/>'))
+
+
+def fmt_with_meta_badges(p, text, size=None, color="1A1A1A", font=FONTE):
+    """Renderiza texto com badges visuais para Metas OMS (Meta 1..Meta 6).
+    Metas ficam em bold + highlight cyan para destaque visual em auditoria."""
+    import re
+    pattern = re.compile(r'(Meta\s+[1-6])', re.IGNORECASE)
+    parts = pattern.split(text)
+    for part in parts:
+        if pattern.match(part):
+            # Badge: bold + highlight
+            run = p.add_run(part)
+            run.bold = True
+            run.font.name = font
+            if size: run.font.size = size
+            run.font.color.rgb = RGBColor.from_string("1A3A6E")
+            _add_highlight_to_run(run, "cyan")
+        else:
+            if part:
+                run = p.add_run(part)
+                run.font.name = font
+                if size: run.font.size = size
+                if color: run.font.color.rgb = RGBColor.from_string(color)
+
 def spacing(p, before=0, after=0, line=1.15):
     pf = p.paragraph_format
     pf.space_before = Pt(before); pf.space_after = Pt(after); pf.line_spacing = line
@@ -260,17 +289,26 @@ def build_footer(section, meta, pal):
         else:
             fmt(c.paragraphs[0], "________________", size=TAM_SMALL, color="AAAAAA")
 
-    # Página
+    # Página X de Y
     pp = footer.add_paragraph()
     pp.alignment = WD_ALIGN_PARAGRAPH.RIGHT
     spacing(pp, 2, 0)
     fmt(pp, "Página ", size=TAM_TINY, color="888888")
+    # Campo PAGE (número atual)
     for x in [f'<w:fldChar {nsdecls("w")} w:fldCharType="begin"/>',
               f'<w:instrText {nsdecls("w")} xml:space="preserve"> PAGE </w:instrText>',
               f'<w:fldChar {nsdecls("w")} w:fldCharType="separate"/>']:
         pp.add_run()._r.append(parse_xml(x))
     r = pp.add_run("1"); r.font.size = TAM_TINY; r.font.color.rgb = RGBColor.from_string("888888")
     pp.add_run()._r.append(parse_xml(f'<w:fldChar {nsdecls("w")} w:fldCharType="end"/>'))
+    # " de " + NUMPAGES (total)
+    fmt(pp, " de ", size=TAM_TINY, color="888888")
+    for x in [f'<w:fldChar {nsdecls("w")} w:fldCharType="begin"/>',
+              f'<w:instrText {nsdecls("w")} xml:space="preserve"> NUMPAGES </w:instrText>',
+              f'<w:fldChar {nsdecls("w")} w:fldCharType="separate"/>']:
+        pp.add_run()._r.append(parse_xml(x))
+    r2 = pp.add_run("1"); r2.font.size = TAM_TINY; r2.font.color.rgb = RGBColor.from_string("888888")
+    pp.add_run()._r.append(parse_xml(f'<w:fldChar {nsdecls("w")} w:fldCharType="end"/>'))
     fmt(pp, f"  |  {meta.get('codigo','POP-XXX-001')}  |  v{meta.get('versao','01')}",
         size=TAM_TINY, color="888888")
 
@@ -308,13 +346,18 @@ def add_bullet(doc, text, bold_prefix=None, indent=IND_ITEM, pal=None):
     else:
         fmt(p, text, size=TAM_CORPO, color="1A1A1A")
 
-def add_num(doc, n, text, indent=IND_SUBITEM, pal=None):
+def add_num(doc, n, text, indent=IND_SUBITEM, pal=None, critico=False):
     c = pal["primary"] if pal else DEFAULT_PRIMARY
     p = doc.add_paragraph(); p.alignment = WD_ALIGN_PARAGRAPH.JUSTIFY
     spacing(p, 1, 4, 1.15)
     p.paragraph_format.left_indent = Cm(indent); p.paragraph_format.first_line_indent = Cm(-0.5)
-    fmt(p, f"{n}.  ", bold=True, size=TAM_CORPO, color=c)
-    fmt(p, text, size=TAM_CORPO, color="1A1A1A")
+    if critico:
+        fmt(p, f"\u26a0  ", bold=True, size=TAM_CORPO, color="C05000")
+        fmt(p, f"{n}. [PASSO CR\u00cdTICO]  ", bold=True, size=TAM_CORPO, color="C05000")
+    else:
+        fmt(p, f"{n}.  ", bold=True, size=TAM_CORPO, color=c)
+    # Texto com badges de Meta OMS
+    fmt_with_meta_badges(p, text, size=TAM_CORPO)
 
 def add_def_item(doc, termo, defn, pal):
     p = doc.add_paragraph(); p.alignment = WD_ALIGN_PARAGRAPH.JUSTIFY
@@ -520,27 +563,46 @@ def gerar_diagrama_png(tipo, conteudo, titulo, pal):
         edge_style = {'color': f'#{pal["primary"]}', 'penwidth': '1.0', 'arrowsize': '0.6'}
 
         if tipo in ("processo", "fluxograma"):
-            etapas = conteudo if isinstance(conteudo, list) else [conteudo]
-            etapas = etapas[:9]  # Max 9 steps (3 rows of 3)
-            n = len(etapas)
+            etapas_raw = conteudo if isinstance(conteudo, list) else [conteudo]
+            etapas_raw = etapas_raw[:9]
+            n = len(etapas_raw)
+
+            # Mapear formas: elipse→ellipse, losango→diamond, retangulo→box
+            forma_map = {'elipse': 'ellipse', 'losango': 'diamond', 'retangulo': 'box',
+                         'ellipse': 'ellipse', 'diamond': 'diamond', 'box': 'box'}
+
+            def _parse_etapa(et, idx, total):
+                """Retorna (texto, forma) de um item do fluxograma."""
+                if isinstance(et, dict):
+                    txt = et.get('texto', et.get('titulo', str(et)))[:30]
+                    frm = forma_map.get(et.get('forma', 'retangulo'), 'box')
+                    return txt, frm
+                txt = str(et)[:30]
+                # Auto-forma: primeiro=elipse, último=elipse, demais=box
+                if idx == 0 or idx == total - 1:
+                    return txt, 'ellipse'
+                # Detectar decisões por "?" no texto
+                if '?' in str(et):
+                    return txt, 'diamond'
+                return txt, 'box'
+
             if n <= 3:
-                # Simple horizontal
                 dot.attr(rankdir='LR', size='8,2!')
-                for i, et in enumerate(etapas):
-                    st = node_style if i % 2 == 0 else node_light
-                    dot.node(f'n{i}', et[:30], **st)
+                for i, et in enumerate(etapas_raw):
+                    txt, frm = _parse_etapa(et, i, n)
+                    st = {**(node_style if i % 2 == 0 else node_light), 'shape': frm}
+                    dot.node(f'n{i}', txt, **st)
                     if i > 0:
                         dot.edge(f'n{i-1}', f'n{i}', **edge_style)
             else:
-                # Zigzag: rows of 3, TB layout
                 dot.attr(rankdir='TB', size='8,6!')
                 per_row = 3
-                for i, et in enumerate(etapas):
-                    st = node_style if i % 2 == 0 else node_light
-                    dot.node(f'n{i}', et[:30], **st)
+                for i, et in enumerate(etapas_raw):
+                    txt, frm = _parse_etapa(et, i, n)
+                    st = {**(node_style if i % 2 == 0 else node_light), 'shape': frm}
+                    dot.node(f'n{i}', txt, **st)
                     if i > 0:
                         dot.edge(f'n{i-1}', f'n{i}', **edge_style)
-                # Group nodes into ranks (rows of 3)
                 for row_start in range(0, n, per_row):
                     row_nodes = ' '.join(f'n{i}' for i in range(row_start, min(row_start + per_row, n)))
                     dot.body.append(f'  {{ rank=same; {row_nodes} }}')
@@ -631,19 +693,29 @@ def _inserir_diagrama_no_corpo(doc, img_path, legenda="", pal=None):
 
 
 def _render_passo(doc, num, item, pal):
-    """Renderiza um item de procedimento: string (texto) ou dict (diagrama inline).
-    Aceita: {"diagrama": {"tipo":..., "titulo":..., "conteudo":...}}
-    Ou:     {"tipo":..., "titulo":..., "conteudo":...}
+    """Renderiza um item de procedimento: string, dict com texto/prioridade, ou dict com diagrama.
+    Formatos aceitos:
+      - "texto simples"
+      - {"texto": "...", "prioridade": "critica"}
+      - {"diagrama": {"tipo":..., "titulo":..., "conteudo":...}}
+      - {"tipo":..., "titulo":..., "conteudo":...}
     """
     if isinstance(item, str):
         add_num(doc, num, item, pal=pal)
     elif isinstance(item, dict):
-        # Extrair dados do diagrama (com ou sem wrapper "diagrama")
+        # Formato 1: {"texto": "...", "prioridade": "critica"}
+        if "texto" in item and "diagrama" not in item and "tipo" not in item:
+            texto = item.get("texto", "")
+            critico = item.get("prioridade", "").lower() == "critica"
+            add_num(doc, num, texto, pal=pal, critico=critico)
+            return
+
+        # Formato 2: diagrama (com ou sem wrapper "diagrama")
         diag = item.get("diagrama", None)
         if diag is None and "tipo" in item:
-            diag = item  # Formato direto sem wrapper
+            diag = item
         if diag is None:
-            # Dict sem formato reconhecido — renderizar como texto
+            # Dict não reconhecido — texto fallback
             add_num(doc, num, str(item), pal=pal)
             return
         tipo = diag.get("tipo", "processo")
@@ -653,7 +725,6 @@ def _render_passo(doc, num, item, pal):
         if img_path:
             _inserir_diagrama_no_corpo(doc, img_path, titulo, pal)
         else:
-            # Fallback: texto descritivo
             items_txt = ', '.join(str(c) for c in conteudo) if isinstance(conteudo, list) else str(conteudo)
             add_num(doc, num, f"[{titulo}]: {items_txt}", pal=pal)
 
@@ -961,6 +1032,8 @@ def gerar_pop_docx(json_str, primary_color=None, logo_bytes=None, palette_overri
     # Seções
     sn = 1
     add_h1(doc, sn, "OBJETIVO / FINALIDADE", pal)
+    # Remover space_before do primeiro H1 (margem superior já cuida do espaço)
+    doc.paragraphs[0].paragraph_format.space_before = Pt(0)
     add_body(doc, secoes.get("objetivo", "[Objetivo não fornecido]"))
 
     sn += 1; add_h1(doc, sn, "CAMPO DE APLICAÇÃO / ÁREA", pal)
@@ -998,7 +1071,25 @@ def gerar_pop_docx(json_str, primary_color=None, logo_bytes=None, palette_overri
         if isinstance(it, dict): add_risk(doc, it.get("risco",""), it.get("barreira",""), pal)
         else: add_bullet(doc, str(it), pal=pal)
     add_h2(doc, f"{sn}.2", "Plano de Contingência", pal)
-    add_body(doc, riscos.get("contingencia","[Contingência não fornecida]"), indent=IND_SUBITEM)
+    # Caixa de destaque com borda lateral vermelha/laranja
+    cont_text = riscos.get("contingencia","[Contingência não fornecida]")
+    cont_tbl = doc.add_table(rows=1, cols=1)
+    cont_tbl.alignment = WD_TABLE_ALIGNMENT.CENTER
+    cc = cont_tbl.cell(0, 0)
+    set_width(cc, LARGURA_DXA - 200)
+    set_margins(cc, 80, 80, 200, 200)
+    # Borda esquerda laranja grossa, demais finas cinza
+    set_borders(cc, ("E87D00","12"), (pal["gray_border"],"1"), (pal["gray_border"],"1"), (pal["gray_border"],"1"))
+    set_shading(cc, "FFF8F0")
+    # Ícone + título
+    pt = cc.paragraphs[0]
+    pt.paragraph_format.space_after = Pt(6)
+    fmt(pt, "\u26a0  ATENÇÃO — Plano de Contingência", bold=True, size=TAM_CORPO, color="C05000")
+    # Texto
+    pb = cc.add_paragraph()
+    pb.alignment = WD_ALIGN_PARAGRAPH.JUSTIFY
+    pb.paragraph_format.line_spacing = 1.15
+    fmt(pb, cont_text, size=TAM_SMALL, color="1A1A1A")
 
     sn += 1; add_h1(doc, sn, "REGISTROS DA QUALIDADE (EVIDÊNCIAS)", pal)
     add_body(doc, "A execução deste procedimento e eventuais intercorrências devem ser "