Update app.py

app.py

@@ -4,6 +4,8 @@ import re
 import json
 import streamlit as st
 from docx import Document
+from docx.shared import Cm, Pt
+from docx.enum.text import WD_ALIGN_PARAGRAPH
 from pydantic import BaseModel
 from google import genai
 from google.genai import types
@@ -26,7 +28,7 @@ class AuditoriaTCC(BaseModel):
 # ─── AUTENTICAÇÃO GEMINI ──────────────────────────────────────────────────
 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.")
+    st.error("❌ ERRO: A variável de ambiente GEMINI_API_KEY não foi encontrada nos Secrets do Hugging Face.")
     st.stop()
 
 client = genai.Client(api_key=api_key)
@@ -37,8 +39,7 @@ def extrair_texto_docx(arquivo_upload) -> str:
     linhas = [para.text.strip() for para in doc.paragraphs if para.text.strip()]
     return "\n\n".join(linhas)
 
-def extrair_referencias_docx(arquivo_upload) -> list:
-    doc = Document(arquivo_upload)
+def extrair_referencias_docx(doc: Document) -> list:
     resultado = []
     em_refs = False
     for i, para in enumerate(doc.paragraphs):
@@ -53,27 +54,87 @@ def extrair_referencias_docx(arquivo_upload) -> list:
     return resultado
 
 def docx_to_bytes(doc: Document) -> bytes:
-    """Converte o documento editado para bytes para o botão de download."""
     byte_io = io.BytesIO()
     doc.save(byte_io)
     return byte_io.getvalue()
 
