Update app.py

app.py

@@ -50,6 +50,7 @@ DICAS_OAB = {
 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')
@@ -168,6 +169,7 @@ 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**", ""),
@@ -205,6 +207,7 @@ def validar_peticao_basica(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)
@@ -285,13 +288,14 @@ def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
           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, ensure_ascii=False)}) ou personalizada.
+          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:
@@ -424,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 = ""
@@ -451,7 +456,7 @@ def formatar_correção(texto):
         if match:
             nota = match.group(1)
             texto_formatado = texto_formatado.replace(f"### ⭐ Nota Final\n\n{nota}", f"### ⭐ Nota Final: {nota}/10\n\n")
-    return texto_formatado  # Corrigido de 'texto_formatada' para 'texto_formatado'
+    return texto_formatado
 
 def extrair_peticao_corrigida(correcao_completa):
     try:
@@ -485,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.", "", "", "", ""
@@ -548,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"""
@@ -579,7 +586,7 @@ def exportar_diferencas_html(caso, peticao_texto, correcao):
             HTML(string=html_content).write_pdf(temp_path)
             return temp_path
     except Exception as e:
-        return f"❌ Erro ao gerar PDF: {str(e)}"  # Retorna erro detalhado
+        return None
     finally:
         if 'temp_path' in locals() and os.path.exists(temp_path):
             os.unlink(temp_path)
@@ -703,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) 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 x else gr.update(visible=True, value=None, label="Erro ao gerar PDF"), gr.update(visible=True)],
             outputs=[download_file, download_btn]
         )