Update app.py

app.py

@@ -17,7 +17,6 @@ from datetime import datetime
 from ratelimiter import RateLimiter
 import difflib
 from html import escape
-import pdfkit  # Requer wkhtmltopdf instalado
 
 load_dotenv()
 
@@ -228,12 +227,13 @@ def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
        - Juntada da guia de custas
        - Fechamento formal da petição
     **Critérios Fundamentais:**
-    1. Identificação do candidato resulta em nota máxima de 3,0
+    1. Identificação do candidato (nome ou número real na seção final) resulta em nota máxima de 3,0
     2. Invenção de fatos não constantes no caso resulta em penalização grave (-5,0 pontos)
     3. Falta de pedido de juntada de custas resulta em perda de 1,0 ponto
     4. Use marcadores para dados não fornecidos (ex: "Rua XXX", "CPF XXX")
     5. Não penalize por informações ausentes no caso original
-    6. A frase "Nestes termos, pede deferimento" com vírgula é incorreta; corrija para "Nestes termos pede deferimento" e destaque o erro na análise.
+    6. A frase "Nestes termos, pede deferimento" com vírgula é incorreta; corrija para "Nestes termos pede deferimento" e destaque o erro na análise
+    7. A petição deve finalizar com "Local e data.\nADVOGADO.\nOAB." sem identificação do aluno (nome, número ou estado na OAB); se houver identificação específica (ex.: número ou estado), zere a nota (0,0)
     **Instruções Especiais para Endereçamento:**
     - CPC (art. 46): foro competente é o domicílio do réu; CDC (art. 101, I): em relação de consumo, foro do autor é correto.
     - Avalie o contexto: priorize o foro do autor em casos consumeristas.
@@ -258,7 +258,7 @@ def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
     **Fechamento**                             |           |
     12. Fechamento formal adequado (0,10)     | / 0,10 |
     **Nota Final:**
-    Calcule a nota final somando os pontos.
+    Calcule a nota final somando os pontos, exceto se houver identificação específica na OAB, que zera a nota.
     ---
 
     ### Petição Corrigida
@@ -273,7 +273,7 @@ def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
     8. Opção por audiência (Art. 319, VII, CPC)
     9. Pedido de citação
     10. Juntada da guia de custas
-    11. Fechamento formal
+    11. Fechamento formal com "Local e data.\nADVOGADO.\nOAB." sem identificação específica
     """
     if modo_treinamento:
         base_prompt += f"""
@@ -359,7 +359,6 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
     if not peticao_conteudo.strip():
         return "⚠️ Forneça uma petição."
     
-    # Validação preliminar como aviso, não como bloqueio
     problemas = validar_peticao_basica(peticao_conteudo)
     aviso_validacao = f"### Validação Inicial\n\n" + "\n".join(problemas) + "\n\n" if problemas else ""
 
@@ -481,15 +480,15 @@ def calcular_progresso():
 
 def comparar_peticao_melhorada(caso, peticao_texto, correcao):
     if not caso:
-        return "⚠️ Forneça um caso antes de comparar.", "", ""
+        return "⚠️ Forneça um caso antes de comparar.", "", "", ""
     peticao = peticao_texto if peticao_texto else (HISTORICO[-1]["peticao"] if HISTORICO else "")
     if not peticao:
-        return "⚠️ Forneça uma petição para comparar ou corrija uma antes.", "", ""
+        return "⚠️ Forneça uma petição para comparar ou corrija uma antes.", "", "", ""
     if not correcao:
-        return "⚠️ Corrija a petição antes de comparar.", "", ""
+        return "⚠️ Corrija a petição antes de comparar.", "", "", ""
     ultima_correcao = HISTORICO[-1]["correcao_completa"] if HISTORICO else ""
     if not ultima_correcao:
-        return "⚠️ Nenhuma correção disponível no histórico.", "", ""
+        return "⚠️ Nenhuma correção disponível no histórico.", "", "", ""
     peticao_corrigida = extrair_peticao_corrigida(ultima_correcao)
 
     orig_lines = peticao.split('\n')
@@ -572,18 +571,14 @@ def comparar_peticao_melhorada(caso, peticao_texto, correcao):
                 resumo_alteracoes.append(f"  - ... e mais {len(alteracoes) - 5} alterações")
 
     resumo_estatistico = f"""
-## 📊 Resumo das Alterações
 - **Linhas Removidas:** {total_removidas}
 - **Linhas Adicionadas:** {total_adicionadas}
 - **Diferença Líquida:** {total_adicionadas - total_removidas} linhas
+"""
 
-## 🚨 Mudanças Críticas
-{''.join(criticas) if criticas else 'Nenhuma mudança crítica detectada.'}
-
-## 🔍 Alterações por Seção
-{''.join(resumo_alteracoes)}
-
-## 📝 Visualização Detalhada
+    criticas_texto = "\n".join(criticas) if criticas else "Nenhuma mudança crítica detectada."
+    alteracoes_texto = "".join(resumo_alteracoes)
+    visualizacao_detalhada = f"""
 <style>
 .linha-igual {{ color: #000000; margin: 2px 0; }}
 .linha-removida {{ color: #ffffff; background-color: #b30000; margin: 2px 0; padding: 2px; }}
@@ -591,20 +586,7 @@ def comparar_peticao_melhorada(caso, peticao_texto, correcao):
 </style>
 {"".join(resultado_diff)}
 """
