Update app.py

app.py

@@ -319,7 +319,7 @@ def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
 def estruturar_prompt_geracao_caso():
     return """
     Crie um caso fictício detalhado para uma petição inicial cível, similar ao exemplo:
-    "Ana, modelo profissional, residente em Manaus, viajou para São Paulo, para o casamento de sua filha. Para lavar, pintar seus cabelos e realizar um penteado para o casamento, Ana procurou os serviços de João Macedo, cabeleireiro e dono do salão de beleza 'Hair', sediado na cidade de São Paulo, que lhe cobrou R$500,00 (quinhentos reais) pela prestação do serviço. Após lavar os cabelos de Ana, João aplicou-lhe uma tintura da marca francesa ABC, importada pela empresa Brasil Connection Ltda, sediada na cidade de Curitiba (PR). Meia hora após a aplicação da tintura, Ana sofreu uma reação alérgica, que demandou atendimento médico-hospitalar, no valor de R$1.000,00 (um mil reais), bem como dois dias de absoluto repouso que impossibilitou sua presença no casamento de sua filha. Além disso, perdeu grande parte de seu cabelo, tendo permanecido com manchas em seu rosto por dois meses, perdendo um ensaio fotográfico, para o qual já havia sido contratada, pelo valor de R$50,000,00 (cinquenta mil reais). Posteriormente constatou-se que a tintura utilizada continha substâncias químicas extremamente perigosas à vida e à saúde das pessoas e a fabricante ABC já havia sido condenada pela justiça francesa a encerrar a fabricação e comercialização do produto. Indignada com os danos sofridos, Ana procura um advogado para pleitear o devido ressarcimento."
+    "Ana, modelo profissional, residente em Manaus, viajou para São Paulo, para o casamento de sua filha. Para lavar, pintar seus cabelos e realizar um penteado para o casamento, Ana procurou os serviços de João Macedo, cabeleireiro e dono do salão de beleza 'Hair', sediado na cidade de São Paulo, que lhe cobrou R$500,00 (quinhentos reais) pela prestação do serviço. Após lavar os cabelos de Ana, João aplicou-lhe uma tintura da marca francesa ABC, importada pela empresa Brasil Connection Ltda, sediada na cidade de Curitiba (PR). Meia hora após a aplicação da tintura, Ana sofreu uma reação alérgica, que demandou atendimento médico-hospitalar, no valor de R$1.000,00 (um mil reais), bem como dois dias de absoluto repouso que impossibilitou sua presença no casamento de sua filha. Além disso, perdeu grande parte de seu cabelo, tendo permanecido com manchas em seu rosto por dois meses, perdendo um ensaio fotográfico, para o qual já havia sido contratada, pelo valor de R$50.000,00 (cinquenta mil reais). Posteriormente constatou-se que a tintura utilizada continha substâncias químicas extremamente perigosas à vida e à saúde das pessoas e a fabricante ABC já havia sido condenada pela justiça francesa a encerrar a fabricação e comercialização do produto. Indignada com os danos sofridos, Ana procura um advogado para pleitear o devido ressarcimento."
     O caso deve:
     1. Conter uma situação clara de violação de direitos que justifique uma ação cível
     2. Incluir valores específicos de danos materiais e/ou morais
@@ -418,23 +418,23 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
                     time.sleep(CONFIG["retry_delay"] * (tentativa + 1))
                     continue
                 return aviso_validacao + "❌ A API retornou uma resposta vazia. Tente novamente."
-            correção_completa = response_data["candidates"][0]["content"]["parts"][0]["text"]
-            if "### Petição Corrigida" in correção_completa:
-                analise_fim = correção_completa.index("### Petição Corrigida")
-                analise = correção_completa[:analise_fim].strip()
+            correcao_completa = response_data["candidates"][0]["content"]["parts"][0]["text"]
+            if "### Petição Corrigida" in correcao_completa:
+                analise_fim = correcao_completa.index("### Petição Corrigida")
+                analise = correcao_completa[:analise_fim].strip()
             else:
-                analise = correção_completa.strip()
-            correção_formatada = formatar_correção(analise)
+                analise = correcao_completa.strip()
+            correcao_formatada = formatar_correcao(analise)
             HISTORICO.append({
                 "data": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
                 "caso": caso,
                 "peticao": peticao_conteudo,
-                "correcao_completa": correção_completa
+                "correcao_completa": correcao_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
+            CORRECAO_CACHE[cache_key] = correcao_formatada
+            return aviso_validacao + correcao_formatada
         except requests.Timeout:
             if tentativa < CONFIG["max_retry_attempts"] - 1:
                 time.sleep(CONFIG["retry_delay"] * (tentativa + 1))
@@ -448,7 +448,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)}"
 
-def formatar_correção(texto):
+def formatar_correcao(texto):
     linhas = texto.split('\n')
     texto_formatado = ""
     for linha in linhas:
@@ -475,17 +475,23 @@ 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:
+        # 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
-        return "⚠️ Petição corrigida não encontrada na resposta da API."
+        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."
     except Exception as e:
         return f"❌ Erro ao extrair petição corrigida: {str(e)}"
 
@@ -520,7 +526,11 @@ def comparar_peticao_melhorada(caso, peticao_texto, correcao):
     ultima_correcao = HISTORICO[-1]["correcao_completa"] if HISTORICO else ""
     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, "", "", "", ""
 
     orig_lines = peticao.split('\n')
     corr_lines = peticao_corrigida.split('\n')
@@ -603,7 +613,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 f"❌ Erro ao gerar PDF: {str(e)}"
     finally:
         if 'temp_path' in locals() and os.path.exists(temp_path):
             os.unlink(temp_path)