Update app.py

app.py

@@ -24,6 +24,21 @@ class AuditoriaTCC(BaseModel):
     objetivos_nao_cumpridos_no_tcc: list[str]
     desvios_de_tema_ou_linguica: list[str]
     indicios_texto_generico_ia: list[str]
+    # NOVOS CAMPOS: Ficha de Avaliação UNIFAGOC (80 Pontos Textuais)
+    nota_tema_problema: float
+    nota_introducao_metodo: float
+    nota_estruturacao: float
+    nota_formatacao_abnt: float
+    justificativa_notas_banca: str
+
+class ErroTexto(BaseModel):
+    original: str
+    corrigido: str
+    motivo: str
+
+class RevisaoInteligente(BaseModel):
+    correcoes: list[ErroTexto]
+    termos_estrangeiros_para_italico: list[str]
 
 # ─── AUTENTICAÇÃO GEMINI ──────────────────────────────────────────────────
 api_key = os.environ.get("GEMINI_API_KEY")
@@ -59,40 +74,32 @@ def docx_to_bytes(doc: Document) -> bytes:
     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 
+        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.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)
 
-    em_referencias = False
-    em_resumo = False
+    em_referencias, em_resumo = False, 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)
@@ -107,10 +114,8 @@ def aplicar_formatacao_estrutural_unifagoc(doc: Document):
             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"):
@@ -118,46 +123,85 @@ def aplicar_formatacao_estrutural_unifagoc(doc: Document):
             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)
+                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 de Conteúdo", "📝 Correção Ortográfica", "📐 Formatação Total ABNT"])
+aba_auditoria, aba_revisao, aba_abnt = st.tabs(["🔍 Auditoria de Conteúdo e Notas", "🧠 Revisão Textual Inteligente", "📐 Formatação Total ABNT"])
 
 # ==============================================================================
-# ABA 1: AUDITORIA
+# ABA 1: AUDITORIA E PONTUAÇÃO UNIFAGOC
 # ==============================================================================
 with aba_auditoria:
-    st.markdown("Cruze o **Pré-Projeto** com o **TCC** para identificar fuga do tema e objetivos não cumpridos.")
+    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("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")
     with col2: arq_tcc = st.file_uploader("2. Envie os Capítulos do TCC (.docx)", type=["docx"], key="tcc_aud")
 
-    if st.button("🔍 Iniciar Auditoria Cruzada", type="primary"):
+    if st.button("🔍 Iniciar Auditoria e Avaliação", type="primary"):
         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 jurídicos e metodológicos..."):
+            with st.spinner("Avaliando argumentos e calculando notas da banca..."):
                 texto_projeto = extrair_texto_docx(arq_projeto)
                 texto_tcc = extrair_texto_docx(arq_tcc)
+                regra_fase = "ATENÇÃO: O aluno está em TCC 1 (Apenas Introdução e Capítulo 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. 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}
                 
