Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 import requests
 import os
 import json
-import tempfile
 import PyPDF2
 import docx
 import io
@@ -17,13 +16,12 @@ from datetime import datetime
 from ratelimiter import RateLimiter
 import difflib
 from html import escape
-from weasyprint import HTML  # Para gerar PDF com cores
 
 load_dotenv()
 
 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.")
+    raise ValueError("❌ A chave GEMINI_API_KEY não foi encontrada. Configure-a no arquivo .env ou como Secret.")
 
 GEMINI_API_URL = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent?key={GEMINI_API_KEY}"
 
@@ -32,22 +30,22 @@ CONFIG = {
     "retry_delay": 5,
     "timeout": 60,
     "temperatura": 0.2,
-    "max_tokens": 12882  # Mantido em 8000, mas dividido em duas chamadas
+    "max_tokens": 12882
 }
 
 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."
+    "Qualificação": "Mencione nome, estado civil (se informado), endereço (use 'Rua XXX' se não fornecido). Dica: Verifique se todos os dados estão corretos.",
+    "Fatos": "Descreva os fatos cronologicamente, sem inventar informações. Exemplo: 'Em [data], o autor celebrou contrato com o réu para [descrição]'.",
+    "Pedidos": "Seja específico nos pedidos (ex.: 'condenação ao pagamento de R$ X'). Inclua danos morais e materiais, se aplicável.",
+    "Custas": "Inclua 'Requer a juntada da guia de custas devidamente recolhida'. Nunca omita para evitar extinção do processo.",
+    "Fundamentação": "Baseie-se em artigos do CPC ou CDC e súmulas do STJ (ex.: Súmula 387 para danos morais).",
+    "Fechamento": "Finalize com 'Nestes termos pede deferimento', sem vírgula. Evite erros de pontuação.",
+    "Estratégia Jurídica": "Utilize precedentes e súmulas para fortalecer a argumentação.",
+    "Provas": "Especifique as provas a produzir. Inclua testemunhas, documentos e perícias, se necessário.",
+    "Contra-argumentação": "Antecipe defesas da parte contrária e prepare respostas."
 }
 
 HISTORICO = []
@@ -172,9 +170,8 @@ def process_file(file_obj):
         return f"❌ Erro ao processar o arquivo: {str(e)}"
 
 def estruturar_prompt_analise(caso, peticao, modo_treinamento=False):
