Update app.py

app.py

@@ -38,16 +38,19 @@ CONFIG = {
 rate_limiter = RateLimiter(max_calls=60, period=60)
 
 DICAS_OAB = {
-    "Endereçamento": "Em relações de consumo, prefira o foro do domicílio do autor (art. 101, I, CDC).",
+    "Endereçamento": "Em relações de consumo, prefira o foro do domicílio do autor (Art. 101, I, CDC).",
     "Qualificação": "Mencione nome, estado civil (se informado), endereço (use 'Rua XXX' se não fornecido).",
     "Fatos": "Descreva os fatos cronologicamente, sem inventar informações.",
     "Pedidos": "Seja específico nos pedidos (ex.: 'condenação ao pagamento de R$ X').",
-    "Custas": "Inclua 'Requer a juntada da guia de custas devidamente recolhida'."
+    "Custas": "Inclua 'Requer a juntada da guia de custas devidamente recolhida'.",
+    "Fundamentação": "Baseie-se em artigos do CPC ou CDC e, se possível, em súmulas do STJ (ex.: Súmula 387 para danos morais).",
+    "Fechamento": "Finalize com 'Nestes termos pede deferimento', sem vírgula."
 }
 
 HISTORICO = []
 CORRECAO_CACHE = {}
 
+# Funções de pré-processamento e extração de texto permanecem inalteradas
 def preprocess_image_for_ocr(img):
     try:
         gray = img.convert('L')
@@ -166,20 +169,21 @@ def process_file(file_obj):
     except Exception as e:
         return f"❌ Erro ao processar o arquivo: {str(e)}"
 
+# Função de cálculo de pontuação aprimorada com explicações
 def calcular_pontuacao(enderecamento, fundamentos, pedidos, fechamento):
     tabela = [
         ("**Endereçamento**", ""),
         ("1. Correto endereçamento ao juízo competente", f"{enderecamento[0]:.2f} / 0,10"),
-        ("2. Qualificação das partes conforme Art. 319, II, CPC", f"{enderecamento[1]:.2f} / 0,20"),
+        ("2. Qualificação das partes (Art. 319, II, CPC)", f"{enderecamento[1]:.2f} / 0,20"),
         ("**Fundamentos**", ""),
-        ("3. Exposição clara dos fatos", f"{fundamentos[0]:.2f} / 0,20"),
+        ("3. Exposição clara dos fatos (Art. 319, III, CPC)", f"{fundamentos[0]:.2f} / 0,20"),
         ("4. Fundamentação jurídica adequada", f"{fundamentos[1]:.2f} / 0,25"),
-        ("5. Demonstração de ilícito e dano", f"{fundamentos[2]:.2f} / 0,25"),
+        ("5. Demonstração de ilícito e dano (Art. 186, CC)", f"{fundamentos[2]:.2f} / 0,25"),
         ("**Pedidos**", ""),
-        ("6. Pedidos claros e específicos", f"{pedidos[0]:.2f} / 0,30"),
-        ("7. Valor da causa indicado", f"{pedidos[1]:.2f} / 0,10"),
-        ("8. Provas pretendidas especificadas", f"{pedidos[2]:.2f} / 0,10"),
-        ("9. Opção sobre audiência de conciliação", f"{pedidos[3]:.2f} / 0,10"),
+        ("6. Pedidos claros e específicos (Art. 319, IV, CPC)", f"{pedidos[0]:.2f} / 0,30"),
+        ("7. Valor da causa indicado (Art. 319, V, CPC)", f"{pedidos[1]:.2f} / 0,10"),
+        ("8. Provas pretendidas (Art. 319, VI, CPC)", f"{pedidos[2]:.2f} / 0,10"),
+        ("9. Opção sobre audiência (Art. 319, VII, CPC)", f"{pedidos[3]:.2f} / 0,10"),
         ("10. Pedido de citação do réu", f"{pedidos[4]:.2f} / 0,10"),
         ("11. Juntada da guia de custas", f"{pedidos[5]:.2f} / 0,10"),
         ("**Fechamento**", ""),
@@ -192,17 +196,18 @@ def calcular_pontuacao(enderecamento, fundamentos, pedidos, fechamento):
 def validar_peticao_basica(peticao):
     problemas = []
     if "EXCELENTÍSSIMO" not in peticao.upper():
-        problemas.append("⚠️ Endereçamento ausente ou incompleto.")
+        problemas.append("⚠️ Endereçamento ausente ou incompleto (Art. 319, I, CPC).")
     if "Art. 319" not in peticao and "CPC" not in peticao:
-        problemas.append("⚠️ Fundamentação jurídica não menciona o CPC.")
+        problemas.append("⚠️ Fundamentação jurídica não menciona o CPC (Art. 319).")
     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.")
+        problemas.append("⚠️ Pedido de juntada da guia de custas ausente (-1,0 ponto).")
     if "Nestes termos" not in peticao:
         problemas.append("⚠️ Fechamento formal ausente.")
     return problemas
 
+# Prompt estruturado aprimorado com feedback pedagógico
 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)
@@ -214,12 +219,12 @@ def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
     ### Análise da Petição
     1. Análise Geral (Breve visão geral sobre a petição - máximo 3 linhas)
     2. Nota Final (0 a 10 com uma casa decimal)
-    3. Pontos Positivos (Liste pelo menos 3 aspectos bem executados)
-    4. Pontos a Melhorar (Liste os problemas encontrados em ordem de gravidade)
+    3. Pontos Positivos (Liste pelo menos 3 aspectos bem executados com justificativa legal)
+    4. Pontos a Melhorar (Liste os problemas encontrados em ordem de gravidade com explica��ão e base legal)
     5. Análise Específica dos Requisitos:
-       - Endereçamento e qualificação do juízo
+       - Endereçamento e qualificação do juízo (Art. 319, I, CPC; Art. 101, I, CDC se aplicável)
        - Qualificação das partes (Art. 319, II, CPC)
-       - Fatos e fundamentos jurídicos (Art. 319, III, CPC)
+       - Fatos e fundamentos jurídicos (Art. 319, III, CPC; Art. 186, CC)
        - Pedidos com especificações (Art. 319, IV, CPC)
        - Valor da causa (Art. 319, V, CPC)
        - Provas com que o autor pretende demonstrar a verdade dos fatos (Art. 319, VI, CPC)
@@ -236,7 +241,7 @@ def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
     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.
+    - 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.
     **Pontuação Detalhada:**
     Inclua uma tabela com:
@@ -263,11 +268,11 @@ def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
     ---
 
     ### Petição Corrigida
-    Forneça a petição do aluno reescrita com todas as correções aplicadas, incluindo partes omitidas (como pedidos ou custas) e ajustando erros identificados nos "Pontos a Melhorar" e "Análise Específica". Use apenas informações do caso e fundamente com leis brasileiras relevantes (ex.: CPC, CDC). Estruture a petição corrigida com:
+    Forneça a petição do aluno reescrita com todas as correções aplicadas, incluindo partes omitidas (como pedidos ou custas) e ajustando erros identificados nos "Pontos a Melhorar" e "Análise Específica". Use apenas informações do caso e fundamente com leis brasileiras relevantes (ex.: CPC, CDC, Súmula 387 do STJ para danos morais). Estruture a petição corrigida com:
     1. Endereçamento correto
     2. Qualificação completa (Art. 319, II, CPC)
     3. Fatos claros (Art. 319, III, CPC)
-    4. Fundamentação jurídica (ex.: CPC, CDC)
+    4. Fundamentação jurídica (ex.: CPC, CDC, Art. 186, CC)
     5. Pedidos específicos (Art. 319, IV, CPC)
     6. Valor da causa (Art. 319, V, CPC)
     7. Provas (Art. 319, VI, CPC)
@@ -284,10 +289,13 @@ def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
           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.
+        - Verifique a coerência entre os fatos do caso e a petição, destacando inconsistências.
+        - Sugira estratégias jurídicas adicionais baseadas no caso (ex.: citar Art. 186, CC ou Súmula 387, STJ para danos morais).
         - Não resuma; detalhe cada erro de forma pedagógica e clara.
         """
     return base_prompt
 
+# Funções de geração de caso e correção permanecem similares
 def estruturar_prompt_geracao_caso():
     return """
     Crie um caso fictício detalhado para uma petição inicial cível, similar ao exemplo:
@@ -403,6 +411,8 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
                 "peticao": peticao_conteudo,
                 "correcao_completa": correção_completa
             })
