Update app.py

app.py

@@ -32,7 +32,7 @@ CONFIG = {
     "retry_delay": 5,
     "timeout": 60,
     "temperatura": 0.2,
-    "max_tokens": 4096
+    "max_tokens": 8000  # Aumentado para 8000 para garantir que a resposta completa caiba
 }
 
 rate_limiter = RateLimiter(max_calls=60, period=60)
@@ -284,7 +284,7 @@ def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
     **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. Siga rigorosamente:
+    Certifique-se de incluir esta seção exatamente como '### 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. 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).
@@ -416,6 +416,10 @@ 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"]
+            # Debug: Inspeciona a resposta se a seção esperada não estiver presente
+            if "### Petição Corrigida" not in correcao_completa:
+                print("Debug: Resposta da API não contém '### Petição Corrigida'. Resposta completa (primeiros 1000 caracteres):")
+                print(correcao_completa[:1000])
             if "### Petição Corrigida" in correcao_completa:
                 analise_fim = correcao_completa.index("### Petição Corrigida")
                 analise = correcao_completa[:analise_fim].strip()
@@ -468,7 +472,7 @@ def formatar_correcao(texto):
     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)
+        match = re.search(nota_pattern, texto_formatada, re.DOTALL)
         if match:
             nota = match.group(1)
             texto_formatado = texto_formatado.replace(f"### ⭐ Nota Final\n\n{nota}", f"### ⭐ Nota Final: {nota}/10\n\n")
@@ -476,6 +480,7 @@ def formatar_correcao(texto):
 
 def extrair_peticao_corrigida(correcao_completa):
     try:
+        # Tenta encontrar a seção exata
         if "### Petição Corrigida" in correcao_completa:
             inicio = correcao_completa.index("### Petição Corrigida") + len("### Petição Corrigida")
             peticao_texto = correcao_completa[inicio:].strip()
@@ -483,6 +488,16 @@ def extrair_peticao_corrigida(correcao_completa):
             peticao_texto = peticao_texto[:fim].strip()
             return peticao_texto
         else:
+            # Fallback: tenta identificar o início da petição corrigida por padrões comuns
+            linhas = correcao_completa.split('\n')
+            inicio_peticao = -1
+            for i, linha in enumerate(linhas):
+                if "EXCELENTÍSSIMO" in linha.upper() and i > len(linhas) // 2:  # Assume que a petição corrigida vem após a análise
+                    inicio_peticao = i
+                    break
+            if inicio_peticao != -1:
+                peticao_texto = "\n".join(linhas[inicio_peticao:]).strip()
+                return peticao_texto
             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)}"