Update app.py

app.py

@@ -18,9 +18,26 @@ from ratelimiter import RateLimiter
 import difflib
 from html import escape
 from weasyprint import HTML  # Para gerar PDF com cores
+import shelve
+import logging
+import re
+
+# Configuração de logging
+logging.basicConfig(filename='app.log', level=logging.ERROR, format='%(asctime)s - %(levelname)s - %(message)s')
 
 load_dotenv()
 
+# Verificação de dependências
+def verificar_dependencias():
+    dependencias = {
+        "gradio": gr, "requests": requests, "pdfplumber": pdfplumber_open, 
+        "pytesseract": pytesseract, "cv2": cv2, "weasyprint": HTML
+    }
+    for nome, modulo in dependencias.items():
+        if modulo is None:
+            raise ImportError(f"❌ Módulo {nome} não encontrado. Instale com 'pip install {nome}'.")
+verificar_dependencias()
+
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 if not GEMINI_API_KEY:
     raise ValueError("❌ A chave GEMINI_API_KEY não foi encontrada. Configure-a como Secret no Hugging Face Spaces.")
@@ -38,47 +55,30 @@ CONFIG = {
 rate_limiter = RateLimiter(max_calls=60, period=60)
 
 DICAS_OAB_MELHORADO = {
-    "Endereçamento": "Em relações de consumo, prefira o foro do domicílio do autor (Art. 101, I, CDC). "
-                     "Exemplo: 'Excelentíssimo Senhor Juiz de Direito da Vara Cível da Comarca de [Cidade do Autor]'.",
-
-    "Qualificação": "Mencione nome, estado civil (se informado), endereço (use 'Rua XXX' se não fornecido). "
-                    "Dica: Sempre verifique se todos os dados das partes estão corretos e completos.",
-
-    "Fatos": "Descreva os fatos cronologicamente, sem inventar informações. "
-              "Exemplo: 'Em [data], o autor celebrou contrato com o réu para a prestação de serviços de [descrição]'.",
-
-    "Pedidos": "Seja específico nos pedidos (ex.: 'condenação ao pagamento de R$ X'). "
-               "Dica: Inclua pedidos de danos morais e materiais, se aplicável, com base em precedentes.",
-
-    "Custas": "Inclua 'Requer a juntada da guia de custas devidamente recolhida'. "
-              "Dica: Nunca se esqueça de mencionar o recolhimento das custas para evitar a extinção do processo.",
-
-    "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). "
-                     "Exemplo: 'Conforme o artigo 186 do Código Civil, a responsabilidade civil pode ser caracterizada'.",
-
-    "Fechamento": "Finalize com 'Nestes termos pede deferimento', sem vírgula. "
-                   "Dica: Evite erros de pontuação que podem comprometer a formalidade do documento.",
-
-    "Estratégia Jurídica": "Utilize precedentes e súmulas para fortalecer sua argumentação. "
-                           "Exemplo: 'Conforme a Súmula 387 do STJ, os danos morais devem ser comprovados e quantificados'.",
-
-    "Provas": "Especifique as provas que pretende produzir. "
-              "Dica: Inclua testemunhas, documentos e perícias, se necessário, para comprovar os fatos alegados.",
-
-    "Contra-argumentação": "Antecipe possíveis defesas da parte contrária e prepare respostas. "
-                          "Dica: Considere os argumentos que a outra parte pode usar e esteja preparado para refutá-los."
+    "Endereçamento": "Em relações de consumo, prefira o foro do domicílio do autor (Art. 101, I, CDC). Exemplo: 'Excelentíssimo Senhor Juiz de Direito da Vara Cível da Comarca de [Cidade do Autor]'.",
+    "Qualificação": "Mencione nome, estado civil (se informado), endereço (use 'Rua XXX' se não fornecido). Dica: Sempre verifique se todos os dados das partes estão corretos e completos.",
+    "Fatos": "Descreva os fatos cronologicamente, sem inventar informações. Exemplo: 'Em [data], o autor celebrou contrato com o réu para a prestação de serviços de [descrição]'.",
+    "Pedidos": "Seja específico nos pedidos (ex.: 'condenação ao pagamento de R$ X'). Dica: Inclua pedidos de danos morais e materiais, se aplicável, com base em precedentes.",
+    "Custas": "Inclua 'Requer a juntada da guia de custas devidamente recolhida'. Dica: Nunca se esqueça de mencionar o recolhimento das custas para evitar a extinção do processo.",
+    "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). Exemplo: 'Conforme o artigo 186 do Código Civil, a responsabilidade civil pode ser caracterizada'.",
+    "Fechamento": "Finalize com 'Nestes termos pede deferimento', sem vírgula. Dica: Evite erros de pontuação que podem comprometer a formalidade do documento.",
+    "Estratégia Jurídica": "Utilize precedentes e súmulas para fortalecer sua argumentação. Exemplo: 'Conforme a Súmula 387 do STJ, os danos morais devem ser comprovados e quantificados'.",
+    "Provas": "Especifique as provas que pretende produzir. Dica: Inclua testemunhas, documentos e perícias, se necessário, para comprovar os fatos alegados.",
+    "Contra-argumentação": "Antecipe possíveis defesas da parte contrária e prepare respostas. Dica: Considere os argumentos que a outra parte pode usar e esteja preparado para refutá-los."
 }
 
 HISTORICO = []
-CORRECAO_CACHE = {}
 
 def preprocess_image_for_ocr(img):
     try:
         gray = img.convert('L')
         np_img = np.array(gray)
-        _, thresh = cv2.threshold(np_img, 150, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
+        clahe = cv2.createCLAHE(clipLimit=2.0, tileGridSize=(8, 8))
+        enhanced = clahe.apply(np_img)
+        _, thresh = cv2.threshold(enhanced, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
         return Image.fromarray(thresh)
     except Exception as e:
+        logging.error(f"Erro no pré-processamento de imagem: {str(e)}")
         return f"❌ Erro no pré-processamento: {str(e)}"
 
 def extract_text_from_pdf(file_path_or_obj):
@@ -114,6 +114,7 @@ def extract_text_from_pdf(file_path_or_obj):
                             text += f"[Erro na página {i+1}: {str(e)}]\n"
         return text if text.strip() else "⚠️ Não foi possível extrair texto do PDF."
     except Exception as e:
+        logging.error(f"Erro ao processar PDF: {str(e)}")
         return f"❌ Erro ao processar o PDF: {str(e)}"
 
 def extract_text_from_docx(file_path_or_obj):
@@ -141,6 +142,7 @@ def extract_text_from_docx(file_path_or_obj):
                     text += "\n"
         return text if text.strip() else "⚠️ Não foi possível extrair texto do DOCX."
     except Exception as e:
+        logging.error(f"Erro ao processar DOCX: {str(e)}")
         return f"❌ Erro ao processar o DOCX: {str(e)}"
 
 def extract_text_from_txt(file_path_or_obj):
@@ -165,9 +167,11 @@ def extract_text_from_txt(file_path_or_obj):
                     continue
             return "❌ Não foi possível decodificar o arquivo TXT."
     except Exception as e:
+        logging.error(f"Erro ao processar TXT: {str(e)}")
         return f"❌ Erro ao processar o TXT: {str(e)}"
 
 def process_file(file_obj):
+    """Processa arquivos PDF, DOCX ou TXT e extrai texto."""
     if file_obj is None:
         return "Nenhum arquivo enviado."
     try:
@@ -188,6 +192,7 @@ def process_file(file_obj):
         else:
             return "❌ Erro: Objeto de arquivo inválido."
     except Exception as e:
+        logging.error(f"Erro ao processar arquivo: {str(e)}")
         return f"❌ Erro ao processar o arquivo: {str(e)}"
 
 def calcular_pontuacao(enderecamento, fundamentos, pedidos, fechamento):
@@ -214,105 +219,175 @@ def calcular_pontuacao(enderecamento, fundamentos, pedidos, fechamento):
     return f"{tabela_formatada}\n\n**Nota Final:** {nota_final:.2f} / 10,00"
 
 def validar_peticao_basica(peticao):
+    """Valida os elementos básicos da petição com base no Art. 319 do CPC e critérios OAB."""
     problemas = []
-    if "EXCELENTÍSSIMO" not in peticao.upper():
+    peticao_upper = peticao.upper()
+    peticao_lower = peticao.lower()
+
+    if "EXCELENTÍSSIMO" not in peticao_upper or "JUIZ" not in peticao_upper:
         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 (Art. 319).")
-    if "Dá-se à causa o valor" not in peticao:
+    if "AUTOR" not in peticao_upper or "RÉU" not in peticao_upper:
+        problemas.append("⚠️ Qualificação incompleta: 'autor' ou 'réu' não mencionados (Art. 319, II, CPC).")
+    if "FATO" not in peticao_upper and "FUNDAMENTO" not in peticao_upper:
+        problemas.append("⚠️ Exposição de fatos ou fundamentos ausente (Art. 319, III, CPC).")
+    if "ART." not in peticao_upper or "CPC" not in peticao_upper:
+        problemas.append("⚠️ Fundamentação jurídica não menciona artigos ou CPC (Art. 319, III, CPC).")
+    if "REQUER" not in peticao_upper and "PEDE" not in peticao_upper:
+        problemas.append("⚠️ Pedidos não especificados ou ausentes (Art. 319, IV, CPC).")
+    if "DÁ-SE À CAUSA O VALOR" not in peticao_upper:
         problemas.append("⚠️ Valor da causa não especificado (Art. 319, V, CPC).")
-    if "juntada da guia de custas" not in peticao.lower():
+    if "R$" not in peticao_upper and "VALOR" not in peticao_upper:
+        problemas.append("⚠️ Valor da causa sem quantia monetária explícita (Art. 319, V, CPC).")
+    if "PROVA" not in peticao_upper and "TESTEMUNHA" not in peticao_upper and "DOCUMENTO" not in peticao_upper:
+        problemas.append("⚠️ Provas pretendidas não especificadas (Art. 319, VI, CPC).")
+    if "AUDIÊNCIA" not in peticao_upper and "CONCILIAÇÃO" not in peticao_upper:
+        problemas.append("⚠️ Opção sobre audiência de conciliação não mencionada (Art. 319, VII, CPC).")
+    if "CITAÇÃO" not in peticao_upper:
+        problemas.append("⚠️ Pedido de citação do réu ausente.")
+    if "juntada da guia de custas" not in peticao_lower:
         problemas.append("⚠️ Pedido de juntada da guia de custas ausente (-1,0 ponto).")
-    if "Nestes termos" not in peticao:
-        problemas.append("⚠️ Fechamento formal ausente.")
+    if "NESTES TERMOS" not in peticao_upper:
+        problemas.append("⚠️ Fechamento formal 'Nestes termos pede deferimento' ausente.")
+    if "NESTES TERMOS," in peticao_upper:
+        problemas.append("⚠️ Uso incorreto de vírgula em 'Nestes termos, pede deferimento'.")
+    if "ADVOGADO" not in peticao_upper or "OAB" not in peticao_upper:
+        problemas.append("⚠️ Fechamento sem 'ADVOGADO' ou 'OAB'.")
+    if "LOCAL E DATA" not in peticao_upper:
+        problemas.append("⚠️ Fechamento sem 'Local e data'.")
+    if any(id in peticao_upper for id in ["CPF", "RG", "MATRÍCULA"]):
+        problemas.append("⚠️ Identificação indevida do candidato detectada (nota máxima 3,0).")
+    if "OAB/" in peticao_upper or any(char.isdigit() for char in peticao.split("OAB")[-1] if "OAB" in peticao_upper):
+        problemas.append("⚠️ Identificação específica na OAB (ex.: número ou estado) detectada (nota 0,0).")
+
     return problemas
 
+def verificar_coerencia_caso_peticao(caso, peticao):
+    """Verifica coerência entre o caso e a petição, detectando omissões e invenções."""
+    incoerencias = []
+    caso_lower = caso.lower()
+    peticao_lower = peticao.lower()
+
+    nomes_caso = [word for word in caso.split() if word.istitle() and len(word) > 2]
+    nomes_peticao = [word for word in peticao.split() if word.istitle() and len(word) > 2]
+    for nome in nomes_caso:
+        if nome.lower() not in peticao_lower:
+            incoerencias.append(f"⚠️ Nome '{nome}' do caso não mencionado na petição (suspeita de omissão).")
+    for frase in peticao.split("."):
+        if frase.strip() and not any(palavra.lower() in caso_lower for palavra in frase.split() if len(palavra) > 3):
+            incoerencias.append(f"⚠️ Frase suspeita de invenção: '{frase.strip()}' (não consta no caso).")
+    valores_caso = re.findall(r"R\$\s*\d+(?:\.\d+)?(?:,\d+)?", caso)
+    valores_peticao = re.findall(r"R\$\s*\d+(?:\.\d+)?(?:,\d+)?", peticao)
+    for valor in valores_peticao:
+        if valor not in valores_caso:
+            incoerencias.append(f"⚠️ Valor '{valor}' na petição não consta no caso (suspeita de invenção).")
+
+    return incoerencias
+
+def validar_resposta_api(correcao_completa):
+    """Valida a qualidade da resposta da API."""
+    problemas_api = []
+    if "Nota Final" not in correcao_completa:
+        problemas_api.append("⚠️ Resposta da API sem 'Nota Final'.")
+    if "Pontos a Melhorar" not in correcao_completa:
+        problemas_api.append("⚠️ Resposta da API sem 'Pontos a Melhorar'.")
+    if "Análise Específica dos Requisitos" not in correcao_completa:
+        problemas_api.append("⚠️ Resposta da API sem 'Análise Específica dos Requisitos'.")
+    if "### Petição Corrigida" not in correcao_completa:
+        problemas_api.append("⚠️ Resposta da API sem 'Petição Corrigida'.")
+    return problemas_api
+
 def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
+    """Estrutura um prompt avançado e especializado para correção de petições."""
     base_prompt = f"""
-    ## Correção Avançada de Petição Inicial Cível (Critérios Realistas da OAB)
+    ## Correção Avançada de Petição Inicial Cível (Critérios Realistas da OAB com Análise Especializada)
     **Caso Fornecido:**
     {caso}
     **Petição do Aluno:**
     {peticao}
     **Estrutura da Avaliação:**
     ### Análise da Petição
-    1. Análise Geral (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 com justificativa legal)
-    4. Pontos a Melhorar (Liste os problemas encontrados em ordem de gravidade com explicação e base legal)
+    1. Análise Geral (Visão geral sobre a petição - máximo 3 linhas, destacando adequação ao caso e consistência jurídica)
+    2. Nota Final (0 a 10 com uma casa decimal, justificando brevemente a pontuação total)
+    3. Pontos Positivos (Liste pelo menos 3 aspectos bem executados com justificativa legal detalhada, incluindo artigos, súmulas ou precedentes aplicáveis)
+    4. Pontos a Melhorar (Liste todos os problemas encontrados em ordem de gravidade, com explicação detalhada, base legal e impacto na nota; não omita erros menores)
     5. Análise Específica dos Requisitos:
-       - 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; 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)
-       - Opção pela realização ou não de audiência de conciliação (Art. 319, VII, CPC)
-       - Pedido de citação do réu
-       - Juntada da guia de custas
-       - Fechamento formal da petição
+       - Endereçamento e qualificação do juízo (Art. 319, I, CPC; Art. 101, I, CDC se aplicável): Verifique a competência territorial e material com precisão.
+       - Qualificação das partes (Art. 319, II, CPC): Avalie nomes, estado civil, endereço e consistência com o caso.
+       - Fatos e fundamentos jurídicos (Art. 319, III, CPC; Art. 186, CC): Compare os fatos narrados com o caso e analise a pertinência dos fundamentos legais.
+       - Pedidos com especificações (Art. 319, IV, CPC): Examine clareza, especificidade e adequação dos pedidos ao caso.
+       - Valor da causa (Art. 319, V, CPC): Confirme se o valor é realista e compatível com os danos alegados.
+       - Provas com que o autor pretende demonstrar a verdade dos fatos (Art. 319, VI, CPC): Avalie a suficiência e relevância das provas indicadas.
+       - Opção pela realização ou não de audiência de conciliação (Art. 319, VII, CPC): Verifique a menção explícita e sua coerência com o caso.
+       - Pedido de citação do réu: Analise se está claro e conforme os meios legais (ex.: Art. 246, CPC).
+       - Juntada da guia de custas: Confirme a presença literal do pedido e sua redação formal.
+       - Fechamento formal da petição: Certifique-se da estrutura exata "Local e data.\nADVOGADO.\nOAB." sem variações ou identificação indevida.
+
     **Critérios Fundamentais:**
-    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
-    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.
+    1. Identificação do candidato (nome, CPF, RG ou número real na seção final) resulta em nota máxima de 3,0; forneça exemplo do trecho problemático.
+    2. Invenção de fatos não constantes no caso resulta em penalização grave (-5,0 pontos); identifique cada fato inventado e compare com o caso.
+    3. Falta de pedido de juntada de custas resulta em perda de 1,0 ponto; verifique a ausência ou redação inadequada.
+    4. Use marcadores para dados não fornecidos (ex.: "Rua XXX", "CPF XXX"); penalize invenção de dados específicos (-0,5 por ocorrência).
+    5. Não penalize por informações ausentes no caso original, mas alerte se a omissão compromete a petição.
+    6. A frase "Nestes termos, pede deferimento" com vírgula é incorreta; corrija para "Nestes termos pede deferimento" e penalize com -0,2 ponto, destacando o erro formal.
+    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.: "OAB/SP" ou "12345"), zere a nota (0,0) e explique o motivo.
+    
+    **Instruções Especiais para Análise Avançada:**
+    - **Competência**: Avalie o foro indicado com base no Art. 46 do CPC (domicílio do réu) ou Art. 101, I do CDC (domicílio do autor em relações de consumo); penalize erros com -0,10 ponto e justifique com jurisprudência (ex.: STJ, REsp 1.234.567).
+    - **Consistência Fática**: Compare cada fato narrado na petição com o caso; destaque omissões ou distorções com base no Art. 319, III, CPC.
+    - **Fundamentação Jurídica**: Exija citação de pelo menos um artigo específico (ex.: Art. 186, CC para responsabilidade civil) e avalie a pertinência; penalize generalidades com -0,15 ponto por item vago.
+    - **Danos Morais e Materiais**: Se o caso sugerir danos, verifique se os pedidos estão quantificados e fundamentados (ex.: Súmula 387/STJ para danos morais); penalize omissões com -0,20 ponto.
+    - **Linguagem Técnica**: Avalie a formalidade e precisão do vocabulário jurídico; penalize informalidades ou erros gramaticais (ex.: "vossa excelência" em minúsculas) com -0,05 ponto por ocorrência.
+    - **Precedentes**: Inclua referências a súmulas ou decisões relevantes (ex.: Súmula 297/STJ para CDC) quando aplicável no caso.
+
     **Pontuação Detalhada:**
     Inclua uma tabela com:
-    | ITEM                                      | PONTUAÇÃO |
-    |-------------------------------------------|-----------|
-    **Endereçamento**                          |           |
-    1. Correto endereçamento ao juízo competente (0,10) | / 0,10 |
-    2. Qualificação das partes conforme Art. 319, II, CPC (0,20) | / 0,20 |
-    **Fundamentos**                            |           |
-    3. Exposição clara dos fatos (0,20)       | / 0,20 |
-    4. Fundamentação jurídica adequada (0,25)  | / 0,25 |
-    5. Demonstração de ilícito e dano (0,25)  | / 0,25 |
-    **Pedidos**                                |           |
-    6. Pedidos claros e específicos (0,30)    | / 0,30 |
-    7. Valor da causa indicado (0,10)         | / 0,10 |
-    8. Provas pretendidas especificadas (0,10) | / 0,10 |
-    9. Opção sobre audiência de conciliação (0,10) | / 0,10 |
-    10. Pedido de citação do réu (0,10)       | / 0,10 |
-    11. Juntada da guia de custas (0,10)     | / 0,10 |
-    **Fechamento**                             |           |
-    12. Fechamento formal adequado (0,10)     | / 0,10 |
-    **Nota Final:**
-    Calcule a nota final somando os pontos, exceto se houver identificação específica na OAB, que zera a nota.
-    ---
+    | ITEM                                      | PONTUAÇÃO | JUSTIFICATIVA |
+    |-------------------------------------------|-----------|---------------|
+    **Endereçamento**                          |           |               |
+    1. Correto endereçamento ao juízo competente (0,10) | / 0,10 | [Explique erro ou acerto] |
+    2. Qualificação das partes conforme Art. 319, II, CPC (0,20) | / 0,20 | [Explique erro ou acerto] |
+    **Fundamentos**                            |           |               |
+    3. Exposição clara dos fatos (0,20)       | / 0,20 | [Explique erro ou acerto] |
+    4. Fundamentação jurídica adequada (0,25) | / 0,25 | [Explique erro ou acerto] |
+    5. Demonstração de ilícito e dano (0,25)  | / 0,25 | [Explique erro ou acerto] |
+    **Pedidos**                                |           |               |
+    6. Pedidos claros e específicos (0,30)    | / 0,30 | [Explique erro ou acerto] |
+    7. Valor da causa indicado (0,10)         | / 0,10 | [Explique erro ou acerto] |
+    8. Provas pretendidas especificadas (0,10)| / 0,10 | [Explique erro ou acerto] |
+    9. Opção sobre audiência de conciliação (0,10) | / 0,10 | [Explique erro ou acerto] |
+    10. Pedido de citação do réu (0,10)       | / 0,10 | [Explique erro ou acerto] |
+    11. Juntada da guia de custas (0,10)      | / 0,10 | [Explique erro ou acerto] |
+    **Fechamento**                             |           |               |
+    12. Fechamento formal adequado (0,10)     | / 0,10 | [Explique erro ou acerto] |
+    **Nota Final:** Calcule a soma dos pontos, aplicando penalizações cumulativas conforme os critérios fundamentais; justifique qualquer desconto adicional.
 
     ### 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, 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, 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)
-    8. Opção por audiência (Art. 319, VII, CPC)
-    9. Pedido de citação
-    10. Juntada da guia de custas
-    11. Fechamento formal com "Local e data.\nADVOGADO.\nOAB." sem identificação específica
+    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. Siga rigorosamente:
+    1. Endereçamento correto (com foro justificado por CPC ou CDC).
+    2. Qualificação completa (Art. 319, II, CPC; use "XXX" para dados ausentes).
+    3. Fatos claros e fiéis ao caso (Art. 319, III, CPC).
+    4. Fundamentação jurídica robusta (cite CPC, CDC, CC e súmulas aplicáveis, como Súmula 387/STJ para danos morais).
+    5. Pedidos específicos e quantificados (Art. 319, IV, CPC).
+    6. Valor da causa realista (Art. 319, V, CPC).
+    7. Provas detalhadas (Art. 319, VI, CPC; ex.: "prova documental e testemunhal").
+    8. Opção por audiência explícita (Art. 319, VII, CPC).
+    9. Pedido de citação formal (ex.: "via postal, nos termos do Art. 246, I, CPC").
+    10. Juntada da guia de custas com redação precisa.
+    11. Fechamento formal com "Local e data.\nADVOGADO.\nOAB." sem identificação específica.
     """
     if modo_treinamento:
         base_prompt += f"""
         **Modo de Treinamento Interativo Avançado:**
         - 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 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_MELHORADO, ensure_ascii=False)}) ou personalizada.
-          e) Uma estratégia jurídica avançada que poderia ser aplicada ao caso.
-        - 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.
-        - Inclua referências a artigos de lei, jurisprudência e doutrina relevantes.
+          a) Explicação detalhada (mínimo 3 frases) com base legal (ex.: Art. 319, CPC) e análise do impacto processual.
+          b) Sugestão de redação corrigida baseada no caso, com linguagem técnica exemplar.
+          c) Exemplo prático em outro contexto (ex.: "Em ação consumerista, ‘Requer a inversão do ônus da prova, nos termos do Art. 6º, VIII, CDC.’").
+          d) Dica prática do banco ({json.dumps(DICAS_OAB_MELHORADO, ensure_ascii=False)}) ou personalizada, com enfoque pedagógico.
+          e) Estratégia jurídica avançada (ex.: "Cite o Art. 944, CC para quantificar danos ou peça tutela antecipada com base no Art. 300, CPC").
+        - Verifique a coerência entre os fatos do caso e a petição, listando todas as inconsistências com trechos específicos.
+        - Sugira estratégias adicionais baseadas no caso (ex.: cumulação de pedidos, Art. 327, CPC; ou danos morais com Súmula 387/STJ).
+        - Detalhe cada erro pedagogicamente, com referências a leis, súmulas (ex.: Súmula 54/STJ para lucros cessantes), e doutrina (ex.: Nelson Nery Jr. sobre Art. 319, CPC).
+        - Avalie a viabilidade processual da petição (ex.: risco de indeferimento por inépcia, Art. 330, CPC).
         """
     return base_prompt
 
@@ -374,9 +449,33 @@ def gerar_caso_ficticio():
                 continue
             return f"❌ Erro ao chamar a API: {e}"
         except Exception as e:
+            logging.error(f"Erro inesperado ao gerar caso: {str(e)}")
             return f"❌ Erro inesperado: {str(e)}"
 
+def sanitize_input(text):
+    """Sanitiza entradas para evitar injeção."""
+    return escape(text.strip()) if text else ""
+
+def salvar_cache(chave, valor):
+    """Salva correção no cache persistente."""
+    try:
+        with shelve.open("correcao_cache.db") as db:
+            db[chave] = valor
+    except Exception as e:
+        logging.error(f"Erro ao salvar no cache: {str(e)}")
+
+def carregar_cache(chave):
+    """Carrega correção do cache persistente."""
+    try:
+        with shelve.open("correcao_cache.db") as db:
+            return db.get(chave)
+    except Exception as e:
+        logging.error(f"Erro ao carregar do cache: {str(e)}")
+        return None
+
 def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treinamento=False):
+    """Corrige uma petição inicial cível com base em critérios da OAB."""
+    caso = sanitize_input(caso)
     if not caso.strip():
         return "⚠️ Forneça o caso da banca."
     if peticao_arquivo is not None:
@@ -384,16 +483,20 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
         if peticao_conteudo.startswith("❌") or peticao_conteudo.startswith("⚠️"):
             return peticao_conteudo
     else:
-        peticao_conteudo = peticao_texto or ""
+        peticao_conteudo = sanitize_input(peticao_texto) or ""
     if not peticao_conteudo.strip():
         return "⚠️ Forneça uma petição."
 
     problemas = validar_peticao_basica(peticao_conteudo)
+    incoerencias = verificar_coerencia_caso_peticao(caso, peticao_conteudo)
     aviso_validacao = f"### Validação Inicial\n\n" + "\n".join(problemas) + "\n\n" if problemas else ""
+    if incoerencias:
+        aviso_validacao += f"### Verificação de Coerência\n\n" + "\n".join(incoerencias) + "\n\n"
 
     cache_key = f"{caso[:50]}|{peticao_conteudo[:50]}|{modo_treinamento}"
-    if cache_key in CORRECAO_CACHE:
-        return aviso_validacao + CORRECAO_CACHE[cache_key]
+    cached_result = carregar_cache(cache_key)
+    if cached_result:
+        return aviso_validacao + cached_result
 
     prompt = estruturar_prompt_correcao(caso, peticao_conteudo, modo_treinamento)
     payload = {
@@ -419,6 +522,9 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
                     continue
                 return aviso_validacao + "❌ A API retornou uma resposta vazia. Tente novamente."
             correcao_completa = response_data["candidates"][0]["content"]["parts"][0]["text"]
+            problemas_api = validar_resposta_api(correcao_completa)
+            if problemas_api:
+                return aviso_validacao + f"### Problemas na Resposta da API\n\n" + "\n".join(problemas_api) + "\n\n⚠️ Tente novamente."
             if "### Peti�o Corrigida" in correcao_completa:
                 analise_fim = correcao_completa.index("### Petição Corrigida")
                 analise = correcao_completa[:analise_fim].strip()
@@ -431,9 +537,9 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
                 "peticao": peticao_conteudo,
                 "correcao_completa": correcao_completa
             })
-            if len(HISTORICO) > 10:  # Limite de histórico
+            if len(HISTORICO) > 10:
                 HISTORICO.pop(0)
-            CORRECAO_CACHE[cache_key] = correcao_formatada
+            salvar_cache(cache_key, correcao_formatada)
             return aviso_validacao + correcao_formatada
         except requests.Timeout:
             if tentativa < CONFIG["max_retry_attempts"] - 1:
@@ -446,6 +552,7 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
                 continue
             return aviso_validacao + f"❌ Erro ao chamar a API: {e}"
         except Exception as e:
+            logging.error(f"Erro inesperado ao corrigir petição: {str(e)}")
             return aviso_validacao + f"❌ Erro inesperado: {str(e)}"
 
 def formatar_correcao(texto):
@@ -469,7 +576,6 @@ def formatar_correcao(texto):
         else:
             texto_formatado += f"{linha}\n"
     if "Nota Final" in texto_formatado:
-        import re
         nota_pattern = r"### Nota Final.*?(\d+[.,]\d+|\d+)"
         match = re.search(nota_pattern, texto_formatado, re.DOTALL)
         if match:
@@ -479,20 +585,15 @@ def formatar_correcao(texto):
 
 def extrair_peticao_corrigida(correcao_completa):
     try:
-        # Verifica se a seção "### Petição Corrigida" está presente na resposta
         if "### Petição Corrigida" in correcao_completa:
             inicio = correcao_completa.index("### Petição Corrigida") + len("### Petição Corrigida")
             peticao_texto = correcao_completa[inicio:].strip()
-            # Encontra o final da petição corrigida, caso haja outra seção marcada com "###"
             fim = peticao_texto.find("###") if "###" in peticao_texto else len(peticao_texto)
             peticao_texto = peticao_texto[:fim].strip()
             return peticao_texto
-        else:
-            # Se a seção não for encontrada, tenta extrair a petição corrigida de outra forma
-            # Aqui, você pode adicionar lógica adicional para tentar localizar a petição corrigida
-            # de outras maneiras, se necessário.
-            return "⚠️ A seção '### Petição Corrigida' não foi encontrada na resposta da API. Verifique a resposta e tente novamente."
+        return "⚠️ A seção '### Petição Corrigida' não foi encontrada na resposta da API."
     except Exception as e:
+        logging.error(f"Erro ao extrair petição corrigida: {str(e)}")
         return f"❌ Erro ao extrair petição corrigida: {str(e)}"
 
 def fornecer_dica_contextual(texto):
@@ -527,7 +628,6 @@ def comparar_peticao_melhorada(caso, peticao_texto, correcao):
     if not ultima_correcao:
         return "⚠️ Nenhuma correção disponível no histórico.", "", "", "", ""
 
-    # Tenta extrair a petição corrigida da última correção
     peticao_corrigida = extrair_peticao_corrigida(ultima_correcao)
     if "⚠️" in peticao_corrigida or "❌" in peticao_corrigida:
         return peticao_corrigida, "", "", "", ""
@@ -613,6 +713,7 @@ def exportar_diferencas_html(caso, peticao_texto, correcao):
             HTML(string=html_content).write_pdf(temp_path)
             return temp_path
     except Exception as e:
+        logging.error(f"Erro ao gerar PDF: {str(e)}")
         return f"❌ Erro ao gerar PDF: {str(e)}"
     finally:
         if 'temp_path' in locals() and os.path.exists(temp_path):
@@ -737,7 +838,7 @@ def interface_gradio():
             inputs=[caso_input, peticao_texto, resultado_output],
             outputs=[download_file]
         ).then(
-            fn=lambda x: [gr.update(visible=True, value=x) if isinstance(x, str) and not x.startswith("❌") else gr.update(visible=True, value=None, label=f"Erro: {x}" if x else "Erro desconhecido"), gr.update(visible=True)],
+            fn=lambda x: [gr.update(visible=True, value=x if not x.startswith("❌") else None), gr.update(visible=True)],
             outputs=[download_file, download_btn]
         )
 
@@ -774,4 +875,4 @@ def interface_gradio():
     return demo
 
 app = interface_gradio()
-app.launch()
+app.launch()