-    """Estrutura o prompt apenas para a análise da petição."""
     base_prompt = f"""
-    ## Correção Avançada de Petição Inicial Cível (Critérios Realistas da OAB com Análise Especializada)
+    ## Correção Avançada de Petição Inicial Cível (Critérios Realistas da OAB)
     **Caso Fornecido:**
     {caso}
     **Petição do Aluno:**
@@ -184,126 +181,87 @@ def estruturar_prompt_analise(caso, peticao, modo_treinamento=False):
     1. Análise Geral (Visão geral sobre a petição - máximo 3 linhas, destacando adequação ao caso e consistência jurídica)
     2. Nota Final (0 a 10 com uma casa decimal, justificando brevemente a pontuação total)
     3. Pontos Positivos (Liste pelo menos 3 aspectos bem executados com justificativa legal detalhada, incluindo artigos, súmulas ou precedentes aplicáveis)
-    4. Pontos a Melhorar (Liste todos os problemas encontrados em ordem de gravidade, com explicação detalhada, base legal e impacto na nota; não omita erros menores)
+    4. Pontos a Melhorar (Liste todos os problemas encontrados em ordem de gravidade, com explicação detalhada, base legal e impacto na nota)
     5. Análise Específica dos Requisitos:
-       - Endereçamento e qualificação do juízo (Art. 319, I, CPC; Art. 101, I, CDC se aplicável): Verifique a competência territorial e material com precisão.
-       - Qualificação das partes (Art. 319, II, CPC): Avalie nomes, estado civil, endereço e consistência com o caso.
-       - Fatos e fundamentos jurídicos (Art. 319, III, CPC; Art. 186, CC): Compare os fatos narrados com o caso e analise a pertinência dos fundamentos legais.
-       - Pedidos com especificações (Art. 319, IV, CPC): Examine clareza, especificidade e adequação dos pedidos ao caso.
-       - Valor da causa (Art. 319, V, CPC): Confirme se o valor é realista e compatível com os danos alegados.
-       - Provas com que o autor pretende demonstrar a verdade dos fatos (Art. 319, VI, CPC): Avalie a suficiência e relevância das provas indicadas.
-       - Opção pela realização ou não de audiência de conciliação (Art. 319, VII, CPC): Verifique a menção explícita e sua coerência com o caso.
-       - Pedido de citação do réu: Analise se está claro e conforme os meios legais (ex.: Art. 246, CPC).
-       - Juntada da guia de custas: Confirme a presença literal do pedido e sua redação formal.
-       - Fechamento formal da petição: Certifique-se da estrutura exata "Local e data.\nADVOGADO.\nOAB." sem variações ou identificação indevida.
+       - Endereçamento e qualificação do juízo (Art. 319, I, CPC; Art. 101, I, CDC se aplicável)
+       - Qualificação das partes (Art. 319, II, CPC)
+       - Fatos e fundamentos jurídicos (Art. 319, III, CPC; Art. 186, CC)
+       - Pedidos com especificações (Art. 319, IV, CPC)
+       - Valor da causa (Art. 319, V, CPC)
+       - Provas com que o autor pretende demonstrar os fatos (Art. 319, VI, CPC)
+       - Opção pela realização ou não de audiência de conciliação (Art. 319, VII, CPC)
+       - Pedido de citação do réu
+       - Juntada da guia de custas
+       - Fechamento formal da petição
 
     **Critérios Fundamentais:**
-    1. Identificação do candidato (nome, CPF, RG ou número real na seção final) resulta em nota máxima de 3,0; forneça exemplo do trecho problemático.
-    2. Invenção de fatos não constantes no caso resulta em penalização grave (-5,0 pontos); identifique cada fato inventado e compare com o caso.
-    3. Falta de pedido de juntada de custas resulta em perda de 1,0 ponto; verifique a ausência ou redação inadequada.
-    4. Use marcadores para dados não fornecidos (ex.: "Rua XXX", "CPF XXX"); penalize invenção de dados específicos (-0,5 por ocorrência).
-    5. Não penalize por informações ausentes no caso original, mas alerte se a omissão compromete a petição.
-    6. A frase "Nestes termos, pede deferimento" com vírgula é incorreta; corrija para "Nestes termos pede deferimento" e penalize com -0,2 ponto, destacando o erro formal.
-    7. A petição deve finalizar com "Local e data.\nADVOGADO.\nOAB." sem identificação do aluno (nome, número ou estado na OAB); se houver identificação específica (ex.: "OAB/SP" ou "12345"), zere a nota (0,0) e explique o motivo.
-
-    **Instruções Especiais para Análise Avançada:**
-    - **Competência**: Avalie o foro indicado com base no Art. 46 do CPC (domicílio do réu) ou Art. 101, I do CDC (domicílio do autor em relações de consumo); penalize erros com -0,10 ponto e justifique com jurisprudência (ex.: STJ, REsp 1.234.567).
-    - **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.
-    - **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:**
-    Inclua uma tabela com:
-    | ITEM                                      | PONTUAÇÃO | JUSTIFICATIVA |
-    |-------------------------------------------|-----------|---------------|
-    **Endereçamento**                          |           |               |
-    1. Correto endereçamento ao juízo competente (0,10) | / 0,10 | [Explique erro ou acerto] |
-    2. Qualificação das partes conforme Art. 319, II, CPC (0,20) | / 0,20 | [Explique erro ou acerto] |
-    **Fundamentos**                            |           |               |
-    3. Exposição clara dos fatos (0,20)       | / 0,20 | [Explique erro ou acerto] |
-    4. Fundamentação jurídica adequada (0,25) | / 0,25 | [Explique erro ou acerto] |
-    5. Demonstração de ilícito e dano (0,25)  | / 0,25 | [Explique erro ou acerto] |
-    **Pedidos**                                |           |               |
-    6. Pedidos claros e específicos (0,30)    | / 0,30 | [Explique erro ou acerto] |
-    7. Valor da causa indicado (0,10)         | / 0,10 | [Explique erro ou acerto] |
-    8. Provas pretendidas especificadas (0,10)| / 0,10 | [Explique erro ou acerto] |
-    9. Opção sobre audiência de conciliação (0,10) | / 0,10 | [Explique erro ou acerto] |
-    10. Pedido de citação do réu (0,10)       | / 0,10 | [Explique erro ou acerto] |
-    11. Juntada da guia de custas (0,10)      | / 0,10 | [Explique erro ou acerto] |
-    **Fechamento**                             |           |               |
-    12. Fechamento formal adequado (0,10)     | / 0,10 | [Explique erro ou acerto] |
-    **Nota Final:** Calcule a soma dos pontos, aplicando penalizações cumulativas conforme os critérios fundamentais; justifique qualquer desconto adicional.
+    1. Identificação do candidato (nome, CPF, RG ou número real na seção final): nota máxima 3,0
+    2. Invenção de fatos não constantes no caso: -5,0 pontos
+    3. Falta de pedido de juntada de custas: -1,0 ponto
+    4. Use marcadores para dados não fornecidos (ex.: "Rua XXX"): penalize invenção de dados (-0,5 por ocorrência)
+    5. Não penalize por informações ausentes no caso original, mas alerte se a omissão compromete a petição
+    6. "Nestes termos, pede deferimento" com vírgula é incorreto: corrija e penalize com -0,2 ponto
+    7. Fechamento deve ser "Local e data.\nADVOGADO.\nOAB." sem identificação específica (ex.: "OAB/SP" ou "12345"): zere a nota (0,0) se houver
+
+    **Instruções Especiais:**
+    - Avalie o foro com base no Art. 46 do CPC ou Art. 101, I do CDC
+    - Compare fatos narrados com o caso
+    - Exija citação de pelo menos um artigo específico
+    - Verifique pedidos quantificados e fundamentados para danos
+    - Avalie formalidade e precisão da linguagem
+    - Inclua precedentes quando aplicável
     """
     if modo_treinamento:
         base_prompt += f"""
-        **Modo de Treinamento Interativo Avançado:**
-        - Para cada item com pontuação inferior ao máximo, forneça:
-          a) Explicação detalhada (máximo 2 frases curtas) com base legal (ex.: Art. 319, CPC) e impacto processual.
-          b) Sugestão de redação corrigida baseada no caso, com linguagem técnica.
-          c) Exemplo prático em outro contexto (máximo 1 frase).
-          d) Dica prática do banco ({json.dumps(DICAS_OAB_MELHORADO, ensure_ascii=False)}) ou personalizada, curta.
-          e) Estratégia jurídica avançada (máximo 1 frase).
-        - Verifique a coerência entre os fatos do caso e a petição, listando inconsistências com trechos específicos.
-        - Sugira estratégias adicionais baseadas no caso (ex.: cumulação de pedidos, Art. 327, CPC).
+        **Modo de Treinamento:**
+        - Para cada item com problemas, forneça:
+          a) Explicação detalhada com base legal
+          b) Sugestão de redação corrigida
+          c) Exemplo prático
+          d) Dica prática do banco ({json.dumps(DICAS_OAB_MELHORADO, ensure_ascii=False)})
+          e) Estratégia jurídica avançada
+        - Liste inconsistências com trechos específicos
+        - Sugira estratégias adicionais (ex.: cumulação de pedidos, Art. 327, CPC)
         """
     return base_prompt
 
 def estruturar_prompt_peticao_corrigida(caso, peticao):
