Update app.py

app.py

@@ -15,6 +15,9 @@ import cv2
 import numpy as np
 from datetime import datetime
 from ratelimiter import RateLimiter
+import difflib
+from html import escape
+import pdfkit  # Requer wkhtmltopdf instalado
 
 load_dotenv()
 
@@ -186,6 +189,20 @@ def calcular_pontuacao(enderecamento, fundamentos, pedidos, fechamento):
     tabela_formatada = "\n".join([f"| {linha[0].ljust(60)} | {linha[1].rjust(10)} |" for linha in tabela])
     return f"{tabela_formatada}\n\n**Nota Final:** {nota_final:.2f} / 10,00"
 
+def validar_peticao_basica(peticao):
+    problemas = []
+    if "EXCELENTÍSSIMO" not in peticao.upper():
+        problemas.append("⚠️ Endereçamento ausente ou incompleto.")
+    if "Art. 319" not in peticao and "CPC" not in peticao:
+        problemas.append("⚠️ Fundamentação jurídica não menciona o CPC.")
+    if "Dá-se à causa o valor" not in peticao:
+        problemas.append("⚠️ Valor da causa não especificado (Art. 319, V, CPC).")
+    if "juntada da guia de custas" not in peticao.lower():
+        problemas.append("⚠️ Pedido de juntada da guia de custas ausente.")
+    if "Nestes termos" not in peticao:
+        problemas.append("⚠️ Fechamento formal ausente.")
+    return problemas
+
 def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
     base_prompt = f"""
     ## Correção Rigorosa de Petição Inicial Cível (Critérios Realistas da OAB)
@@ -262,9 +279,10 @@ def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
         base_prompt += f"""
         **Modo de Treinamento Interativo Ativado:**
         - Para cada item com pontuação inferior ao máximo, forneça:
