Update app.py

app.py

@@ -13,6 +13,11 @@ from google.genai import types
 # ─── 1. CONFIGURAÇÃO DA PÁGINA E ESTADO ───────────────────────────────────
 st.set_page_config(page_title="Corretor TCC - UNIFAGOC", page_icon="⚖️", layout="wide")
 
+# Limpa o cache se o usuário trocar de arquivo, evitando downloads cruzados
+def reset_estado(chave):
+    if chave in st.session_state:
+        st.session_state[chave] = None
+
 if "resultado_auditoria" not in st.session_state: st.session_state.resultado_auditoria = None
 if "doc_revisado_bytes" not in st.session_state: st.session_state.doc_revisado_bytes = None
 if "doc_formatado_bytes" not in st.session_state: st.session_state.doc_formatado_bytes = None
@@ -56,13 +61,13 @@ class AnaliseABNT(BaseModel):
 def get_client():
     api_key = os.environ.get("GEMINI_API_KEY")
     if not api_key:
-        st.error("❌ ERRO: A variável de ambiente GEMINI_API_KEY não foi encontrada nos Secrets do Hugging Face.")
+        st.error("❌ ERRO: A variável de ambiente GEMINI_API_KEY não foi encontrada nos Secrets.")
         st.stop()
     return genai.Client(api_key=api_key)
 
 client = get_client()
 
-# ─── 4. MANIPULAÇÃO DE DOCUMENTOS (DOCX) ──────────────────────────────────
+# ─── 4. MANIPULAÇÃO AVANÇADA DE DOCUMENTOS (DOCX) ─────────────────────────
 def extrair_texto_docx(arquivo_upload) -> str:
     doc = Document(arquivo_upload)
     return "\n\n".join([para.text.strip() for para in doc.paragraphs if para.text.strip()])