-    """Estrutura o prompt para a petição corrigida com instruções mais enfáticas."""
     return f"""
     **Caso Fornecido:**
     {caso}
     **Petição do Aluno:**
     {peticao}
     ### Petição Corrigida
-    Gere a petição corrigida abaixo desta seção, intitulada exatamente como '### Petição Corrigida'. Você DEVE seguir TODOS os critérios abaixo rigorosamente, sem exceção, aplicando cada correção necessária com base no caso e na petição do aluno. Não ignore nenhum item, pois todos são obrigatórios:
-
-    1. **Endereçamento correto**: Corrija o foro com base no Art. 46 do CPC (domicílio do réu) ou Art. 101, I do CDC (domicílio do autor em relações de consumo, se aplicável). Exemplo: "Excelentíssimo Senhor Juiz de Direito da Vara Cível da Comarca de [Cidade correta]".
-    2. **Qualificação completa**: Inclua nome, estado civil, endereço (use "XXX" para dados ausentes), conforme Art. 319, II, CPC. Não invente dados específicos.
-    3. **Fatos claros e fiéis ao caso**: Narre os fatos do caso em ordem cronológica, sem adicionar informações não fornecidas, conforme Art. 319, III, CPC.
-    4. **Fundamentação jurídica robusta**: Cite pelo menos um artigo específico do CPC, CDC ou CC (ex.: Art. 186, CC para responsabilidade civil) e, se aplicável, súmulas (ex.: Súmula 387/STJ para danos morais).
-    5. **Pedidos específicos e quantificados**: Detalhe todos os pedidos com clareza e valores exatos (ex.: "R$ 50.000,00 por danos morais"), conforme Art. 319, IV, CPC.
-    6. **Valor da causa realista**: Inclua o valor total do benefício econômico pretendido, conforme Art. 319, V, CPC (ex.: soma de danos + valor do contrato, se aplicável).
-    7. **Provas detalhadas**: Especifique provas como "prova documental e testemunhal", conforme Art. 319, VI, CPC.
-    8. **Opção por audiência explícita**: Inclua "Requer a designação de audiência de conciliação" ou "Dispensa a audiência de conciliação", conforme Art. 319, VII, CPC.
-    9. **Pedido de citação formal**: Use "via postal, nos termos do Art. 246, I, CPC" ou outro meio legal aplicável.
-    10. **Juntada da guia de custas**: Adicione "Requer a juntada da guia de custas devidamente recolhida" em todas as petições corrigidas, sem exceção.
-    11. **Fechamento formal**: Finalize EXATAMENTE com "Local e data.\nADVOGADO.\nOAB." sem identificação específica (ex.: "OAB/SP" ou "12345" resultam em nota zero). A frase "Nestes termos, pede deferimento" com vírgula DEVE SER corrigida para "Nestes termos pede deferimento" (sem vírgula), conforme exigência formal da OAB.
-
-    **Instruções Obrigatórias**:
-    - Não omita nenhum dos 11 itens acima. Cada um é essencial e deve ser verificado e corrigido se necessário.
-    - Siga estritamente as regras formais da OAB, especialmente a correção da vírgula em "Nestes termos pede deferimento".
-    - Baseie-se exclusivamente no caso fornecido e na petição do aluno, sem inventar fatos ou dados.
-    - Gere a petição completa, incluindo todas as seções (endereçamento, qualificação, fatos, direito, pedidos, valor da causa, provas, fechamento).
+    Gere a petição corrigida abaixo desta seção, intitulada '### Petição Corrigida'. Siga TODOS os critérios abaixo:
+    1. Endereçamento correto: Art. 46 do CPC ou Art. 101, I do CDC
+    2. Qualificação completa: Nome, estado civil, endereço (use "XXX" para dados ausentes), Art. 319, II, CPC
+    3. Fatos fiéis ao caso: Ordem cronológica, sem invenção, Art. 319, III, CPC
+    4. Fundamentação jurídica: Cite pelo menos um artigo (ex.: Art. 186, CC) e súmulas, se aplicável
+    5. Pedidos específicos: Detalhe valores (ex.: "R$ 50.000,00 por danos morais"), Art. 319, IV, CPC
+    6. Valor da causa realista: Soma do benefício econômico, Art. 319, V, CPC
+    7. Provas detalhadas: "Prova documental e testemunhal", Art. 319, VI, CPC
+    8. Opção por audiência: "Requer a designação" ou "Dispensa", Art. 319, VII, CPC
+    9. Pedido de citação: "Via postal, Art. 246, I, CPC"
+    10. Juntada da guia: "Requer a juntada da guia de custas devidamente recolhida"
+    11. Fechamento: "Local e data.\nADVOGADO.\nOAB." sem identificação específica; corrija "Nestes termos, pede deferimento" para "Nestes termos pede deferimento"
     """
 
 def pos_corrigir_peticao(peticao_corrigida):