-    return peticao, peticao_corrigida, resumo_estatistico
-
-def exportar_diferencas(caso, peticao_texto, correcao):
-    _, _, resumo_estatistico = comparar_peticao_melhorada(caso, peticao_texto, correcao)
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".html") as tmp:
-        tmp.write(resumo_estatistico.encode('utf-8'))
-        pdf_path = tmp.name.replace(".html", ".pdf")
-        try:
-            pdfkit.from_file(tmp.name, pdf_path)
-            return pdf_path
-        except Exception as e:
-            return f"❌ Erro ao gerar PDF: {str(e)}"
-        finally:
-            os.unlink(tmp.name)
+    return peticao, peticao_corrigida, resumo_estatistico, criticas_texto, alteracoes_texto, visualizacao_detalhada
 
 def interface_gradio():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -628,6 +610,7 @@ def interface_gradio():
                 - ❌ Identificação do candidato: nota máxima 3,0
                 - ❌ Invenção de fatos: -5,0 pontos
                 - ❌ Sem pedido de custas: -1,0 ponto
+                - ❌ Identificação específica na OAB (nome, número ou estado): nota 0,0
                 """)
                 modo_treinamento = gr.Checkbox(label="Modo de Treinamento Interativo", value=False)
                 with gr.Accordion("⚙️ Configurações", open=False):
@@ -674,9 +657,11 @@ def interface_gradio():
             peticao_corrigida_output = gr.Textbox(label="Petição Corrigida", lines=15, interactive=False)
 
         with gr.Row():
-            comparacao_diferencas = gr.HTML(label="Diferenças Destacadas")
-            export_btn = gr.Button("📥 Exportar Diferenças", visible=False)
-            export_file = gr.File(label="Baixar Diferenças", visible=False)
+            with gr.Column(scale=1):
+                resumo_alteracoes_output = gr.Markdown(label="📊 Resumo das Alterações")
+                criticas_output = gr.Markdown(label="🚨 Mudanças Críticas")
+            with gr.Column(scale=2):
+                visualizacao_detalhada_output = gr.Markdown(label="📝 Alterações Detalhadas")
 
         with gr.Accordion("📜 Histórico de Correções", open=False):
             historico_output = gr.Markdown()
@@ -698,25 +683,17 @@ def interface_gradio():
             outputs=[resultado_output]
         )
         clear_btn.click(
-            fn=lambda: [None, None, None, None, "", "", "", "", gr.update(visible=False), gr.update(visible=False)],
+            fn=lambda: [None, None, None, None, "", "", "", "", "", "", ""],
             inputs=None,
-            outputs=[caso_input, peticao_texto, peticao_arquivo, arquivo_preview, resultado_output, dica_contextual, peticao_aluno_output, peticao_corrigida_output, export_btn, export_file]
+            outputs=[caso_input, peticao_texto, peticao_arquivo, arquivo_preview, resultado_output, dica_contextual, 
+                     peticao_aluno_output, peticao_corrigida_output, resumo_alteracoes_output, criticas_output, 
+                     visualizacao_detalhada_output]
         )
         comparar_btn.click(
             fn=comparar_peticao_melhorada,
             inputs=[caso_input, peticao_texto, resultado_output],
-            outputs=[peticao_aluno_output, peticao_corrigida_output, comparacao_diferencas]
-        ).then(
-            fn=lambda: [gr.update(visible=True), gr.update(visible=False)],
-            outputs=[export_btn, export_file]
-        )
-        export_btn.click(
-            fn=exportar_diferencas,
-            inputs=[caso_input, peticao_texto, resultado_output],
-            outputs=export_file
-        ).then(
-            fn=lambda x: [gr.update(visible=True, value=x), gr.update(visible=True)],
-            outputs=[export_file, export_btn]
+            outputs=[peticao_aluno_output, peticao_corrigida_output, resumo_alteracoes_output, criticas_output, 
+                     visualizacao_detalhada_output]
         )
 
         with gr.Accordion("📘 Critérios Detalhados", open=False):
@@ -729,9 +706,12 @@ def interface_gradio():
             - ✅ CORRETO: Pedir citação conforme dados do caso
             #### 3. Pedidos Obrigatórios
             - ✅ OBRIGATÓRIO: Incluir "a juntada da guia de custas"
-            #### 4. Penalização Grave
-            - ❌ INCORRETO: Identificação do candidato
+            #### 4. Fechamento Formal
+            - ✅ OBRIGATÓRIO: Finalizar com "Local e data.\nADVOGADO.\nOAB." sem identificação específica
+            #### 5. Penalização Grave
+            - ❌ INCORRETO: Identificação do candidato (nome ou número)
             - ❌ INCORRETO: Invenção de fatos
+            - ❌ INCORRETO: Identificação na OAB (número ou estado)
             """)
         gr.Markdown("""
         ### ℹ️ Modo de Treinamento Interativo