+            if len(HISTORICO) > 10:  # Limite de histórico
+                HISTORICO.pop(0)
             CORRECAO_CACHE[cache_key] = correção_formatada
             return aviso_validacao + correção_formatada
         except requests.Timeout:
@@ -418,6 +428,7 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
         except Exception as e:
             return aviso_validacao + f"❌ Erro inesperado: {str(e)}"
 
+# Função de formatação com mais detalhes pedagógicos
 def formatar_correção(texto):
     linhas = texto.split('\n')
     texto_formatado = ""
@@ -431,7 +442,7 @@ def formatar_correção(texto):
         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"
+            texto_formatado += f"**🔴 {linha.strip()} (-5,0 pontos)**\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 or "Exemplo" in linha:
@@ -479,6 +490,7 @@ def calcular_progresso():
     tendencia = "⬆️ Melhorando" if len(notas) > 1 and notas[-1] > notas[-2] else "⬇️ Estável ou piorando"
     return f"📊 Progresso: Média {media:.1f}/10 | Tentativas: {len(notas)} | Tendência: {tendencia}"
 
+# Função de comparação aprimorada com feedback visual
 def comparar_peticao_melhorada(caso, peticao_texto, correcao):
     if not caso:
         return "⚠️ Forneça um caso antes de comparar.", "", "", "", ""