-          a) Uma explicação detalhada (mínimo 2 frases) descrevendo o erro e por que ele está errado conforme o CPC ou CDC.
-          b) Uma sugestão de redação corrigida baseada exclusivamente no caso fornecido.
-          c) Uma dica prática extraída do banco de dicas ({json.dumps(DICAS_OAB)}) ou uma dica personalizada se o banco não se aplicar.
+          a) Uma explicação detalhada (mínimo 2 frases) descrevendo o erro e sua base legal (ex.: Art. 319, CPC).
+          b) Uma sugestão de redação corrigida baseada no caso.
+          c) Um exemplo prático de redação correta em outro contexto (ex.: "Em uma ação de dano moral, poderia ser: 'Requer a condenação ao pagamento de R$ 10.000,00 por danos morais, conforme Súmula 387 do STJ.'").
+          d) Uma dica prática do banco ({json.dumps(DICAS_OAB)}) ou personalizada.
         - Não resuma; detalhe cada erro de forma pedagógica e clara.
         """
     return base_prompt
@@ -341,6 +359,10 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
     if not peticao_conteudo.strip():
         return "⚠️ Forneça uma petição."
     
+    problemas = validar_peticao_basica(peticao_conteudo)
+    if problemas:
+        return f"### Validação Inicial\n\n" + "\n".join(problemas) + "\n\nCorrija os itens acima antes de prosseguir."
+
     cache_key = f"{caso[:50]}|{peticao_conteudo[:50]}|{modo_treinamento}"
     if cache_key in CORRECAO_CACHE:
         return CORRECAO_CACHE[cache_key]
@@ -376,7 +398,7 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
                 analise = correção_completa.strip()
             correção_formatada = formatar_correção(analise)
             HISTORICO.append({
-                "data": datetime.now().strftime("%Y-%-m-%d %H:%M:%S"),
+                "data": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
                 "caso": caso,
                 "peticao": peticao_conteudo,
                 "correcao_completa": correção_completa
@@ -400,11 +422,19 @@ def formatar_correção(texto):
     linhas = texto.split('\n')
     texto_formatado = ""
     for linha in linhas:
-        if "Análise Geral" in linha or "Nota Final" in linha or "Pontos Positivos" in linha or "Pontos a Melhorar" in linha:
-            texto_formatado += f"### {linha.strip()}\n\n"
+        if "Análise Geral" in linha:
+            texto_formatado += f"### ℹ️ {linha.strip()}\n\n"
+        elif "Nota Final" in linha:
+            texto_formatado += f"### ⭐ {linha.strip()}\n\n"
+        elif "Pontos Positivos" in linha:
+            texto_formatado += f"### ✅ {linha.strip()}\n\n"
+        elif "Pontos a Melhorar" in linha:
+            texto_formatado += f"### ⚠️ {linha.strip()}\n\n"
+        elif "- Invenção de fatos" in linha:
+            texto_formatado += f"**🔴 {linha.strip()}**\n"
         elif linha.strip().startswith('-') and len(linha) > 2 and any(termo in linha for termo in ["Endereçamento", "Qualificação", "Fatos", "Pedidos", "Valor", "Provas"]):
             texto_formatado += f"**{linha.strip()}**\n"
-        elif "Explicação" in linha or "Sugestão" in linha or "Dica" in linha:
+        elif "Explicação" in linha or "Sugestão" in linha or "Dica" in linha or "Exemplo" in linha:
             texto_formatado += f"*{linha.strip()}*\n"
         else:
             texto_formatado += f"{linha}\n"
@@ -414,7 +444,7 @@ def formatar_correção(texto):
         match = re.search(nota_pattern, texto_formatado, re.DOTALL)
         if match:
             nota = match.group(1)
-            texto_formatado = texto_formatado.replace(f"### Nota Final\n\n{nota}", f"### Nota Final: {nota}/10\n\n")
+            texto_formatado = texto_formatado.replace(f"### ⭐ Nota Final\n\n{nota}", f"### ⭐ Nota Final: {nota}/10\n\n")
     return texto_formatado
 
 def extrair_peticao_corrigida(correcao_completa):
@@ -450,9 +480,6 @@ def calcular_progresso():
     return f"📊 Progresso: Média {media:.1f}/10 | Tentativas: {len(notas)} | Tendência: {tendencia}"
 
 def comparar_peticao_melhorada(caso, peticao_texto, correcao):
-    import difflib
-    from html import escape
-
     if not caso:
         return "⚠️ Forneça um caso antes de comparar.", "", ""
     peticao = peticao_texto if peticao_texto else (HISTORICO[-1]["peticao"] if HISTORICO else "")
@@ -497,6 +524,7 @@ def comparar_peticao_melhorada(caso, peticao_texto, correcao):
 
     secao_atual = "OUTROS"
     resultado_diff = []
+    criticas = []
     for linha in diff:
         if linha.startswith('  '):
             texto = linha[2:]
@@ -506,6 +534,8 @@ def comparar_peticao_melhorada(caso, peticao_texto, correcao):
             resultado_diff.append(f"<div class='linha-igual'>{escape(texto)}</div>")
         elif linha.startswith('- '):
             texto = linha[2:]
+            if any(kw in texto.lower() for kw in ["invenção", "custas", "art. 319"]):
+                criticas.append(f"🔴 REMOVIDO: {texto}")
             resultado_diff.append(f"<div class='linha-removida'><del>{escape(texto)}</del></div>")
             secoes[secao_atual].append(f"- REMOVIDO: {texto}")
         elif linha.startswith('+ '):
@@ -513,6 +543,8 @@ def comparar_peticao_melhorada(caso, peticao_texto, correcao):
             nova_secao = identificar_secao(texto)
             if nova_secao != "OUTROS":
                 secao_atual = nova_secao
+            if any(kw in texto.lower() for kw in ["custas", "art. 319"]):
+                criticas.append(f"🟢 ADICIONADO: {texto}")
             resultado_diff.append(f"<div class='linha-adicionada'><ins>{escape(texto)}</ins></div>")
             secoes[secao_atual].append(f"+ ADICIONADO: {texto}")
         elif linha.startswith('? '):
@@ -524,31 +556,56 @@ def comparar_peticao_melhorada(caso, peticao_texto, correcao):
     resumo_alteracoes = []
     for secao, alteracoes in secoes.items():
         if alteracoes:
-            resumo_alteracoes.append(f"### {secao}\n")
+            adicionadas = sum(1 for alt in alteracoes if alt.startswith("+"))
+            removidas = sum(1 for alt in alteracoes if alt.startswith("-"))
+            resumo_alteracoes.append(f"""
+### {secao}
+| Tipo         | Quantidade |
+|--------------|------------|
+| Adicionadas  | {adicionadas}      |
+| Removidas    | {removidas}      |
+""")
             for alt in alteracoes[:5]:
-                resumo_alteracoes.append(f"  {alt}")
+                texto = alt[:60] + "..." if len(alt) > 60 else alt
+                resumo_alteracoes.append(f"  {texto}")
             if len(alteracoes) > 5:
-                resumo_alteracoes.append(f"  - ... e mais {len(alteracoes) - 5} alterações\n")
+                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
-
-    ## 🔍 Alterações por Seção
-    {''.join(resumo_alteracoes)}
-
-    ## 📝 Visualização Detalhada
-    <style>
-    .linha-igual {{ color: #000000; margin: 2px 0; }}
-    .linha-removida {{ color: #ffffff; background-color: #b30000; margin: 2px 0; padding: 2px; }}
-    .linha-adicionada {{ color: #ffffff; background-color: #006600; margin: 2px 0; padding: 2px; }}
-    </style>
-    {"".join(resultado_diff)}
-    """
+## 📊 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
+<style>
+.linha-igual {{ color: #000000; margin: 2px 0; }}
+.linha-removida {{ color: #ffffff; background-color: #b30000; margin: 2px 0; padding: 2px; }}
+.linha-adicionada {{ color: #ffffff; background-color: #006600; margin: 2px 0; padding: 2px; }}
+</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)
+
 def interface_gradio():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("# 📝 Corretor Avançado de Petições Iniciais (Critérios OAB)")
@@ -592,7 +649,7 @@ def interface_gradio():
                 with gr.Row():
                     gr.Markdown("### 🧩 Gerar Caso Fictício")
                     caso_btn = gr.Button("🔄 Gerar Novo Caso Fictício", variant="secondary")
-                caso_input = gr.Textbox(label="� - 📋 Caso Fornecido pela Banca", lines=8, placeholder="Insira o texto COMPLETO do caso...")
+                caso_input = gr.Textbox(label="📋 Caso Fornecido pela Banca", lines=8, placeholder="Insira o texto COMPLETO do caso...")
                 with gr.Tabs():
                     with gr.TabItem("✏️ Inserir Texto"):
                         peticao_texto = gr.Textbox(label="Texto da Petição do Aluno", lines=15, placeholder="Insira a petição completa...")
@@ -618,6 +675,8 @@ def interface_gradio():
 
         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.Accordion("📜 Histórico de Correções", open=False):
             historico_output = gr.Markdown()
@@ -639,14 +698,25 @@ def interface_gradio():
             outputs=[resultado_output]
         )
         clear_btn.click(
-            fn=lambda: [None, None, None, None, "", "", "", ""],
+            fn=lambda: [None, None, None, None, "", "", "", "", gr.update(visible=False), gr.update(visible=False)],
             inputs=None,
-            outputs=[caso_input, peticao_texto, peticao_arquivo, arquivo_preview, resultado_output, dica_contextual, peticao_aluno_output, peticao_corrigida_output]
+            outputs=[caso_input, peticao_texto, peticao_arquivo, arquivo_preview, resultado_output, dica_contextual, peticao_aluno_output, peticao_corrigida_output, export_btn, export_file]
         )
         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]
         )
 
         with gr.Accordion("📘 Critérios Detalhados", open=False):