-                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."
+                === PRÉ-PROJETO ===
+                {texto_projeto}
+                
+                === TCC ===
+                {texto_tcc}
+                
+                TAREFA 1: Identifique o Problema, Objetivos e procure por fugas de tema genéricas.
+                TAREFA 2: Avalie o TCC conforme os 80 pontos da FICHA DE AVALIAÇÃO DA UNIFAGOC:
+                - Tema, Delimitação do Problema e Objeto (0 a 20 pts)
+                - Introdução e Método (0 a 20 pts)
+                - Estruturação e Desenvolvimento (0 a 20 pts)
+                - Formatação e ABNT (0 a 20 pts)
+                Seja rigoroso e acadêmico na atribuição das notas.
+                """
                 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"))
+                    
+                    st.success(f"Auditoria e Avaliação concluídas para {fase_tcc.split(' ')[0]}!")
+                    
+                    # --- NOVO BLOCO DE NOTAS DA BANCA ---
+                    st.header("🏆 Avaliação da Banca UNIFAGOC")
+                    st.markdown("*Distribuição de 80 pontos textuais (20 pts de apresentação não inclusos).*")
+                    
+                    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)
+                    ])
+                    
+                    c_nota1, c_nota2, c_nota3, c_nota4 = st.columns(4)
+                    c_nota1.metric("Tema/Problema", f"{dados.get('nota_tema_problema')}/20")
+                    c_nota2.metric("Intro/Método", f"{dados.get('nota_introducao_metodo')}/20")
+                    c_nota3.metric("Estrutura/Linguagem", f"{dados.get('nota_estruturacao')}/20")
+                    c_nota4.metric("Formatação/ABNT", f"{dados.get('nota_formatacao_abnt')}/20")
+                    
+                    st.info(f"**NOTA TEXTUAL FINAL: {nota_total}/80 pontos** \n*Justificativa da Banca:* {dados.get('justificativa_notas_banca')}")
+                    
+                    st.divider()
+                    # --- FIM DO BLOCO DE NOTAS ---
+                    
+                    st.metric(label="Índice de Aderência ao Projeto", value=f"{dados.get('aderencia_percentual')}%")
+                    st.write(f"**Diagnóstico:** {dados.get('diagnostico_geral')}")
+                    
                     c1, c2 = st.columns(2)
                     with c1:
                         st.subheader("🎯 Problema de Pesquisa")
@@ -169,77 +213,108 @@ with aba_auditoria:
                         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 fuga in dados.get("desvios_de_tema_ou_linguica", []): st.warning(f"Fuga de Tema: {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}")
+                    if "429" in str(e): st.error("❌ A API está sobrecarregada. Aguarde 1 minuto.")
+                    else: st.error(f"Erro: {e}")
 
 # ==============================================================================
-# ABA 2: CORREÇÃO DE ORTOGRAFIA
+# ABA 2: REVISÃO TEXTUAL INTELIGENTE (CAIXA ALTA E ITÁLICO)
 # ==============================================================================
-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 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.\nTEXTO: {texto[:20000]}\nRetorne um JSON: {{\"erros\": [{{\"original\": \"trecho errado\", \"corrigido\": \"trecho certo\", \"motivo\": \"explicacao\"}}]}}"
+with aba_revisao:
+    st.markdown("A IA analisa o contexto para corrigir **Ortografia**, parágrafos indevidamente em **CAIXA ALTA** e aplica **Itálico** em termos estrangeiros/latinos.")
+    arq_revisao = st.file_uploader("Envie o TCC (.docx) para revisão inteligente", type=["docx"], key="rev")
+    
+    if st.button("🧠 Realizar Revisão Inteligente"):
+        if arq_revisao:
+            with st.spinner("Lendo o documento para identificar jargões, caixa alta e erros... Isso pode levar um tempo."):
+                texto = extrair_texto_docx(arq_revisao)
+                prompt_rev = f"""Analise o texto acadêmico de Direito abaixo e identifique:
+                1. Erros ortográficos e gramaticais.
+                2. Trechos do corpo do texto que estão indevidamente em CAIXA ALTA (retorne a versão corrigida). Ignore títulos.
+                3. Termos estrangeiros ou jargões em latim (ex: habeas corpus, fake news) que necessitam de itálico.
+                TEXTO: {texto[:25000]}"""
+                
                 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"))
-                    erros = json.loads(res.text).get("erros", [])
-                    if not erros: st.success("🎉 Nenhum erro encontrado! O texto está limpo.")
+                    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)
+                    )
+                    dados_rev = json.loads(res.text)
+                    correcoes = dados_rev.get("correcoes", [])
+                    italicos = dados_rev.get("termos_estrangeiros_para_italico", [])
+                    
+                    if not correcoes and not italicos:
+                        st.success("🎉 O texto está perfeitamente revisado e sem jargões para formatar.")
                     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, corr = e.get("original", ""), e.get("corrigido", "")
-                            st.write(f"❌ `{orig}` ➔ ✅ `{corr}` (*{e.get('motivo')}*)")
-                            for para in doc_typos.paragraphs:
-                                if orig in para.text:
-                                    for run in para.runs:
-                                        if orig in run.text:
-                                            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("⬇️ Baixar Documento Corrigido (.docx)", data=docx_to_bytes(doc_typos), file_name="TCC_Sem_Erros.docx", mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
+                        doc_rev = Document(arq_revisao)
+                        aplicadas_texto = 0
+                        aplicadas_italico = 0
+                        
+                        if correcoes:
+                            st.subheader("📝 Correções de Texto e Caixa Alta")
+                            for c in correcoes:
+                                orig, corr = c.get("original", ""), c.get("corrigido", "")
+                                st.write(f"❌ `{orig}` ➔ ✅ `{corr}` (*{c.get('motivo')}*)")
+                                for para in doc_rev.paragraphs:
+                                    if orig in para.text:
+                                        for run in para.runs:
+                                            if orig in run.text:
+                                                run.text = run.text.replace(orig, corr, 1)
+                                                aplicadas_texto += 1
+                                                break
+
+                        if italicos:
+                            st.subheader("🪄 Termos Estrangeiros (Itálico Aplicado)")
+                            st.write(", ".join([f"*{i}*" for i in italicos]))
+                            for termo in italicos:
+                                for para in doc_rev.paragraphs:
+                                    matches = list(re.finditer(r'\b' + re.escape(termo) + r'\b', para.text, re.IGNORECASE))
+                                    for match in matches:
+                                        pos_ini = match.start()
+                                        pos_fim = match.end()
+                                        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:
+                                                run.italic = True
+                                                aplicadas_italico += 1
+                                            pos_atual += len(run.text)
+                        
+                        st.success(f"✅ {aplicadas_texto} correções de texto e {aplicadas_italico} itálicos aplicados!")
+                        st.download_button("⬇️ Baixar Documento Revisado (.docx)", data=docx_to_bytes(doc_rev), file_name="TCC_Revisado_Inteligente.docx", mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
                 except Exception as e:
-                    st.error(f"Erro ao processar: {e}")
+                    if "429" in str(e): st.error("❌ A API está sobrecarregada. Aguarde 1 minuto.")
+                    else: st.error(f"Erro ao processar: {e}")
 
 # ==============================================================================
 # ABA 3: FORMATAÇÃO TOTAL E REFERÊNCIAS ABNT
 # ==============================================================================
 with aba_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.")
+    st.warning("⏱️ **Aviso de Sobrecarga:** Se muitos alunos tentarem formatar ao mesmo tempo, aguarde 60 segundos.")
     
-    arq_abnt = st.file_uploader("Envie o TCC (.docx) para formatar o layout", type=["docx"], key="abnt")
+    arq_abnt = st.file_uploader("Envie o TCC (.docx) para formatar o layout estrutural", type=["docx"], key="abnt")
     
     if st.button("📐 Formatar Layout e Referências"):
         if 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).")
+                if paginas_estimadas > 18: st.error(f"🚨 **Atenção:** Seu TCC tem aprox. {paginas_estimadas} páginas. O limite da banca é 18.")
+                else: st.info(f"📄 Extensão estimada: {paginas_estimadas} páginas.")
                 
-                # 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("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")
+                    st.download_button("⬇️ Baixar TCC Formatado (Sem 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:\n{{\"problemas\": [{{\"numero\": int, \"erros\": [\"erro\"], \"elemento_negrito\": \"palavra destaque\"}}]}}\nREFERÊNCIAS:\n{texto_refs}"
@@ -252,9 +327,8 @@ with aba_abnt:
                         )
                         problemas = json.loads(res.text).get("problemas", [])
                         
-                        def aplicar_negrito(doc, refs_problematicas):
-                            em_refs = False
-                            ref_idx = 0
+                        def aplicar_negrito_ref(doc, refs_problematicas):
+                            em_refs, ref_idx = False, 0
                             for para in doc.paragraphs:
                                 texto = para.text.strip()
                                 if not texto: continue
@@ -277,11 +351,10 @@ with aba_abnt:
                                             pos_atual += len(run.text)
                                 ref_idx += 1
 
-                        aplicar_negrito(doc_abnt, problemas)
+                        aplicar_negrito_ref(doc_abnt, problemas)
                         
                         if problemas:
-                            for p in problemas:
-                                st.write(f"⚠️ Referência #{p['numero']} ajustada: {', '.join(p['erros'])}")
+                            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(
@@ -291,5 +364,5 @@ with aba_abnt:
                             mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
                         )
                     except Exception as 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.")
+                        if "429" in str(e): st.error("❌ A API está sobrecarregada. Aguarde 1 minuto.")
                         else: st.error(f"Erro na verificação pela IA: {e}")