Update app.py

app.py

@@ -18,26 +18,9 @@ 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.")
@@ -49,36 +32,44 @@ CONFIG = {
     "retry_delay": 5,
     "timeout": 60,
     "temperatura": 0.2,
-    "max_tokens": 8192
+    "max_tokens": 4096
 }
 
 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)
-        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)
+        _, thresh = cv2.threshold(np_img, 150, 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,7 +105,6 @@ 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):
@@ -142,7 +132,6 @@ 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):
@@ -167,11 +156,9 @@ 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:
@@ -192,7 +179,6 @@ 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):
@@ -219,85 +205,19 @@ 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 = []
-    peticao_upper = peticao.upper()
-    peticao_lower = peticao.lower()
-
-    if "EXCELENTÍSSIMO" not in peticao_upper or "JUIZ" not in peticao_upper:
+    if "EXCELENTÍSSIMO" not in peticao.upper():
         problemas.append("⚠️ Endereçamento ausente ou incompleto (Art. 319, I, CPC).")
-    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:
+    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:
         problemas.append("⚠️ Valor da causa não especificado (Art. 319, V, CPC).")
-    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:
+    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_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).")
-
+    if "Nestes termos" not in peticao:
+        problemas.append("⚠️ Fechamento formal ausente.")
     return problemas
 
-# Função comentada (removida da execução, mantida para referência)
-"""
-def verificar_coerencia_caso_peticao(caso, peticao):
-    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"""
@@ -338,7 +258,7 @@ def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
     - **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.
+    - **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) con -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:**
@@ -451,33 +371,9 @@ 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:
@@ -485,7 +381,7 @@ 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 = sanitize_input(peticao_texto) or ""
+        peticao_conteudo = peticao_texto or ""
     if not peticao_conteudo.strip():
         return "⚠️ Forneça uma petição."
 
@@ -493,9 +389,8 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
     aviso_validacao = f"### Validação Inicial\n\n" + "\n".join(problemas) + "\n\n" if problemas else ""
 
     cache_key = f"{caso[:50]}|{peticao_conteudo[:50]}|{modo_treinamento}"
-    cached_result = carregar_cache(cache_key)
-    if cached_result:
-        return aviso_validacao + cached_result
+    if cache_key in CORRECAO_CACHE:
+        return aviso_validacao + CORRECAO_CACHE[cache_key]
 
     prompt = estruturar_prompt_correcao(caso, peticao_conteudo, modo_treinamento)
     payload = {
@@ -521,9 +416,6 @@ 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()
@@ -536,9 +428,9 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
                 "peticao": peticao_conteudo,
                 "correcao_completa": correcao_completa
             })
-            if len(HISTORICO) > 10:
+            if len(HISTORICO) > 10:  # Limite de histórico
                 HISTORICO.pop(0)
-            salvar_cache(cache_key, correcao_formatada)
+            CORRECAO_CACHE[cache_key] = correcao_formatada
             return aviso_validacao + correcao_formatada
         except requests.Timeout:
             if tentativa < CONFIG["max_retry_attempts"] - 1:
@@ -551,7 +443,6 @@ 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):
@@ -575,6 +466,7 @@ 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:
@@ -590,9 +482,9 @@ def extrair_peticao_corrigida(correcao_completa):
             fim = peticao_texto.find("###") if "###" in peticao_texto else len(peticao_texto)
             peticao_texto = peticao_texto[:fim].strip()
             return peticao_texto
-        return "⚠️ A seção '### Petição Corrigida' não foi encontrada na resposta da API."
+        else:
+            return "⚠️ A seção '### Petição Corrigida' não foi encontrada na resposta da API. Verifique a resposta e tente novamente."
     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):
@@ -712,7 +604,6 @@ 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):
@@ -837,7 +728,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 not x.startswith("❌") else None), gr.update(visible=True)],
+            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)],
             outputs=[download_file, download_btn]
         )