Update app.py

app.py

@@ -16,30 +16,24 @@ import numpy as np
 from datetime import datetime
 from ratelimiter import RateLimiter
 
-# Carrega variáveis de ambiente de um arquivo .env (opcional para dev local)
 load_dotenv()
 
-# Obtém a chave da API do ambiente (no Hugging Face, configure como Secret)
 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.")
 
-# URL da API Gemini
 GEMINI_API_URL = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent?key={GEMINI_API_KEY}"
 
-# Configurações da aplicação
 CONFIG = {
     "max_retry_attempts": 3,
-    "retry_delay": 10,  # Aumentado para 10 segundos para respeitar limites
+    "retry_delay": 10,
     "timeout": 60,
     "temperatura": 0.2,
     "max_tokens": 4096
 }
 
-# Limite de 60 requisições por minuto (ajuste conforme seu limite real)
 rate_limiter = RateLimiter(max_calls=60, period=60)
 
-# Banco de dicas contextuais
 DICAS_OAB = {
     "Endereçamento": "Em relações de consumo, prefira o foro do domicílio do autor (art. 101, I, CDC).",
     "Qualificação": "Mencione nome, estado civil (se informado), endereço (use 'Rua XXX' se não fornecido).",
@@ -48,7 +42,6 @@ DICAS_OAB = {
     "Custas": "Inclua 'Requer a juntada da guia de custas devidamente recolhida'."
 }
 
-# Histórico de correções e cache
 HISTORICO = []
 CORRECAO_CACHE = {}
 
@@ -223,6 +216,7 @@ def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
     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.
     **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.
@@ -267,17 +261,18 @@ def estruturar_prompt_correcao(caso, peticao, modo_treinamento=False):
     if modo_treinamento:
         base_prompt += f"""
         **Modo de Treinamento Interativo Ativado:**
-        - Para cada item com pontuação abaixo do máximo, forneça:
-          a) Uma explicação detalhada do erro.
-          b) Uma sugestão de redação corrigida baseada no caso.
-          c) Uma dica prática para evitar o erro no futuro (use este banco de dicas: {json.dumps(DICAS_OAB)}).
+        - 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 por que ele está errado conforme o CPC ou CDC.
+          b) Uma sugestão de redação corrigida baseada exclusivamente no caso fornecido.
+          c) Uma dica prática extraída do banco de dicas ({json.dumps(DICAS_OAB)}) ou uma dica personalizada se o banco não se aplicar.
+        - Não resuma; detalhe cada erro de forma pedagógica e clara.
         """
     return base_prompt
 
 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
@@ -346,7 +341,6 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
     if not peticao_conteudo.strip():
         return "⚠️ Forneça uma petição."
     
-    # Chave única para o cache
     cache_key = f"{caso[:50]}|{peticao_conteudo[:50]}|{modo_treinamento}"
     if cache_key in CORRECAO_CACHE:
         return CORRECAO_CACHE[cache_key]
@@ -382,12 +376,12 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
                 analise = correção_completa.strip()
             correção_formatada = formatar_correção(analise)
             HISTORICO.append({
-                "data": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+                "data": datetime.now().strftime("%Y-%-m-%d %H:%M:%S"),
                 "caso": caso,
                 "peticao": peticao_conteudo,
                 "correcao_completa": correção_completa
             })
-            CORRECAO_CACHE[cache_key] = correção_formatada  # Armazena no cache
+            CORRECAO_CACHE[cache_key] = correção_formatada
             return correção_formatada
         except requests.Timeout:
             if tentativa < CONFIG["max_retry_attempts"] - 1:
@@ -427,11 +421,10 @@ def extrair_peticao_corrigida(correcao_completa):
     try:
         if "### Petição Corrigida" in correcao_completa:
             inicio = correcao_completa.index("### Petição Corrigida") + len("### Petição Corrigida")
-            peticao_texto = ""
-            for line in correcao_completa[inicio:].split('\n'):
-                if line.strip() and not line.strip().startswith('#') and not line.strip().startswith('*') and not any(prefix in line for prefix in ["Análise", "Nota Final", "Pontos Positivos", "Pontos a Melhorar"]):
-                    peticao_texto += line + '\n'
-            return peticao_texto.strip()
+            peticao_texto = correcao_completa[inicio:].strip()
+            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."
     except Exception as e:
         return f"❌ Erro ao extrair petição corrigida: {str(e)}"
@@ -525,32 +518,32 @@ def comparar_peticao_melhorada(caso, peticao_texto, correcao):
         elif linha.startswith('? '):
             continue
 
+    total_removidas = len([linha for linha in resultado_diff if "linha-removida" in linha])
+    total_adicionadas = len([linha for linha in resultado_diff if "linha-adicionada" in linha])
+
     resumo_alteracoes = []
     for secao, alteracoes in secoes.items():
         if alteracoes:
-            resumo_alteracoes.append(f"### {secao}:")
+            resumo_alteracoes.append(f"### {secao}\n")
             for alt in alteracoes[:5]:
-                resumo_alteracoes.append(f"- {alt}")
+                resumo_alteracoes.append(f"  {alt}")
             if len(alteracoes) > 5:
-                resumo_alteracoes.append(f"- ... e mais {len(alteracoes) - 5} alterações")
-
-    total_removidas = len([linha for linha in resultado_diff if "linha-removida" in linha])
-    total_adicionadas = len([linha for linha in resultado_diff if "linha-adicionada" in linha])
+                resumo_alteracoes.append(f"  - ... e mais {len(alteracoes) - 5} alterações\n")
 
     resumo_estatistico = f"""
     ## 📊 Resumo das Alterações
-    - Total de linhas removidas: {total_removidas}
-    - Total de linhas adicionadas: {total_adicionadas}
-    - Diferença líquida: {total_adicionadas - total_removidas} linhas
+    - **Linhas Removidas:** {total_removidas}
+    - **Linhas Adicionadas:** {total_adicionadas}
+    - **Diferença Líquida:** {total_adicionadas - total_removidas} linhas
 
     ## 🔍 Alterações por Seção
-    {chr(10).join(resumo_alteracoes)}
+    {''.join(resumo_alteracoes)}
 
     ## 📝 Visualização Detalhada
     <style>
-    .linha-igual {{ color: #000000; }}
-    .linha-removida {{ color: #ffffff; background-color: #b30000; }}
-    .linha-adicionada {{ color: #ffffff; background-color: #006600; }}
+    .linha-igual {{ color: #000000; margin: 2px 0; }}
+    .linha-removida {{ color: #ffffff; background-color: #b30000; margin: 2px 0; padding: 2px; }}
+    .linha-adicionada {{ color: #ffffff; background-color: #006600; margin: 2px 0; padding: 2px; }}
     </style>
     {"".join(resultado_diff)}
     """
@@ -599,7 +592,7 @@ def interface_gradio():
                 with gr.Row():
                     gr.Markdown("### 🧩 Gerar Caso Fictício")
                     caso_btn = gr.Button("🔄 Gerar Novo Caso Fictício", variant="secondary")
-                caso_input = gr.Textbox(label="📋 Caso Fornecido pela Banca", lines=8, placeholder="Insira o texto COMPLETO do caso...")
+                caso_input = gr.Textbox(label="� - 📋 Caso Fornecido pela Banca", lines=8, placeholder="Insira o texto COMPLETO do caso...")
                 with gr.Tabs():
                     with gr.TabItem("✏️ Inserir Texto"):
                         peticao_texto = gr.Textbox(label="Texto da Petição do Aluno", lines=15, placeholder="Insira a petição completa...")