-    """Pós-processamento para garantir que todas as regras sejam aplicadas."""
-    # Garantir a correção da vírgula
     if "Nestes termos, pede deferimento" in peticao_corrigida:
         peticao_corrigida = peticao_corrigida.replace("Nestes termos, pede deferimento", "Nestes termos pede deferimento")
-
-    # Garantir o pedido de juntada da guia de custas
     if "juntada da guia de custas" not in peticao_corrigida.lower():
         peticao_corrigida = peticao_corrigida.replace("Nestes termos pede deferimento", "Requer a juntada da guia de custas devidamente recolhida.\nNestes termos pede deferimento")
-
-    # Garantir o fechamento formal correto
     if not peticao_corrigida.strip().endswith("Local e data.\nADVOGADO.\nOAB."):
         linhas = peticao_corrigida.split('\n')
-        while linhas and not linhas[-1].strip():  # Remove linhas em branco no final
+        while linhas and not linhas[-1].strip():
             linhas.pop()
         if len(linhas) >= 3:
             linhas[-3:] = ["Local e data.", "ADVOGADO.", "OAB."]
         else:
             linhas.extend(["Local e data.", "ADVOGADO.", "OAB."])
         peticao_corrigida = "\n".join(linhas)
-
-    # Garantir opção por audiência explícita
     if "audiência de conciliação" not in peticao_corrigida.lower():