@@ -511,12 +523,12 @@ def comparar_peticao_melhorada(caso, peticao_texto, correcao):
         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}")
+                criticas.append(f"🔴 REMOVIDO: {texto} (inconsistente ou incorreto)")
             resultado_diff.append(f"<div class='linha-removida'><del>{escape(texto)}</del></div>")
         elif linha.startswith('+ '):
             texto = linha[2:]
             if any(kw in texto.lower() for kw in ["custas", "art. 319"]):
-                criticas.append(f"🟢 ADICIONADO: {texto}")
+                criticas.append(f"🟢 ADICIONADO: {texto} (essencial para conformidade)")
             resultado_diff.append(f"<div class='linha-adicionada'><ins>{escape(texto)}</ins></div>")
         elif linha.startswith('? '):
             continue
@@ -542,6 +554,7 @@ Diferença Líquida: {total_adicionadas - total_removidas} linhas
 """
     return peticao, peticao_corrigida, resumo_estatistico, criticas_texto, visualizacao_detalhada
 
+# Função de exportação corrigida para o problema do download
 def exportar_diferencas_html(caso, peticao_texto, correcao):
     _, _, resumo_estatistico, criticas_texto, visualizacao_detalhada = comparar_peticao_melhorada(caso, peticao_texto, correcao)
     html_content = f"""
@@ -567,14 +580,16 @@ def exportar_diferencas_html(caso, peticao_texto, correcao):
 </body>
 </html>
 """
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf", mode='w') as tmp:
-        try:
-            HTML(string=html_content).write_pdf(tmp.name)
-            return tmp.name
-        except Exception as e:
-            return f"❌ Erro ao gerar PDF: {str(e)}"
-        finally:
-            os.unlink(tmp.name) if os.path.exists(tmp.name) else None
+    try:
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
+            temp_path = tmp.name
+            HTML(string=html_content).write_pdf(temp_path)
+            return temp_path
+    except Exception as e:
+        return None
+    finally:
+        if 'temp_path' in locals() and os.path.exists(temp_path):
+            os.unlink(temp_path)
 
 def interface_gradio():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -584,13 +599,13 @@ def interface_gradio():
             with gr.Column(scale=1):
                 gr.Markdown("""
                 ### 🔍 Critérios de Avaliação
-                - ✅ Endereçamento correto
-                - ✅ Qualificação das partes
-                - ✅ Fatos e fundamentos
-                - ✅ Pedidos claros e específicos
-                - ✅ Valor da causa
-                - ✅ Provas pretendidas
-                - ✅ Opção sobre conciliação
+                - ✅ Endereçamento correto (Art. 319, I, CPC)
+                - ✅ Qualificação das partes (Art. 319, II, CPC)
+                - ✅ Fatos e fundamentos (Art. 319, III, CPC)
+                - ✅ Pedidos claros e específicos (Art. 319, IV, CPC)
+                - ✅ Valor da causa (Art. 319, V, CPC)
+                - ✅ Provas pretendidas (Art. 319, VI, CPC)
+                - ✅ Opção sobre conciliação (Art. 319, VII, CPC)
                 - ✅ Pedido de citação adequado
                 - ✅ Juntada da guia de custas
                 - ✅ Fechamento formal
@@ -598,7 +613,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
+                - ❌ Identificação específica na OAB: nota 0,0
                 """)
                 modo_treinamento = gr.Checkbox(label="Modo de Treinamento Interativo", value=False)
                 with gr.Accordion("⚙️ Configurações", open=False):
@@ -695,7 +710,7 @@ def interface_gradio():
             inputs=[caso_input, peticao_texto, resultado_output],
             outputs=[download_file]
         ).then(
-            fn=lambda x: [gr.update(visible=True, value=x), gr.update(visible=True)],
+            fn=lambda x: [gr.update(visible=True, value=x) if x else gr.update(visible=True, value=None, label="Erro ao gerar PDF"), gr.update(visible=True)],
             outputs=[download_file, download_btn]
         )
 
@@ -718,7 +733,7 @@ def interface_gradio():
             """)
         gr.Markdown("""
         ### ℹ️ Modo de Treinamento Interativo
-        Ative para explicações detalhadas.
+        Ative para explicações detalhadas com exemplos práticos e sugestões.
         ### 📋 Como usar
         1. Gere ou cole o caso
         2. Insira a petição