+def estimar_paginas(doc: Document) -> int:
+    """Estima o número de páginas com base na contagem de palavras do corpo do texto."""
+    palavras_totales = 0
+    for para in doc.paragraphs:
+        if re.search(r"^(?:\d+\.?\s*)?REFER[ÊE]NCIAS", para.text, re.IGNORECASE):
+            break 
+        palavras_totales += len(para.text.split())
+    return max(1, palavras_totales // 350)
+
+def aplicar_formatacao_estrutural_unifagoc(doc: Document):
+    """Varre o documento inteiro e força o template estrutural físico da UNIFAGOC."""
+    # 1. Aplicar Margens Oficiais (3cm sup/esq, 2cm inf/dir)
+    for section in doc.sections:
+        section.page_height = Cm(29.7) # A4
+        section.page_width = Cm(21.0)  # A4
+        section.top_margin = Cm(3.0)
+        section.left_margin = Cm(3.0)
+        section.bottom_margin = Cm(2.0)
+        section.right_margin = Cm(2.0)
+
+    em_referencias = False
+    em_resumo = False
+
+    # 2. Varrer Parágrafos para Fonte, Alinhamento e Tamanho
+    for para in doc.paragraphs:
+        texto = para.text.strip()
+        if not texto: continue
+
+        texto_upper = texto.upper()
+        
+        if texto_upper.startswith("RESUMO"): em_resumo = True
+        elif texto_upper.startswith("INTRODUÇÃO"): em_resumo = False
+        elif re.search(r"^(?:\d+\.?\s*)?REFER[ÊE]NCIAS", texto, re.IGNORECASE): em_referencias = True
+
+        # Alinhamento Base e Recuos
+        if em_referencias:
+            para.alignment = WD_ALIGN_PARAGRAPH.LEFT
+            para.paragraph_format.first_line_indent = Cm(0)
+            para.paragraph_format.space_after = Pt(6)
+        elif em_resumo:
+            para.alignment = WD_ALIGN_PARAGRAPH.JUSTIFY
+            para.paragraph_format.first_line_indent = Cm(0)
+        elif texto.isupper() and len(texto) < 100:
+            para.alignment = WD_ALIGN_PARAGRAPH.CENTER
+            para.paragraph_format.first_line_indent = Cm(0)
+        else:
+            para.alignment = WD_ALIGN_PARAGRAPH.JUSTIFY
+            para.paragraph_format.first_line_indent = Cm(1.25)
+
+        # Forçar Fonte e Tamanho
+        for run in para.runs:
+            run.font.name = 'Calibri'
+            
+            if em_resumo:
+                run.font.size = Pt(11)
+                if texto_upper != "RESUMO" and not texto_upper.startswith("PALAVRAS-CHAVE"):
+                    run.font.italic = True
+            elif em_referencias:
+                run.font.size = Pt(11)
+            elif texto.isupper() and len(texto) < 100:
+                if "TULO" in texto_upper or "DIREITO" in texto_upper:
+                    run.font.size = Pt(14)
+                else:
+                    run.font.size = Pt(12)
+            else:
+                run.font.size = Pt(12)
+
 # ─── INTERFACE PRINCIPAL (ABAS) ───────────────────────────────────────────
 st.title("⚖️ Assistente Acadêmico - Direito UNIFAGOC")
-aba_auditoria, aba_typos, aba_abnt = st.tabs(["🔍 Auditoria (Projeto vs TCC)", "📝 Correção de Ortografia", "📚 Referências ABNT"])
+aba_auditoria, aba_typos, aba_abnt = st.tabs(["🔍 Auditoria de Conteúdo", "📝 Correção Ortográfica", "📐 Formatação Total ABNT"])
 
 # ==============================================================================
-# ABA 1: AUDITORIA E TCC 1 vs TCC 2
+# ABA 1: AUDITORIA
 # ==============================================================================
 with aba_auditoria:
     st.markdown("Cruze o **Pré-Projeto** com o **TCC** para identificar fuga do tema e objetivos não cumpridos.")
-    
-    # OPÇÃO DE FASE DO TCC
-    fase_tcc = st.radio(
-        "Selecione a fase atual do trabalho:",
-        ["TCC 1 (Apenas Introdução e Capítulo 1)", "TCC 2 (TCC Completo com Resumo e Conclusão)"],
-        horizontal=True
-    )
+    fase_tcc = st.radio("Selecione a fase atual do trabalho:", ["TCC 1 (Apenas Introdução e Capítulo 1)", "TCC 2 (TCC Completo com Resumo e Conclusão)"], horizontal=True)
     
     col1, col2 = st.columns(2)
     with col1: arq_projeto = st.file_uploader("1. Envie o Pré-Projeto (.docx)", type=["docx"], key="proj")
@@ -83,103 +144,58 @@ with aba_auditoria:
         if not arq_projeto or not arq_tcc:
             st.warning("⚠️ Envie os DOIS arquivos para realizar a comparação.")
         else:
-            with st.spinner("Avaliando argumentos..."):
+            with st.spinner("Avaliando argumentos jurídicos e metodológicos..."):
                 texto_projeto = extrair_texto_docx(arq_projeto)
                 texto_tcc = extrair_texto_docx(arq_tcc)
                 
-                # Regra customizada baseada no TCC 1 ou TCC 2
-                regra_fase = ""
-                if "TCC 1" in fase_tcc:
-                    regra_fase = "ATENÇÃO MÁXIMA: O aluno está na disciplina TCC 1. Ele DEVE entregar apenas a Introdução e o Capítulo 1. NÃO penalize e NÃO aponte como erro a ausência de Resumo, Conclusão ou capítulos posteriores."
-                else:
-                    regra_fase = "O aluno está no TCC 2. O trabalho deve estar completo (Resumo, Introdução, Desenvolvimento, Conclusão e Referências)."
-
-                prompt = f"""
-                Você é um Professor Examinador de Direito da UNIFAGOC.
-                {regra_fase}
-                
-                === PRÉ-PROJETO ===
-                {texto_projeto}
-                
-                === TCC ===
-                {texto_tcc}
-                
-                Identifique o Problema, Objetivos e procure por fugas de tema genéricas.
-                """
-                
-                config = types.GenerateContentConfig(
-                    temperature=0.0,
-                    response_mime_type="application/json",
-                    response_schema=AuditoriaTCC,
-                )
+                regra_fase = "ATENÇÃO MÁXIMA: O aluno está na disciplina TCC 1. Ele DEVE entregar apenas a Introdução e o Capítulo 1. NÃO penalize e NÃO aponte como erro a ausência de Resumo, Conclusão ou capítulos posteriores." if "TCC 1" in fase_tcc else "O aluno está no TCC 2. O trabalho deve estar completo (Resumo, Introdução, Desenvolvimento, Conclusão e Referências)."
+                prompt = f"Você é um Professor Examinador de Direito da UNIFAGOC.\n{regra_fase}\n\n=== PRÉ-PROJETO ===\n{texto_projeto}\n\n=== TCC ===\n{texto_tcc}\nIdentifique o Problema, Objetivos e procure por fugas de tema genéricas."
+                config = types.GenerateContentConfig(temperature=0.0, response_mime_type="application/json", response_schema=AuditoriaTCC)
                 
                 try:
                     resposta = client.models.generate_content(model="gemini-3.5-flash", contents=prompt, config=config)
                     dados = json.loads(resposta.text)
-                    
                     st.success(f"Auditoria concluída para os parâmetros do {fase_tcc.split(' ')[0]}!")
                     st.metric(label="Índice de Aderência", value=f"{dados.get('aderencia_percentual')}%")
                     st.info(dados.get("diagnostico_geral"))
-                    
                     c1, c2 = st.columns(2)
                     with c1:
                         st.subheader("🎯 Problema de Pesquisa")
                         st.write(f"**Promessa:** {dados.get('problema_identificado_no_projeto')}")
-                        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 c2:
                         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("✅ Objetivos 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"Texto Genérico/IA: {ia}")
-                    
                 except Exception as e:
                     st.error(f"Erro: {e}")
 
 # ==============================================================================
-# ABA 2: CORREÇÃO DE ORTOGRAFIA E DIGITAÇÃO (TYPOS)
+# ABA 2: CORREÇÃO DE ORTOGRAFIA
 # ==============================================================================
 with aba_typos:
     st.markdown("Identifica e corrige automaticamente palavras duplicadas, erros ortográficos e gramaticais.")
-    arq_typos = st.file_uploader("Envie o TCC (.docx) para correção", type=["docx"], key="typos")
-    
+    arq_typos = st.file_uploader("Envie o TCC (.docx) para correção gramatical", type=["docx"], key="typos")
     if st.button("📝 Corrigir Ortografia"):
         if arq_typos:
             with st.spinner("Caçando erros de digitação..."):
                 texto = extrair_texto_docx(arq_typos)
-                prompt_typos = f"""
-                Analise o texto e identifique erros OBJETIVOS (ortografia, gramática, palavras duplicadas).
-                NÃO aponte termos em inglês ou jargões jurídicos.
-                TEXTO: {texto[:20000]}
-                
-                Retorne um JSON: {{"erros": [{{"original": "trecho errado", "corrigido": "trecho certo", "motivo": "explicacao"}}]}}
-                """
-                
+                prompt_typos = f"Analise o texto e identifique erros OBJETIVOS (ortografia, gramática, palavras duplicadas). NÃO aponte termos em inglês ou jargões jurídicos.\nTEXTO: {texto[:20000]}\nRetorne um JSON: {{\"erros\": [{{\"original\": \"trecho errado\", \"corrigido\": \"trecho certo\", \"motivo\": \"explicacao\"}}]}}"
                 try:
-                    res = client.models.generate_content(
-                        model="gemini-3.5-flash", 
-                        contents=prompt_typos, 
-                        config=types.GenerateContentConfig(temperature=0.1, response_mime_type="application/json")
-                    )
+                    res = client.models.generate_content(model="gemini-3.5-flash", contents=prompt_typos, config=types.GenerateContentConfig(temperature=0.1, response_mime_type="application/json"))
                     erros = json.loads(res.text).get("erros", [])
-                    
-                    if not erros:
-                        st.success("🎉 Nenhum erro encontrado! O texto está limpo.")
+                    if not erros: st.success("🎉 Nenhum erro encontrado! O texto está limpo.")
                     else:
                         st.warning(f"{len(erros)} erro(s) encontrado(s). Aplicando correções...")
                         doc_typos = Document(arq_typos)
                         aplicadas = 0
                         for e in erros:
-                            orig = e.get("original", "")
-                            corr = e.get("corrigido", "")
+                            orig, corr = e.get("original", ""), e.get("corrigido", "")
                             st.write(f"❌ `{orig}` ➔ ✅ `{corr}` (*{e.get('motivo')}*)")
-                            
-                            # Lógica de substituição no Documento
                             for para in doc_typos.paragraphs:
                                 if orig in para.text:
                                     for run in para.runs:
@@ -187,40 +203,46 @@ with aba_typos:
                                             run.text = run.text.replace(orig, corr, 1)
                                             aplicadas += 1
                                             break
-                        
                         if aplicadas > 0:
                             st.success(f"{aplicadas} correções aplicadas no documento!")
-                            st.download_button(
-                                label="⬇️ Baixar Documento Corrigido (.docx)",
-                                data=docx_to_bytes(doc_typos),
-                                file_name="TCC_Sem_Erros.docx",
-                                mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-                            )
+                            st.download_button("⬇️ Baixar Documento Corrigido (.docx)", data=docx_to_bytes(doc_typos), file_name="TCC_Sem_Erros.docx", mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
                 except Exception as e:
                     st.error(f"Erro ao processar: {e}")
 
 # ==============================================================================
-# ABA 3: FORMATAÇÃO DE REFERÊNCIAS ABNT
+# ABA 3: FORMATAÇÃO TOTAL E REFERÊNCIAS ABNT
 # ==============================================================================
 with aba_abnt:
-    st.markdown("Verifica a formatação NBR 6023:2018 e aplica o negrito nos elementos corretos (Ex: Título do livro ou Nome do Periódico).")
-    arq_abnt = st.file_uploader("Envie o TCC (.docx) para checar referências", type=["docx"], key="abnt")
+    st.markdown("Varre todo o documento forçando a fonte **Calibri**, Margens **3x2**, Alinhamento e Espaçamentos do edital, além de aplicar destaques ABNT nas Referências.")
+    st.warning("⏱️ **Aviso de Sobrecarga:** Esta ferramenta utiliza inteligência artificial. Se muitos alunos tentarem formatar ao mesmo tempo, aguarde 60 segundos e tente novamente.")
+    
+    arq_abnt = st.file_uploader("Envie o TCC (.docx) para formatar o layout", type=["docx"], key="abnt")
     
-    if st.button("📚 Corrigir Referências"):
+    if st.button("📐 Formatar Layout e Referências"):
         if arq_abnt:
-            with st.spinner("Avaliando normas ABNT..."):
-                referencias = extrair_referencias_docx(arq_abnt)
+            with st.spinner("Padronizando fontes, margens e referências... Isso pode levar um minuto."):
+                doc_abnt = Document(arq_abnt)
+                
+                # Estimativa de Páginas
+                paginas_estimadas = estimar_paginas(doc_abnt)
+                if paginas_estimadas > 18:
+                    st.error(f"🚨 **Atenção:** O corpo do seu TCC possui aproximadamente {paginas_estimadas} páginas. O limite máximo permitido pela banca (sem contar as referências) é de 18 páginas. Revise a extensão do seu texto.")
+                else:
+                    st.info(f"📄 Extensão estimada: {paginas_estimadas} páginas (dentro do limite de 18).")
+                
+                # Aplica as regras estruturais
+                aplicar_formatacao_estrutural_unifagoc(doc_abnt)
+                st.success("✅ Margens (3x2) ajustadas!")
+                st.success("✅ Fonte Calibri, tamanhos (11, 12 e 14) e alinhamentos aplicados perfeitamente!")
+                
+                # Analisa referências com a IA
+                referencias = extrair_referencias_docx(doc_abnt)
                 if not referencias:
-                    st.warning("Nenhuma referência encontrada. Certifique-se de que o título da seção seja 'REFERÊNCIAS'.")
+                    st.warning("Aviso: Nenhuma seção com o título 'REFERÊNCIAS' foi localizada para revisão de negrito.")
+                    st.download_button("⬇️ Baixar TCC Formatado (Sem Correção de Referências)", data=docx_to_bytes(doc_abnt), file_name="TCC_Padrao_UNIFAGOC.docx", mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
                 else:
                     texto_refs = "\n".join(f"{i+1}. {texto}" for i, (_, texto) in enumerate(referencias))
-                    prompt_abnt = f"""
-                    Verifique as referências abaixo seguindo a ABNT NBR 6023:2018.
-                    Retorne um JSON com a avaliação:
-                    {{"problemas": [{{"numero": int, "erros": ["erro 1"], "elemento_negrito": "palavra a receber destaque"}}]}}
-                    REFERÊNCIAS:
-                    {texto_refs}
-                    """
+                    prompt_abnt = f"Verifique as referências abaixo seguindo a ABNT NBR 6023:2018. Retorne um JSON:\n{{\"problemas\": [{{\"numero\": int, \"erros\": [\"erro\"], \"elemento_negrito\": \"palavra destaque\"}}]}}\nREFERÊNCIAS:\n{texto_refs}"
                     
                     try:
                         res = client.models.generate_content(
@@ -230,9 +252,6 @@ with aba_abnt:
                         )
                         problemas = json.loads(res.text).get("problemas", [])
                         
-                        doc_abnt = Document(arq_abnt)
-                        
-                        # Função interna de negrito adaptada do seu script original
                         def aplicar_negrito(doc, refs_problematicas):
                             em_refs = False
                             ref_idx = 0
@@ -244,13 +263,12 @@ with aba_abnt:
                                     continue
                                 if not em_refs or len(texto) < 10: continue
                                 
-                                # Aplica a lógica de negrito se a referência foi sinalizada
                                 nums_prob = {p.get("numero", 0) - 1 for p in refs_problematicas}
                                 if ref_idx in nums_prob:
                                     match = re.match(r"^((?:[A-ZÁÉÍÓÚÀÂÊÔÃÕÇ][^.;]+[.;]\s*)+)\s*(.+?)(\.\s+(?:\d+\.\s+ed|[A-Z]).*|$)", para.text)
                                     if match:
                                         autoria, titulo = match.group(1).strip(), match.group(2).strip()
-                                        pos_atual, pos_ini, pos_fim = 0, para.text.find(titulo, len(autoria)), 0
+                                        pos_atual, pos_ini = 0, para.text.find(titulo, len(autoria))
                                         pos_fim = pos_ini + len(titulo)
                                         for run in para.runs:
                                             fim_run = pos_atual + len(run.text)
@@ -261,15 +279,17 @@ with aba_abnt:
 
                         aplicar_negrito(doc_abnt, problemas)
                         
-                        for p in problemas:
-                            st.write(f"⚠️ Referência #{p['numero']} possui erros: {', '.join(p['erros'])}")
-                            
-                        st.success("Formatação de destaque (negrito) aplicada com base na ABNT!")
+                        if problemas:
+                            for p in problemas:
+                                st.write(f"⚠️ Referência #{p['numero']} ajustada: {', '.join(p['erros'])}")
+                        st.success("✅ Formatação de destaque (negrito) nas referências aplicada!")
+                        
                         st.download_button(
-                            label="⬇️ Baixar TCC com Referências Corrigidas (.docx)",
+                            label="⬇️ Baixar TCC Totalmente Formatado (.docx)",
                             data=docx_to_bytes(doc_abnt),
-                            file_name="TCC_ABNT_Corrigido.docx",
+                            file_name="TCC_Padrao_UNIFAGOC.docx",
                             mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
                         )
                     except Exception as e:
-                        st.error(f"Erro na verificação: {e}")
+                        if "429" in str(e): st.error("❌ A API está sobrecarregada no momento (muitos alunos usando). Por favor, aguarde 1 minuto e tente novamente.")
+                        else: st.error(f"Erro na verificação pela IA: {e}")