-        peticao_corrigida = peticao_corrigida.replace("Nestes termos pede deferimento", "Requer a designação de audiência de conciliação, nos termos do Art. 319, VII, CPC.\nNestes termos pede deferimento")
-
+        peticao_corrigida = peticao_corrigida.replace("Nestes termos pede deferimento", "Requer a designação de audiência de conciliação, Art. 319, VII, CPC.\nNestes termos pede deferimento")
     return peticao_corrigida
 
 def chamar_api(prompt):
@@ -317,29 +275,28 @@ def chamar_api(prompt):
             with rate_limiter:
                 response = requests.post(GEMINI_API_URL, headers=headers, data=json.dumps(payload), timeout=CONFIG["timeout"])
             if response.status_code != 200:
-                response_data = response.json()
-                error_msg = response_data.get("error", {}).get("message", "Erro desconhecido")
+                error_msg = response.json().get("error", {}).get("message", "Erro desconhecido")
                 if tentativa < CONFIG["max_retry_attempts"] - 1:
                     time.sleep(CONFIG["retry_delay"] * (tentativa + 1))
                     continue
-                return f"❌ Erro da API (Tentativa {tentativa+1}/{CONFIG['max_retry_attempts']}): {error_msg}"
+                return f"❌ Erro da API: {error_msg}"
             response_data = response.json()
             if "candidates" not in response_data or not response_data["candidates"]:
                 if tentativa < CONFIG["max_retry_attempts"] - 1:
                     time.sleep(CONFIG["retry_delay"] * (tentativa + 1))
                     continue