@@ -79,239 +84,124 @@ def estimar_paginas(doc: Document) -> int:
     )
     return max(1, palavras_totais // 350)
 
-def _set_run_font(run, nome: str, tamanho_pt: float,
-                  negrito: bool = None, italico: bool = None):
-    """Aplica fonte, tamanho e atributos opcionais a um run."""
+def _set_run_font(run, nome: str, tamanho_pt: float, negrito: bool = None, italico: bool = None):
     run.font.name = nome
     run.font.size = Pt(tamanho_pt)
-    if negrito is not None:
-        run.font.bold = negrito
-    if italico is not None:
-        run.font.italic = italico
+    if negrito is not None: run.font.bold = negrito
+    if italico is not None: run.font.italic = italico
+
+def substituir_texto_cross_run(para, original, corrigido) -> int:
+    """Substitui textos mesmo quando fragmentados em múltiplos runs pelo Word."""
+    if original not in para.text: return 0
+    
+    # Se está num único run (caso simples)
+    for run in para.runs:
+        if original in run.text:
+            run.text = run.text.replace(original, corrigido, 1)
+            return 1
+            
+    # Se está fragmentado (cross-run) - Preserva o estilo do primeiro run afetado
+    estilo_base = para.runs[0].style if para.runs else None
+    texto_completo = para.text.replace(original, corrigido, 1)
+    para.clear()
+    novo_run = para.add_run(texto_completo)
+    if estilo_base: novo_run.style = estilo_base
+    return 1
 
+def formatacao_cruzada_indices(para, texto_alvo, aplicar_italico=False, aplicar_negrito=False) -> int:
+    aplicacoes = 0
+    if aplicar_italico:
+        matches = list(re.finditer(r'\b' + re.escape(texto_alvo) + r'\b', para.text, re.IGNORECASE))
+    else:
+        idx = para.text.find(texto_alvo)
+        matches = [(idx, idx + len(texto_alvo))] if idx != -1 else []
+
+    for match in matches:
+        pos_ini, pos_fim = match if isinstance(match, tuple) else (match.start(), match.end())
+        if pos_ini == -1: continue
+        pos_atual = 0
+        for run in para.runs:
+            fim_run = pos_atual + len(run.text)
+            if pos_atual < pos_fim and fim_run > pos_ini:
+                if aplicar_italico: run.italic = True
+                if aplicar_negrito: run.bold = True
+                aplicacoes += 1
+            pos_atual += len(run.text)
+    return aplicacoes
 
 def aplicar_formatacao_unifagoc_abnt(doc: Document):
-    """
-    Aplica a formatação exata do template UNIFAGOC / ABNT 6023:2018:
-
-    PÁGINA
-      • A4 (21 × 29,7 cm)
-      • Margens: superior 3 cm | esquerda 3 cm | inferior 2 cm | direita 2 cm
-      • Cabeçalho / rodapé: 0,5 cm das bordas
-
-    CORPO DO TEXTO
-      • Fonte: Calibri 12 pt
-      • Alinhamento: justificado
-      • Recuo de primeira linha: 1,25 cm
-      • Espaçamento entre linhas: 1,5
-      • Espaço antes/depois: 0 pt
-
-    TÍTULOS DAS SEÇÕES PRIMÁRIAS (caixa-alta, estilo Heading 2)
-      • Calibri 12 pt, negrito, centralizado
-      • Espaço antes: 0 pt | depois: 12 pt | entre linhas: 1,5
-
-    TÍTULOS SECUNDÁRIOS (estilo Heading 3)
-      • Calibri 12 pt, negrito, justificado
-      • Espaço antes: 12 pt | depois: 0 pt | entre linhas: 1,5
-
-    TÍTULOS TERCIÁRIOS (estilo Heading 4)
-      • Calibri 12 pt, negrito + itálico, justificado
-      • Espaço antes: 0 pt | depois: 0 pt | entre linhas: 1,5
-
-    RESUMO / ABSTRACT
-      • Calibri 11 pt, itálico (rótulo "RESUMO" e "Palavras-chave" em negrito sem itálico)
-      • Alinhamento: justificado | sem recuo | entre linhas: 1,5
-
-    CITAÇÕES LONGAS (> 4 linhas)
-      • Calibri 11 pt, justificado
-      • Recuo esquerdo: 4 cm | sem recuo de primeira linha
-      • Espaçamento simples (1,0) | 6 pt antes e depois
-
-    REFERÊNCIAS (linhas individuais)
-      • Calibri 11 pt, alinhamento esquerdo
-      • Sem recuo | espaçamento simples (1,0) | 6 pt depois (entre entradas)
-      • Espaço antes: 0 pt
-    """
-
-    # ── Página ───────────────────────────────────────────────────────────────
     for section in doc.sections:
-        section.page_height     = Cm(29.7)
-        section.page_width      = Cm(21.0)
-        section.top_margin      = Cm(3.0)
-        section.left_margin     = Cm(3.0)
-        section.bottom_margin   = Cm(2.0)
-        section.right_margin    = Cm(2.0)
-        section.header_distance = Cm(0.5)
-        section.footer_distance = Cm(0.5)
-
-    em_referencias = False
-    em_resumo      = False
+        section.page_height, section.page_width = Cm(29.7), Cm(21.0)
+        section.top_margin, section.left_margin = Cm(3.0), Cm(3.0)
+        section.bottom_margin, section.right_margin = Cm(2.0), Cm(2.0)
+        section.header_distance, section.footer_distance = Cm(0.5), Cm(0.5)
+
+    em_referencias, em_resumo = False, False
 
     for para in doc.paragraphs:
         texto = para.text.strip()
-        if not texto:
-            continue
+        if not texto: continue
 
-        pf         = para.paragraph_format
-        style_name = (para.style.name or "").lower()
+        pf, style_name = para.paragraph_format, (para.style.name or "").lower()
 
-        # ── Detecção de seções-chave ─────────────────────────────────────────
         if re.match(r"^RESUMO\b", texto, re.IGNORECASE):
-            em_resumo      = True
-            em_referencias = False
+            em_resumo, em_referencias = True, False
         elif re.match(r"^(?:INTRODU[ÇC]|ABSTRACT|1[\s\.])", texto, re.IGNORECASE):
             em_resumo = False
 
         is_titulo_refs = bool(re.search(r"^(?:\d+\.?\s*)?REFER[ÊE]NCIAS\b", texto, re.IGNORECASE))
-        if is_titulo_refs:
-            em_referencias = True
+        if is_titulo_refs: em_referencias = True
 
-        # ─────────────────────────────────────────────────────────────────────
-        # A) REFERÊNCIAS – corpo (entradas individuais)
-        # ─────────────────────────────────────────────────────────────────────
+        # REFERÊNCIAS (Entradas)
         if em_referencias and not is_titulo_refs:
-            pf.alignment         = WD_ALIGN_PARAGRAPH.LEFT
-            pf.first_line_indent = Cm(0)
-            pf.left_indent       = Cm(0)
-            pf.space_before      = Pt(0)
-            pf.space_after       = Pt(6)
-            pf.line_spacing      = 1.0          # simples
-            for run in para.runs:
-                _set_run_font(run, "Calibri", 11)
+            pf.alignment, pf.first_line_indent, pf.left_indent = WD_ALIGN_PARAGRAPH.LEFT, Cm(0), Cm(0)
+            pf.space_before, pf.space_after, pf.line_spacing = Pt(0), Pt(6), 1.0
+            for run in para.runs: _set_run_font(run, "Calibri", 11)
 
-        # ─────────────────────────────────────────────────────────────────────
-        # B) TÍTULO "REFERÊNCIAS"
-        # ─────────────────────────────────────────────────────────────────────
+        # TÍTULO "REFERÊNCIAS"
         elif is_titulo_refs:
-            pf.alignment         = WD_ALIGN_PARAGRAPH.CENTER
-            pf.first_line_indent = Cm(0)
-            pf.left_indent       = Cm(0)
-            pf.space_before      = Pt(12)
-            pf.space_after       = Pt(12)
-            pf.line_spacing      = 1.5
-            for run in para.runs:
-                _set_run_font(run, "Calibri", 12, negrito=True, italico=False)
+            pf.alignment, pf.first_line_indent, pf.left_indent = WD_ALIGN_PARAGRAPH.CENTER, Cm(0), Cm(0)
+            pf.space_before, pf.space_after, pf.line_spacing = Pt(12), Pt(12), 1.5
+            for run in para.runs: _set_run_font(run, "Calibri", 12, negrito=True, italico=False)
 
-        # ─────────────────────────────────────────────────────────────────────
-        # C) RESUMO / ABSTRACT
-        # ─────────────────────────────────────────────────────────────────────
+        # RESUMO
         elif em_resumo:
-            pf.alignment         = WD_ALIGN_PARAGRAPH.JUSTIFY
-            pf.first_line_indent = Cm(0)
-            pf.left_indent       = Cm(0)
-            pf.space_before      = Pt(0)
-            pf.space_after       = Pt(0)
-            pf.line_spacing      = 1.5
+            pf.alignment, pf.first_line_indent, pf.left_indent = WD_ALIGN_PARAGRAPH.JUSTIFY, Cm(0), Cm(0)
+            pf.space_before, pf.space_after, pf.line_spacing = Pt(0), Pt(0), 1.5
             for run in para.runs:
-                is_label = bool(re.match(
-                    r"^(RESUMO|PALAVRAS.CHAVE|ABSTRACT|KEYWORDS)",
-                    run.text.strip(), re.IGNORECASE
-                ))
-                if is_label:
-                    _set_run_font(run, "Calibri", 11, negrito=True, italico=False)
-                else:
-                    _set_run_font(run, "Calibri", 11, negrito=False, italico=True)
+                is_label = bool(re.match(r"^(RESUMO|PALAVRAS.CHAVE|ABSTRACT|KEYWORDS)", run.text.strip(), re.IGNORECASE))
+                _set_run_font(run, "Calibri", 11, negrito=is_label, italico=not is_label)
 
-        # ─────────────────────────────────────────────────────────────────────
-        # D) TÍTULOS DE SEÇÕES – detectados por estilo Word ou caixa-alta curta
-        # ─────────────────────────────────────────────────────────────────────
+        # SEÇÕES (Títulos e Subtítulos)
         elif "heading" in style_name or (texto.isupper() and 3 < len(texto) < 120):
-            # Heading 2 → seção primária (centralizado, maiúsculas)
-            if "heading 2" in style_name or "heading2" in style_name or (
-                texto.isupper() and "heading" not in style_name
-            ):
-                pf.alignment         = WD_ALIGN_PARAGRAPH.CENTER
-                pf.first_line_indent = Cm(0)
-                pf.left_indent       = Cm(0)
-                pf.space_before      = Pt(0)
-                pf.space_after       = Pt(12)
-                pf.line_spacing      = 1.5
-                for run in para.runs:
-                    _set_run_font(run, "Calibri", 12, negrito=True, italico=False)
-
-            # Heading 3 → seção secundária (justificado, negrito)
+            if "heading 2" in style_name or "heading2" in style_name or (texto.isupper() and "heading" not in style_name):
+                pf.alignment, pf.first_line_indent, pf.left_indent = WD_ALIGN_PARAGRAPH.CENTER, Cm(0), Cm(0)
+                pf.space_before, pf.space_after, pf.line_spacing = Pt(0), Pt(12), 1.5
+                for run in para.runs: _set_run_font(run, "Calibri", 12, negrito=True, italico=False)
             elif "heading 3" in style_name or "heading3" in style_name:
-                pf.alignment         = WD_ALIGN_PARAGRAPH.JUSTIFY
-                pf.first_line_indent = Cm(0)
-                pf.left_indent       = Cm(0)
-                pf.space_before      = Pt(12)
-                pf.space_after       = Pt(0)
-                pf.line_spacing      = 1.5
-                for run in para.runs:
-                    _set_run_font(run, "Calibri", 12, negrito=True, italico=False)
-
-            # Heading 4 → seção terciária (justificado, negrito + itálico)
+                pf.alignment, pf.first_line_indent, pf.left_indent = WD_ALIGN_PARAGRAPH.JUSTIFY, Cm(0), Cm(0)
+                pf.space_before, pf.space_after, pf.line_spacing = Pt(12), Pt(0), 1.5
+                for run in para.runs: _set_run_font(run, "Calibri", 12, negrito=True, italico=False)
             elif "heading 4" in style_name or "heading4" in style_name:
-                pf.alignment         = WD_ALIGN_PARAGRAPH.JUSTIFY
-                pf.first_line_indent = Cm(0)
-                pf.left_indent       = Cm(0)
-                pf.space_before      = Pt(0)
-                pf.space_after       = Pt(0)
-                pf.line_spacing      = 1.5
-                for run in para.runs:
-                    _set_run_font(run, "Calibri", 12, negrito=True, italico=True)
-
-            # Heading 1 → título principal do artigo (centralizado, maior)
+                pf.alignment, pf.first_line_indent, pf.left_indent = WD_ALIGN_PARAGRAPH.JUSTIFY, Cm(0), Cm(0)
+                pf.space_before, pf.space_after, pf.line_spacing = Pt(0), Pt(0), 1.5
+                for run in para.runs: _set_run_font(run, "Calibri", 12, negrito=True, italico=True)
             else:
-                pf.alignment         = WD_ALIGN_PARAGRAPH.CENTER
-                pf.first_line_indent = Cm(0)
-                pf.left_indent       = Cm(0)
-                pf.space_before      = Pt(6)
-                pf.space_after       = Pt(12)
-                pf.line_spacing      = 1.5
-                for run in para.runs:
-                    _set_run_font(run, "Calibri", 14, negrito=True, italico=False)
-
-        # ─────────────────────────────────────────────────────────────────────
-        # E) CITAÇÕES LONGAS – recuo ≥ 3,5 cm (bloco com recuo existente)
-        # ─────────────────────────────────────────────────────────────────────
+                pf.alignment, pf.first_line_indent, pf.left_indent = WD_ALIGN_PARAGRAPH.CENTER, Cm(0), Cm(0)
+                pf.space_before, pf.space_after, pf.line_spacing = Pt(6), Pt(12), 1.5
+                for run in para.runs: _set_run_font(run, "Calibri", 14, negrito=True, italico=False)
+
+        # CITAÇÕES LONGAS
         elif pf.left_indent is not None and pf.left_indent >= Cm(3.5):
-            pf.alignment         = WD_ALIGN_PARAGRAPH.JUSTIFY
-            pf.first_line_indent = Cm(0)
-            pf.left_indent       = Cm(4.0)
-            pf.space_before      = Pt(6)
-            pf.space_after       = Pt(6)
-            pf.line_spacing      = 1.0          # simples
-            for run in para.runs:
-                _set_run_font(run, "Calibri", 11)
+            pf.alignment, pf.first_line_indent, pf.left_indent = WD_ALIGN_PARAGRAPH.JUSTIFY, Cm(0), Cm(4.0)
+            pf.space_before, pf.space_after, pf.line_spacing = Pt(6), Pt(6), 1.0
+            for run in para.runs: _set_run_font(run, "Calibri", 11)
 
-        # ─────────────────────────────────────────────────────────────────────
-        # F) CORPO DO TEXTO (padrão)
-        # ─────────────────────────────────────────────────────────────────────
+        # CORPO DO TEXTO
         else:
-            pf.alignment         = WD_ALIGN_PARAGRAPH.JUSTIFY
-            pf.first_line_indent = Cm(1.25)
-            pf.left_indent       = Cm(0)
-            pf.space_before      = Pt(0)
-            pf.space_after       = Pt(0)
-            pf.line_spacing      = 1.5
-            for run in para.runs:
-                _set_run_font(run, "Calibri", 12)
-
-
-def formatacao_cruzada_indices(para, texto_alvo, aplicar_italico=False, aplicar_negrito=False) -> int:
-    aplicacoes = 0
-    if aplicar_italico:
-        matches = list(re.finditer(r'\b' + re.escape(texto_alvo) + r'\b', para.text, re.IGNORECASE))
-    else:
-        idx = para.text.find(texto_alvo)
-        matches = [(idx, idx + len(texto_alvo))] if idx != -1 else []
-
-    for match in matches:
-        pos_ini, pos_fim = match if isinstance(match, tuple) else (match.start(), match.end())
-        if pos_ini == -1:
-            continue
-        pos_atual = 0
-        for run in para.runs:
-            fim_run = pos_atual + len(run.text)
-            if pos_atual < pos_fim and fim_run > pos_ini:
-                if aplicar_italico:
-                    run.italic = True
-                if aplicar_negrito:
-                    run.bold = True
-                aplicacoes += 1
-            pos_atual += len(run.text)
-    return aplicacoes
+            pf.alignment, pf.first_line_indent, pf.left_indent = WD_ALIGN_PARAGRAPH.JUSTIFY, Cm(1.25), Cm(0)
+            pf.space_before, pf.space_after, pf.line_spacing = Pt(0), Pt(0), 1.5
+            for run in para.runs: _set_run_font(run, "Calibri", 12)
 
 # ─── 5. INTERFACE PRINCIPAL (ABAS) ────────────────────────────────────────
 st.title("⚖️ Assistente Acadêmico - Direito UNIFAGOC")
@@ -326,15 +216,11 @@ aba_auditoria, aba_revisao, aba_abnt = st.tabs([
 # ==============================================================================
 with aba_auditoria:
     st.markdown("Cruze o **Pré-Projeto** com o **TCC** para identificar fuga do tema e gerar a **Nota da Banca (80 pontos)**.")
-    fase_tcc = st.radio(
-        "Fase atual:",
-        ["TCC 1 (Apenas Introdução e Capítulo 1)", "TCC 2 (TCC Completo)"],
-        horizontal=True
-    )
+    fase_tcc = st.radio("Fase atual:", ["TCC 1 (Apenas Introdução e Capítulo 1)", "TCC 2 (TCC Completo)"], horizontal=True, on_change=reset_estado, args=("resultado_auditoria",))
 
     col1, col2 = st.columns(2)
-    with col1: arq_projeto = st.file_uploader("1. Pré-Projeto (.docx)", type=["docx"], key="proj")
-    with col2: arq_tcc     = st.file_uploader("2. TCC (.docx)", type=["docx"], key="tcc_aud")
+    with col1: arq_projeto = st.file_uploader("1. Pré-Projeto (.docx)", type=["docx"], key="proj", on_change=reset_estado, args=("resultado_auditoria",))
+    with col2: arq_tcc     = st.file_uploader("2. TCC (.docx)", type=["docx"], key="tcc_aud", on_change=reset_estado, args=("resultado_auditoria",))
 
     if st.button("🔍 Iniciar Auditoria e Avaliação", type="primary"):
         if not arq_projeto or not arq_tcc:
@@ -345,15 +231,10 @@ with aba_auditoria:
                 texto_projeto = extrair_texto_docx(arq_projeto)
                 texto_tcc     = extrair_texto_docx(arq_tcc)
 
-                regra_fase = (
-                    "ATENÇÃO: O aluno está em TCC 1. NÃO penalize a ausência de Resumo e Conclusão."
-                    if "TCC 1" in fase_tcc
-                    else "O aluno está em TCC 2. Exija o trabalho completo."
-                )
+                regra_fase = "ATENÇÃO: O aluno está em TCC 1. NÃO penalize a ausência de Resumo e Conclusão." if "TCC 1" in fase_tcc else "O aluno está em TCC 2. Exija o trabalho completo."
                 prompt = (
                     f"Como Professor Examinador titular da banca de Direito da UNIFAGOC:\n{regra_fase}\n"
-                    f"=== PRÉ-PROJETO ===\n{texto_projeto}\n"
-                    f"=== TCC ===\n{texto_tcc}\n"
+                    f"=== PRÉ-PROJETO ===\n{texto_projeto}\n=== TCC ===\n{texto_tcc}\n"
                     "TAREFA 1: Avalie o cruzamento de dados, fuga de tema e objetivos.\n"
                     "TAREFA 2: Distribua 80 pontos rigorosamente de acordo com os critérios da ficha UNIFAGOC."
                 )
@@ -361,13 +242,8 @@ with aba_auditoria:
                 st.write("A submeter dados à IA do Google...")
                 try:
                     res = client.models.generate_content(
-                        model="gemini-3.5-flash",
-                        contents=prompt,
-                        config=types.GenerateContentConfig(
-                            temperature=0.0,
-                            response_mime_type="application/json",
-                            response_schema=AuditoriaTCC
-                        )
+                        model="gemini-3.5-flash", contents=prompt,
+                        config=types.GenerateContentConfig(temperature=0.0, response_mime_type="application/json", response_schema=AuditoriaTCC)
                     )
                     st.session_state.resultado_auditoria = json.loads(res.text)
                     status.update(label="Avaliação da Banca concluída!", state="complete", expanded=False)
@@ -378,52 +254,38 @@ with aba_auditoria:
     if st.session_state.resultado_auditoria:
         dados = st.session_state.resultado_auditoria
         st.header("🏆 Avaliação da Banca UNIFAGOC")
-        nota_total = sum([
-            dados.get("nota_tema_problema", 0),
-            dados.get("nota_introducao_metodo", 0),
-            dados.get("nota_estruturacao", 0),
-            dados.get("nota_formatacao_abnt", 0)
-        ])
+        nota_total = sum([dados.get("nota_tema_problema", 0), dados.get("nota_introducao_metodo", 0), dados.get("nota_estruturacao", 0), dados.get("nota_formatacao_abnt", 0)])
 
         c1, c2, c3, c4 = st.columns(4)
-        c1.metric("Tema/Problema",        f"{dados.get('nota_tema_problema')}/20")
-        c2.metric("Intro/Método",          f"{dados.get('nota_introducao_metodo')}/20")
-        c3.metric("Estrutura/Linguagem",   f"{dados.get('nota_estruturacao')}/20")
-        c4.metric("Formatação/ABNT",       f"{dados.get('nota_formatacao_abnt')}/20")
-
-        st.info(
-            f"**NOTA FINAL TEXTUAL: {nota_total}/80 pontos**\n"
-            f"*Parecer da Banca:* {dados.get('justificativa_notas_banca')}"
-        )
+        c1.metric("Tema/Problema", f"{dados.get('nota_tema_problema')}/20")
+        c2.metric("Intro/Método", f"{dados.get('nota_introducao_metodo')}/20")
+        c3.metric("Estrutura/Linguagem", f"{dados.get('nota_estruturacao')}/20")
+        c4.metric("Formatação/ABNT", f"{dados.get('nota_formatacao_abnt')}/20")
+
+        st.info(f"**NOTA FINAL TEXTUAL: {nota_total}/80 pontos**\n*Parecer da Banca:* {dados.get('justificativa_notas_banca')}")
         st.divider()
 
         st.metric("Índice de Aderência ao Projeto", f"{dados.get('aderencia_percentual')}%")
         col_a, col_b = st.columns(2)
         with col_a:
             st.subheader("🎯 Problema de Pesquisa")
-            if dados.get("tcc_respondeu_ao_problema"):
-                st.success(f"✅ O TCC respondeu: {dados.get('justificativa_problema')}")
-            else:
-                st.error(f"❌ Não respondeu: {dados.get('justificativa_problema')}")
+            if dados.get("tcc_respondeu_ao_problema"): st.success(f"✅ O TCC respondeu: {dados.get('justificativa_problema')}")
+            else: st.error(f"❌ Não respondeu: {dados.get('justificativa_problema')}")
         with col_b:
             st.subheader("📋 Objetivos")
-            for f in dados.get("objetivos_nao_cumpridos_no_tcc", []):
-                st.error(f"Faltou: {f}")
-            if not dados.get("objetivos_nao_cumpridos_no_tcc"):
-                st.success("✅ Todos os objetivos foram abordados.")
+            for f in dados.get("objetivos_nao_cumpridos_no_tcc", []): st.error(f"Faltou: {f}")
+            if not dados.get("objetivos_nao_cumpridos_no_tcc"): st.success("✅ Todos os objetivos foram abordados.")
 
         st.subheader("🚨 Filtro Anti-Embromação")
-        for fuga in dados.get("desvios_de_tema_ou_linguica", []):
-            st.warning(f"Fuga: {fuga}")
-        for ia in dados.get("indicios_texto_generico_ia", []):
-            st.error(f"Indício de IA/Texto Genérico: {ia}")
+        for fuga in dados.get("desvios_de_tema_ou_linguica", []): st.warning(f"Fuga: {fuga}")
+        for ia in dados.get("indicios_texto_generico_ia", []): st.error(f"Indício de IA/Texto Genérico: {ia}")
 
 # ==============================================================================
 # ABA 2: REVISÃO TEXTUAL INTELIGENTE
 # ==============================================================================
 with aba_revisao:
     st.markdown("A IA analisa o contexto para corrigir **Ortografia**, **CAIXA ALTA** indevida e aplica **Itálico** em termos latinos/estrangeiros.")
-    arq_rev = st.file_uploader("Envie o TCC (.docx)", type=["docx"], key="revisao")
+    arq_rev = st.file_uploader("Envie o TCC (.docx)", type=["docx"], key="revisao", on_change=reset_estado, args=("doc_revisado_bytes",))
 
     if st.button("🧠 Realizar Revisão"):
         if arq_rev:
@@ -439,13 +301,8 @@ with aba_revisao:
                 st.write("A aguardar correções da Inteligência Artificial...")
                 try:
                     res = client.models.generate_content(
-                        model="gemini-3.5-flash",
-                        contents=prompt_rev,
-                        config=types.GenerateContentConfig(
-                            temperature=0.1,
-                            response_mime_type="application/json",
-                            response_schema=RevisaoInteligente
-                        )
+                        model="gemini-3.5-flash", contents=prompt_rev,
+                        config=types.GenerateContentConfig(temperature=0.1, response_mime_type="application/json", response_schema=RevisaoInteligente)
                     )
                     dados = json.loads(res.text)
 
@@ -456,11 +313,7 @@ with aba_revisao:
                     if dados.get("correcoes"):
                         for c in dados["correcoes"]:
                             for para in doc_rev.paragraphs:
-                                if c["original"] in para.text:
-                                    for run in para.runs:
-                                        if c["original"] in run.text:
-                                            run.text = run.text.replace(c["original"], c["corrigido"], 1)
-                                            aplicadas_texto += 1
+                                aplicadas_texto += substituir_texto_cross_run(para, c["original"], c["corrigido"])
 
                     if dados.get("termos_estrangeiros_para_italico"):
                         for termo in dados["termos_estrangeiros_para_italico"]:
@@ -468,10 +321,8 @@ with aba_revisao:
                                 aplicadas_italico += formatacao_cruzada_indices(para, termo, aplicar_italico=True)
 
                     st.session_state.doc_revisado_bytes = docx_to_bytes(doc_rev)
-                    status.update(
-                        label=f"Processo finalizado com {aplicadas_texto + aplicadas_italico} intervenções automáticas.",
-                        state="complete", expanded=False
-                    )
+                    st.session_state.nome_arq_rev = f"{arq_rev.name.replace('.docx', '')}_Revisado.docx"
+                    status.update(label=f"Processo finalizado com {aplicadas_texto + aplicadas_italico} intervenções automáticas.", state="complete", expanded=False)
                 except Exception as e:
                     status.update(label="Erro no processamento.", state="error")
                     st.error(f"Erro: {e}")
@@ -480,7 +331,7 @@ with aba_revisao:
         st.download_button(
             "⬇️ Baixar TCC Revisado (.docx)",
             data=st.session_state.doc_revisado_bytes,
-            file_name="TCC_Revisado.docx",
+            file_name=st.session_state.nome_arq_rev,
             mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
         )
 
@@ -488,101 +339,70 @@ with aba_revisao:
 # ABA 3: FORMATAÇÃO TOTAL ABNT (UNIFAGOC)
 # ==============================================================================
 with aba_abnt:
-    st.markdown(
-        "Motor de formatação **fiel ao template UNIFAGOC**. "
-        "Aplica: Calibri 12/11 pt · Margens 3×2 cm · Espaçamento 1,5 · "
-        "Recuo 1,25 cm · Referências simples + 6 pt · Cabeçalho/rodapé 0,5 cm."
-    )
-    arq_abnt = st.file_uploader("Envie o TCC (.docx)", type=["docx"], key="abnt_file")
+    st.markdown("Motor de formatação **fiel ao template UNIFAGOC**.")
+    arq_abnt = st.file_uploader("Envie o TCC (.docx)", type=["docx"], key="abnt_file", on_change=reset_estado, args=("doc_formatado_bytes",))
 
     if st.button("📐 Formatar TCC"):
         if arq_abnt:
             with st.status("A reestruturar formatação conforme template UNIFAGOC...", expanded=True) as status:
                 doc_abnt = Document(arq_abnt)
 
-                # ── Estimativa de páginas ────────────────────────────────────
                 paginas_estimadas = estimar_paginas(doc_abnt)
-                if paginas_estimadas > 18:
-                    st.warning(f"🚨 Extensão de {paginas_estimadas} páginas. O limite recomendado é 18.")
-                else:
-                    st.info(f"📄 Extensão estimada: {paginas_estimadas} páginas.")
+                if paginas_estimadas > 18: st.warning(f"🚨 Extensão de {paginas_estimadas} páginas. O limite recomendado é 18.")
+                else: st.info(f"📄 Extensão estimada: {paginas_estimadas} páginas.")
 
-                # ── Aplicação das regras físicas do template ─────────────────
                 st.write("A aplicar regras do template (Calibri, margens, recuos, espaçamentos)...")
                 aplicar_formatacao_unifagoc_abnt(doc_abnt)
 
-                # ── Extração limpa das referências bibliográficas ────────────
                 st.write("A analisar Referências Bibliográficas com IA...")
-                refs_limpas = []
-                em_refs = False
+                refs_limpas, em_refs = [], False
                 for i, para in enumerate(doc_abnt.paragraphs):
                     txt = para.text.strip()
-                    if not txt:
-                        continue
+                    if not txt: continue
                     if re.search(r"^(?:\d+\.?\s*)?REFER[ÊE]NCIAS", txt, re.IGNORECASE):
-                        em_refs = True
-                        continue
+                        em_refs = True; continue
                     if em_refs:
-                        # Interrompe se encontrar outro título de seção em caixa-alta
-                        if len(txt) < 120 and txt.isupper() and not txt[0].isdigit():
-                            break
+                        if len(txt) < 120 and txt.isupper() and not txt[0].isdigit(): break
                         refs_limpas.append((i, txt))
 
+                st.session_state.nome_arq_abnt = f"{arq_abnt.name.replace('.docx', '')}_Formatado.docx"
+
                 if refs_limpas:
                     texto_refs = "\n".join(f"[{idx}] {txt}" for idx, txt in refs_limpas)
                     prompt_abnt = (
-                        "Analise as referências (ABNT 6023:2018). "
-                        "Para cada uma, devolva o índice exato entre colchetes como 'numero', "
-                        "e extraia o trecho EXATO (título do livro ou nome da revista/periódico) "
-                        "que deve receber negrito.\n"
+                        "Analise as referências (ABNT 6023:2018). Para cada uma, devolva o índice exato entre colchetes como 'numero', "
+                        "e extraia o trecho EXATO (título do livro ou nome da revista/periódico) que deve receber negrito.\n"
                         f"REFERÊNCIAS:\n{texto_refs}"
                     )
                     try:
                         res = client.models.generate_content(
-                            model="gemini-3.5-flash",
-                            contents=prompt_abnt,
-                            config=types.GenerateContentConfig(
-                                temperature=0.0,
-                                response_mime_type="application/json",
-                                response_schema=AnaliseABNT
-                            )
+                            model="gemini-3.5-flash", contents=prompt_abnt,
+                            config=types.GenerateContentConfig(temperature=0.0, response_mime_type="application/json", response_schema=AnaliseABNT)
                         )
                         dados_abnt = json.loads(res.text)
 
-                        # Constrói mapa: índice-parágrafo → objeto parágrafo
-                        mapa_idx_para = {idx: doc_abnt.paragraphs[idx] for idx, _ in refs_limpas
-                                         if idx < len(doc_abnt.paragraphs)}
-
+                        mapa_idx_para = {idx: doc_abnt.paragraphs[idx] for idx, _ in refs_limpas if idx < len(doc_abnt.paragraphs)}
                         negs_aplicados = 0
+                        
                         for problema in dados_abnt.get("problemas", []):
-                            num = problema.get("numero")
-                            elem = problema.get("elemento_negrito", "").strip()
+                            num, elem = problema.get("numero"), problema.get("elemento_negrito", "").strip()
                             if elem and num in mapa_idx_para:
-                                negs_aplicados += formatacao_cruzada_indices(
-                                    mapa_idx_para[num], elem, aplicar_negrito=True
-                                )
+                                negs_aplicados += formatacao_cruzada_indices(mapa_idx_para[num], elem, aplicar_negrito=True)
 
                         st.session_state.doc_formatado_bytes = docx_to_bytes(doc_abnt)
-                        status.update(
-                            label=f"TCC formatado! {negs_aplicados} negritos aplicados nas referências.",
-                            state="complete", expanded=False
-                        )
+                        status.update(label=f"TCC formatado! {negs_aplicados} negritos aplicados nas referências.", state="complete", expanded=False)
                     except Exception as e:
-                        # Salva mesmo sem o negrito nas referências
                         st.session_state.doc_formatado_bytes = docx_to_bytes(doc_abnt)
                         status.update(label="Formatação aplicada (negrito de referências ignorado).", state="complete")
                         st.warning(f"Aviso na análise ABNT: {e}")
                 else:
                     st.session_state.doc_formatado_bytes = docx_to_bytes(doc_abnt)
-                    status.update(
-                        label="Formatação aplicada (nenhuma referência encontrada para análise de negrito).",
-                        state="complete"
-                    )
+                    status.update(label="Formatação aplicada (nenhuma referência encontrada).", state="complete")
 
     if st.session_state.doc_formatado_bytes:
         st.download_button(
             "⬇️ Baixar TCC Final Formatado (.docx)",
             data=st.session_state.doc_formatado_bytes,
-            file_name="TCC_Final_Formatado.docx",
+            file_name=st.session_state.nome_arq_abnt,
             mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
         )