-                return "❌ A API retornou uma resposta vazia. Tente novamente."
+                return "❌ Resposta vazia da API."
             return response_data["candidates"][0]["content"]["parts"][0]["text"].strip()
         except requests.Timeout:
             if tentativa < CONFIG["max_retry_attempts"] - 1:
                 time.sleep(CONFIG["retry_delay"] * (tentativa + 1))
                 continue
-            return "⏱️ A resposta demorou muito. Tente novamente mais tarde."
+            return "⏱️ Tempo esgotado."
         except requests.RequestException as e:
             if tentativa < CONFIG["max_retry_attempts"] - 1:
                 time.sleep(CONFIG["retry_delay"] * (tentativa + 1))
                 continue
-            return f"❌ Erro ao chamar a API: {e}"
+            return f"❌ Erro na requisição: {e}"
         except Exception as e:
             return f"❌ Erro inesperado: {str(e)}"
 
@@ -362,22 +319,18 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
     if cache_key in CORRECAO_CACHE:
         return aviso_validacao + CORRECAO_CACHE[cache_key]
 
-    # Primeira chamada: análise
     prompt_analise = estruturar_prompt_analise(caso, peticao_conteudo, modo_treinamento)
     analise = chamar_api(prompt_analise)
     if "❌" in analise or "⏱️" in analise:
         return aviso_validacao + analise
     analise_formatada = formatar_correcao(analise)
 
-    # Segunda chamada: petição corrigida
     prompt_peticao = estruturar_prompt_peticao_corrigida(caso, peticao_conteudo)
     peticao_corrigida = chamar_api(prompt_peticao)
     if "❌" in peticao_corrigida or "⏱️" in peticao_corrigida:
         return aviso_validacao + analise_formatada + "\n\n" + peticao_corrigida
 
-    # Pós-correção para garantir conformidade
     peticao_corrigida = pos_corrigir_peticao(peticao_corrigida)
-    print("Debug: Petição corrigida após pós-processamento:\n", peticao_corrigida[:1000])  # Debug para verificar a saída
 
     correcao_completa = analise + "\n\n" + peticao_corrigida
     HISTORICO.append({
@@ -405,7 +358,7 @@ def formatar_correcao(texto):
             texto_formatado += f"### ⚠️ {linha.strip()}\n\n"
         elif "- Invenção de fatos" in linha:
             texto_formatado += f"**🔴 {linha.strip()} (-5,0 pontos)**\n"
-        elif linha.strip().startswith('-') and len(linha) > 2 and any(termo in linha for termo in ["Endereçamento", "Qualificação", "Fatos", "Pedidos", "Valor", "Provas"]):
+        elif linha.strip().startswith('-') and any(termo in linha for termo in ["Endereçamento", "Qualificação", "Fatos", "Pedidos", "Valor", "Provas"]):
             texto_formatado += f"**{linha.strip()}**\n"
         elif "Explicação" in linha or "Sugestão" in linha or "Dica" in linha or "Exemplo" in linha:
             texto_formatado += f"*{linha.strip()}*\n"
@@ -426,21 +379,10 @@ def extrair_peticao_corrigida(correcao_completa):
             inicio = correcao_completa.index("### Petição Corrigida") + len("### Petição Corrigida")
             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
-        else:
-            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:
-                    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."
+            return peticao_texto[:fim].strip()
+        return "⚠️ Seção '### Petição Corrigida' não encontrada."
     except Exception as e:
-        return f"❌ Erro ao extrair petição corrigida: {str(e)}"
+        return f"❌ Erro ao extrair petição: {str(e)}"
 
 def fornecer_dica_contextual(texto):
     dicas = [f"💡 {chave}: {dica}" for chave, dica in DICAS_OAB_MELHORADO.items() if chave.lower() in texto.lower()]
@@ -530,6 +472,7 @@ Diferença Líquida: {total_adicionadas - total_removidas} linhas
 
 def exportar_diferencas_html(caso, peticao_texto, correcao):
     _, _, resumo_estatistico, criticas_texto, visualizacao_detalhada = comparar_peticao_melhorada(caso, peticao_texto, correcao)
+    
     html_content = f"""
 <html>
 <head>
@@ -553,21 +496,16 @@ def exportar_diferencas_html(caso, peticao_texto, correcao):
 </body>
 </html>
 """
-    try:
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
-            temp_path = tmp.name
-            HTML(string=html_content).write_pdf(temp_path)
-            return temp_path
-    except Exception as e:
-        return f"❌ Erro ao gerar PDF: {str(e)}"
-    finally:
-        if 'temp_path' in locals() and os.path.exists(temp_path):
-            os.unlink(temp_path)
+    
+    nome_arquivo = f"diferencas_peticao_{datetime.now().strftime('%Y%m%d_%H%M%S')}.html"
+    with open(nome_arquivo, 'w', encoding='utf-8') as f:
+        f.write(html_content)
+    return nome_arquivo
 
 def interface_gradio():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("# 📝 Corretor Avançado de Petições Iniciais (Critérios OAB)")
-        gr.Markdown("⚠️ Nota: Devido a limites de cota da API, pode haver atrasos ou erros se muitas requisições forem feitas rapidamente.")
+        gr.Markdown("⚠️ Nota: Devido a limites da API, pode haver atrasos ou erros em muitas requisições rápidas.")
         with gr.Row():
             with gr.Column(scale=1):
                 gr.Markdown("""
@@ -640,7 +578,7 @@ def interface_gradio():
                 visualizacao_detalhada_output = gr.HTML(label="📝 Alterações Detalhadas")
 
         with gr.Row():
-            download_btn = gr.Button("📥 Baixar Diferenças em PDF", visible=False)
+            download_btn = gr.Button("📥 Baixar Diferenças em HTML", visible=False)
             download_file = gr.File(label="Download", visible=False)
 
         with gr.Accordion("📜 Histórico de Correções", open=False):
@@ -683,7 +621,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), gr.update(visible=True)],
             outputs=[download_file, download_btn]
         )
 
@@ -706,7 +644,7 @@ def interface_gradio():
             """)
         gr.Markdown("""
         ### ℹ️ Modo de Treinamento Interativo
-        Ative para explicações detalhadas com exemplos práticos e sugestões.
+        Ative para explicações detalhadas com exemplos e sugestões.
         ### 📋 Como usar
         1. Gere ou cole o caso
         2. Insira a petição
@@ -728,19 +666,13 @@ def estruturar_prompt_geracao_caso():
     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."
     O caso deve:
-    1. Conter uma situação clara de violação de direitos que justifique uma ação cível
+    1. Conter uma violação clara de direitos que justifique uma ação cível
     2. Incluir valores específicos de danos materiais e/ou morais
-    3. Mencionar nomes completos, cidades e informações suficientes para identificar as partes
-    4. Estabelecer relação entre as partes (contratual, extracontratual, consumo, etc.)
+    3. Mencionar nomes completos, cidades e informações para identificar as partes
+    4. Estabelecer relação entre as partes (contratual, consumo, etc.)
     5. Ter complexidade adequada para uma petição inicial de nível OAB
-    6. Ter entre 15-25 linhas de texto
-    7. Focar em um dos seguintes temas (escolha APENAS UM aleatoriamente):
-       - Responsabilidade civil por acidente de consumo
-       - Inadimplemento contratual com danos
-       - Responsabilidade civil por dano ambiental
-       - Violação de direitos da personalidade
-       - Relação de consumo com vício de produto/serviço
-       - Reparação por danos patrimoniais
+    6. Ter entre 15-25 linhas
+    7. Focar em um tema aleatório: responsabilidade civil por acidente de consumo, inadimplemento contratual, dano ambiental, direitos da personalidade, vício de produto/serviço, ou danos patrimoniais
     Termine com: "Situação problema: Como advogado(a) de [NOME DO CLIENTE], promova a demanda cabível."
     """