Update app.py

app.py

@@ -37,17 +37,15 @@ CONFIG = {
     "max_retry_attempts": 3,
     "retry_delay": 5,
     "timeout": 180,
-    "temperatura": 0.3, # Default temp for general IA correction
+    "temperatura": 0.3,
     "max_tokens": 8192
 }
 
-# Lower temperature for more deterministic OAB-based generation/correction
 CONFIG_OAB = {
     "temperatura": 0.2,
     "max_tokens": 8192
 }
 
-
 rate_limiter = RateLimiter(max_calls=30, period=60)
 
 TIPOS_DE_PETICAO = [
@@ -89,7 +87,6 @@ DICAS_OAB_MELHORADO = {
 HISTORICO = []
 CORRECAO_CACHE = {}
 
-# --- Funções de Extração de Texto ---
 def preprocess_image_for_ocr(img):
     try:
         gray = img.convert('L')
@@ -111,7 +108,7 @@ def extract_text_from_pdf(file_path_or_obj):
             page_text = page.extract_text()
             if page_text:
                 page_full_text += page_text + "\n"
-                text += page_text + "\n" # Keep track of non-OCR text separately
+                text += page_text + "\n"
             else:
                 logger.warning(f"Página {i+1} sem texto extraível, tentando OCR.")
                 try:
@@ -120,7 +117,7 @@ def extract_text_from_pdf(file_path_or_obj):
                     page_ocr_text = pytesseract.image_to_string(preprocessed_img, lang='por')
                     if page_ocr_text.strip():
                         page_full_text += page_ocr_text + "\n"
-                        ocr_text += page_ocr_text + "\n" # Keep track of OCR text separately
+                        ocr_text += page_ocr_text + "\n"
                     else:
                         logger.warning(f"OCR falhou na página {i+1}.")
                         page_full_text += f"[OCR falhou na página {i+1}]\n"
@@ -162,7 +159,6 @@ def extract_text_from_pdf(file_path_or_obj):
             logger.error(f"Erro no fallback com PyPDF2: {str(fallback_e)}")
             return f"❌ Erro crítico ao processar o PDF: {str(e)}"
 
-
 def extract_text_from_docx(file_path_or_obj):
     try:
         doc = docx.Document(file_path_or_obj)
@@ -235,8 +231,6 @@ def process_file(file_obj):
         logger.error(f"Erro geral ao processar o arquivo '{getattr(file_obj, 'name', 'N/A')}': {str(e)}", exc_info=True)
         return f"❌ Erro inesperado ao processar o arquivo: {str(e)}"
 
-
-# --- Funções da API Gemini ---
 def chamar_api(prompt, use_oab_config=False):
     config_to_use = CONFIG_OAB if use_oab_config else CONFIG
     payload = {
@@ -311,12 +305,6 @@ def chamar_api(prompt, use_oab_config=False):
 
     return "❌ Falha API após todas tentativas."
 
-
-# --- Funções de Correção Geral (IA) ---
-# (Inclui as funções restauradas: get_instrucoes_gerais_avaliacao, estruturar_prompt_analise,
-#  estruturar_prompt_peticao_corrigida, pos_corrigir_peticao, corrigir_peticao_geral_ia,
-#  formatar_correcao, avaliar_criterios_peticao (renomeada para avaliacao_criterios_ia), etc.)
-
 def get_instrucoes_gerais_avaliacao(modo_treinamento=False):
     base = """
     **Estrutura da Avaliação (Geral - IA):**
@@ -394,7 +382,7 @@ def estruturar_prompt_analise_geral(caso, peticao, tipo_peticao, modo_treinament
        - **Pedidos:** Acolhimento preliminares; IMPROCEDÊNCIA TOTAL; Condenação custas/honorários.
        - Indicação de provas. Fechamento formal.
          """
-    # Add other petition types...
+    # Add other types...
 
     prompt_especifico += "\n\n**Instruções Finais:** Analise a peça do aluno com base no caso e nos critérios gerais e específicos. Use seu conhecimento jurídico para simular um examinador da OAB. Forneça feedback construtivo e detalhado. Gere a resposta **exclusivamente** no formato solicitado (Análise Geral, Nota, Pontos Positivos, etc.)."
     return prompt_especifico
@@ -451,32 +439,32 @@ def pos_corrigir_peticao(peticao_corrigida, tipo_peticao):
     if tipo_peticao in ["Petição Inicial", "Ação de Alimentos Gravídicos"]:
         if "juntada da guia de custas" not in peticao_corrigida.lower() and "gratuidade de justiça" not in peticao_corrigida.lower():
              partes = peticao_corrigida.split("Nestes termos pede deferimento")
-             prefixo = partes[0] if len(partes) > 1 else peticao_corrigida
-             sufixo = "Nestes termos pede deferimento" + partes[1] if len(partes) > 1 else ""
-             peticao_corrigida = prefixo.strip() + "\nRequer, ainda, a juntada da guia de custas devidamente recolhida.\n" + sufixo
+             prefixo = partes[0].strip() if len(partes) > 1 else peticao_corrigida.strip()
+             sufixo = "\nNestes termos pede deferimento" + partes[1] if len(partes) > 1 else ""
+             peticao_corrigida = prefixo + "\nRequer, ainda, a juntada da guia de custas devidamente recolhida." + sufixo
 
         if "audiência de conciliação" not in peticao_corrigida.lower() and "audiência de mediação" not in peticao_corrigida.lower():
              partes = peticao_corrigida.split("Nestes termos pede deferimento")
-             prefixo = partes[0] if len(partes) > 1 else peticao_corrigida
-             sufixo = "Nestes termos pede deferimento" + partes[1] if len(partes) > 1 else ""
-             peticao_corrigida = prefixo.strip() + "\nOpta o Autor pela realização da audiência de conciliação ou mediação (Art. 319, VII, CPC).\n" + sufixo
+             prefixo = partes[0].strip() if len(partes) > 1 else peticao_corrigida.strip()
+             sufixo = "\nNestes termos pede deferimento" + partes[1] if len(partes) > 1 else ""
+             peticao_corrigida = prefixo + "\nOpta o Autor pela realização da audiência de conciliação ou mediação (Art. 319, VII, CPC)." + sufixo
 
     linhas = [l.strip() for l in peticao_corrigida.strip().split('\n') if l.strip()]
     fechamento_padrao = ["\nLocal, data.", "ADVOGADO(A)", "OAB/..."]
-    # Improved removal of existing closing lines
-    while linhas and any(term.lower() in linhas[-1].lower() for term in ["advogado", "oab", "local", "data", "pede deferimento"]):
-        # Check if it's the standard closing part or just content
-        is_standard_closing = any(std_close in linhas[-1] for std_close in fechamento_padrao)
-        if not is_standard_closing and "pede deferimento" in linhas[-1].lower() and len(linhas) > 3: # Avoid removing the actual request line
+
+    # Remove potential multiple closing lines before adding the standard one
+    while linhas and any(term.lower() in linhas[-1].lower() for term in ["advogado", "oab", "local", "data", "pede deferimento", "termos em que"]):
+        # More robust check to avoid removing essential content like the final request
+        last_line_lower = linhas[-1].lower()
+        is_closing_line = any(std_close.strip().lower() in last_line_lower for std_close in fechamento_padrao) or "pede deferimento" in last_line_lower or "termos em que" in last_line_lower
+        if not is_closing_line and len(linhas[-1]) > 30: # Heuristic: unlikely to be just closing
              break
         linhas.pop()
 
-    # Add standard closing cleanly
-    peticao_corrigida = "\n".join(linhas).strip() + "\n" + "\n".join(fechamento_padrao)
+    peticao_corrigida = "\n".join(linhas).strip() + "".join(fechamento_padrao) # Join directly
 
     return peticao_corrigida.strip()
 
-
 def formatar_correcao(texto):
     texto = texto.replace("### Análise Geral", "## 📝 Análise Geral")
     texto = texto.replace("### Nota Final", "## ⭐ Nota Final")
@@ -512,15 +500,6 @@ def formatar_correcao(texto):
         texto_final = re.sub(nota_pattern, f"\\1**{nota}/10**", texto_final, count=1, flags=re.IGNORECASE)
     return texto_final
 
-def avaliar_criterios_ia(caso, peticao_original, tipo_peticao):
-    # This function remains from the previous step, evaluating based on IA criteria
-    # Ensure it's called by corrigir_peticao_geral_ia if needed
-    logger.info(f"Iniciando avaliação por critérios (IA) para: {tipo_peticao}")
-    # ... (rest of the function logic as provided in the previous answer) ...
-    # It needs: obter_criterios_avaliacao, estruturar_prompt_avaliacao_criterios, chamar_api
-    # parse the JSON response and format to Markdown.
-    return "Avaliação por Critérios (IA) - Implementação Pendente ou Removida" # Placeholder
-
 def corrigir_peticao_geral_ia(caso, tipo_peticao, peticao_texto=None, peticao_arquivo=None, modo_treinamento=False):
     logger.info(f"Iniciando correção GERAL (IA) para tipo: {tipo_peticao}")
     start_time = time.time()
@@ -537,7 +516,7 @@ def corrigir_peticao_geral_ia(caso, tipo_peticao, peticao_texto=None, peticao_ar
          peticao_conteudo = peticao_texto
     else: return "⚠️ Forneça a petição.", "", "", "", "", "", ""
 
-    troll_triggers = [ "escreva a petição", "faça a petição"] # Shortened
+    troll_triggers = [ "escreva a petição", "faça a petição"]
     if any(trigger in peticao_conteudo.lower() for trigger in troll_triggers):
         return random.choice(["Nope! 😉", "Faça você e eu corrijo!"]), "", "", "", "", "", ""
 
@@ -565,8 +544,7 @@ def corrigir_peticao_geral_ia(caso, tipo_peticao, peticao_texto=None, peticao_ar
     else:
         peticao_corrigida_final = pos_corrigir_peticao(peticao_corrigida_bruta, tipo_peticao)
 
-    # avaliacao_criterios_md = avaliar_criterios_ia(caso, peticao_conteudo, tipo_peticao) # Keep this call commented/removed if not needed now
-    avaliacao_criterios_md = "" # Set to empty for now
+    avaliacao_criterios_md = ""
 
     visualizacao_detalhada_html, resumo_diff, criticas_diff = calcular_diff_interno(peticao_conteudo, peticao_corrigida_final)
 
@@ -583,11 +561,7 @@ def corrigir_peticao_geral_ia(caso, tipo_peticao, peticao_texto=None, peticao_ar
     return ( correcao_completa_formatada, peticao_conteudo, peticao_corrigida_final,
              resumo_diff, criticas_diff, avaliacao_criterios_md, visualizacao_detalhada_html )
 
-
-
 # --- Funções de Correção Padrão OAB ---
-# (Inclui as funções: parse_oab_pdf, estruturar_prompt_geracao_modelo_oab,
-#  gerar_peticao_modelo_oab, estruturar_prompt_correcao_pela_oab, corrigir_pela_oab)
 def parse_oab_pdf(file_obj):
     if file_obj is None: return None, None, None, "⚠️ Nenhum PDF OAB."
     try:
@@ -602,13 +576,11 @@ def parse_oab_pdf(file_obj):
             line_strip = line.strip()
             line_lower = line_strip.lower()
 
-            # Section Start Markers (allow variations)
             if line_strip.startswith("Enunciado"): current_section = "enunciado"; continue
             if line_strip.startswith("Gabarito comentado"): current_section = "gabarito"; continue
             if line_strip.startswith("Distribuição dos Pontos"): current_section = "distribuicao"; continue
-            # Break Markers (end of relevant sections)
             if line_strip.startswith("PADRÃO DE RESPOSTA – QUESTÃO"): current_section = None; break
-            if line_strip.startswith("Padrão de Resposta"): continue # Skip sub-headers
+            if line_strip.startswith("Padrão de Resposta"): continue
 
             if current_section == "enunciado": enunciado += line + "\n"
             elif current_section == "gabarito": gabarito += line + "\n"
@@ -618,10 +590,9 @@ def parse_oab_pdf(file_obj):
         gabarito = gabarito.strip()
         distribuicao = distribuicao.strip()
 
-        # Basic check if sections were captured
         if not enunciado or not gabarito or not distribuicao:
              logger.warning(f"Parse OAB PDF falhou. E:{bool(enunciado)}, G:{bool(gabarito)}, D:{bool(distribuicao)}")
-             return None, None, None, f"⚠️ Falha separar seções. Verifique o texto completo:\n\n{full_text[:500]}..." # Show preview
+             return None, None, None, f"⚠️ Falha separar seções. Verifique o texto completo:\n\n{full_text[:500]}..."
 
         logger.info("PDF OAB parseado.")
         return enunciado, gabarito, distribuicao, "✅ PDF OAB carregado e seções extraídas."
@@ -629,7 +600,6 @@ def parse_oab_pdf(file_obj):
         logger.error(f"Erro parsear PDF OAB: {str(e)}", exc_info=True)
         return None, None, None, f"❌ Erro processar PDF OAB: {str(e)}"
 
-
 def estruturar_prompt_geracao_modelo_oab(enunciado, gabarito, distribuicao):
     prompt = f"""
     **Tarefa:** Elabore a peça processual completa e adequada para o **Enunciado**, seguindo **estritamente** as diretrizes do **Gabarito Comentado** e os critérios da **Distribuição dos Pontos**. Crie uma peça modelo perfeita para este caso OAB.
@@ -666,7 +636,7 @@ def gerar_peticao_modelo_oab(enunciado, gabarito, distribuicao):
 
     logger.info("Gerando peça modelo OAB...")
     prompt = estruturar_prompt_geracao_modelo_oab(enunciado, gabarito, distribuicao)
-    peticao_modelo = chamar_api(prompt, use_oab_config=True) # Use OAB config
+    peticao_modelo = chamar_api(prompt, use_oab_config=True)
 
     if peticao_modelo.startswith("❌") or peticao_modelo.startswith("⏱️"):
         logger.error(f"Erro API gerar modelo OAB: {peticao_modelo}")
@@ -704,7 +674,7 @@ def estruturar_prompt_correcao_pela_oab(peticao_usuario, enunciado, gabarito, di
     1. **Avaliação Pontual:** Para **cada item** na **Distribuição**, verifique se foi abordado pelo Usuário conforme o Gabarito. Atribua a pontuação correspondente.
     2. **Justificativa:** Justifique brevemente a pontuação de cada item.
     3. **Feedback:** Aponte erros/omissões do Usuário vs Gabarito/Distribuição.
-    4. **Peça Corrigida (Opcional - Foco na Avaliação):** Se possível, sugira trechos corrigidos para alinhar ao padrão, mas priorize a avaliação pontual.
+    4. **Peça Corrigida (Opcional):** Se possível, sugira trechos corrigidos para alinhar ao padrão.
     5. **Output Markdown:**
 
     ```markdown
@@ -738,7 +708,7 @@ def corrigir_pela_oab(peticao_usuario, enunciado, gabarito, distribuicao):
 
     logger.info("Corrigindo pela OAB...")
     prompt = estruturar_prompt_correcao_pela_oab(peticao_usuario, enunciado, gabarito, distribuicao)
-    correcao_oab = chamar_api(prompt, use_oab_config=True) # Use OAB config
+    correcao_oab = chamar_api(prompt, use_oab_config=True)
 
     if correcao_oab.startswith("❌") or correcao_oab.startswith("⏱️"):
         logger.error(f"Erro API corrigir OAB: {correcao_oab}")
@@ -747,9 +717,7 @@ def corrigir_pela_oab(peticao_usuario, enunciado, gabarito, distribuicao):
     logger.info("Correção/Avaliação OAB concluída.")
     return correcao_oab
 
-
-# --- Funções Auxiliares (Diff, Histórico, Salvar/Carregar, Gerar Caso Fictício) ---
-# (Keep these largely as they were in the previous correct version)
+# --- Funções Auxiliares (Diff, Histórico, Salvar/Carregar, Gerar Caso Fictício, Dicas) ---
 def calcular_diff_interno(texto_original, texto_corrigido):
     if not texto_original or not texto_corrigido or texto_corrigido.startswith("⚠️") or texto_corrigido.startswith("❌"): return "","",""
     orig_lines = texto_original.splitlines(); corr_lines = texto_corrigido.splitlines()
@@ -774,7 +742,6 @@ def calcular_diff_interno(texto_original, texto_corrigido):
 
 def calcular_progresso():
     if not HISTORICO: return "📊 Progresso vazio."
-    # Simplified progress focusing on General IA for now
     notas_geral = []
     tipos_geral = set()
     count_geral = 0
@@ -851,15 +818,71 @@ def gerar_caso_ficticio(tipo_peticao="Petição Inicial"):
     logger.info("Caso gerado.")
     return caso
 
+def fornecer_dica_contextual(texto_peticao, tipo_peticao):
+    if not texto_peticao or not texto_peticao.strip():
+        return "Digite algo na petição para receber dicas..."
+    dicas_relevantes = []
+    texto_lower = texto_peticao.lower()
+    for chave, dica in DICAS_OAB_MELHORADO.items():
+         if chave in ["Clareza e Concisão", "Fundamentação Legal", "Coerência Argumentativa", "Revisão Ortográfica e Gramatical", "Formatação"]:
+              dicas_relevantes.append(f"💡 **{chave}:** {dica}")
+    keywords_por_tipo = {
+        "Petição Inicial": ["endereçamento", "qualificação", "fatos", "fundamentos", "pedidos", "valor da causa", "provas", "audiência", "citação", "custas"],
+        "Contestação": ["preliminar", "mérito", "impugnação", "fatos", "improcedência", "provas", "honorários"],
+        "Reconvenção": ["reconvinte", "reconvindo", "conexão", "pedidos reconvencionais"],
+        "Impugnação à Contestação": ["réplica", "preliminar", "fatos novos", "documentos", "ratifica"],
+        "Embargos de Declaração": ["omissão", "contradição", "obscuridade", "erro material", "decisão embargada"],
+        "Recurso de Apelação": ["sentença", "reforma", "anulação", "dialeticidade", "preparo", "tribunal", "razões"],
+        "Agravo de Instrumento": ["decisão interlocutória", "tribunal", "cabimento", "urgência", "efeito suspensivo", "peças obrigatórias"],
+        "Petição de Tutela Provisória": ["urgência", "evidência", "liminar", "periculum", "fumus boni", "probabilidade", "risco"],
+        "Ação de Alimentos Gravídicos": ["gravídicos", "nascituro", "indícios de paternidade", "binômio", "conversão", "lei 11.804"],
+    }
+    if tipo_peticao in keywords_por_tipo:
+         for keyword in keywords_por_tipo[tipo_peticao]:
+              if keyword in texto_lower:
+                   for chave, dica in DICAS_OAB_MELHORADO.items():
+                        chave_lower = chave.lower()
+                        tipo_simples = tipo_peticao.split()[0].lower() if tipo_peticao else ""
+                        if f"({tipo_simples})" in chave_lower or keyword in chave_lower:
+                            dica_formatada = f"💡 **{chave}:** {dica}"
+                            if dica_formatada not in dicas_relevantes:
+                                 dicas_relevantes.append(dica_formatada)
+    if not dicas_relevantes:
+        return "💡 Continue escrevendo ou selecione o tipo de peça para dicas mais específicas."
+    return "\n\n".join(dicas_relevantes[:5])
+
+def estruturar_prompt_geracao_caso(tipo_peticao="Petição Inicial"):
+    prompt = f"""
+    Crie um caso fictício **detalhado e realista** para um exercício prático de elaboração de uma peça do tipo **"{tipo_peticao}"**, simulando um cenário de Exame de Ordem (OAB).
+    **Instruções Gerais para o Caso:**
+    1.  **Complexidade Adequada:** (Ideal: 15-25 linhas).
+    2.  **Dados Essenciais:** Nomes, locais, datas, valores, relação jurídica, conflito.
+    3.  **Coerência:** Fatos lógicos para a peça indicada.
+    4.  **Foco no Tipo de Peça:** Cenário exige **{tipo_peticao}**.
+    **Instruções Específicas para Gerar Caso para "{tipo_peticao}":**
+    """
+    if tipo_peticao == "Petição Inicial" or tipo_peticao == "Ação de Alimentos Gravídicos":
+        prompt += "* Descreva violação de direitos, busca inicial da tutela. Elementos para danos. Dados para qualificação. Temas: consumo, contrato, cobrança, despejo, família (alimentos gravídicos - incluir indícios paternidade, necessidade mãe/nascituro)."
+    elif tipo_peticao == "Contestação":
+        prompt += "* Resumo da Inicial recebida. Versão dos fatos do RÉU. Elementos para Preliminares (Art. 337), Impugnação Específica, Defesa de Mérito (direta/indireta). Dados qualificação réu."
+    # ... (Add other types as before) ...
+    else: prompt += "* Crie um cenário jurídico cível com conflito claro."
+
+    prompt += """
+    **Finalize o caso com:**
+    "**Situação-Problema:** Na qualidade de advogado(a) de [Nome do Cliente], elabore a peça processual cabível ({tipo_peticao})."
+    **Importante:** Evite temas excessivamente específicos. Foco em Civil/Processo Civil comum.
+    """
+    return prompt
+
 
-# --- Interface Gradio (Reestruturada com Abas) ---
+# --- Interface Gradio ---
 def interface_gradio():
     with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="orange")) as demo:
         gr.Markdown("# ⚖️ Corretor Jurídico OAB Master ⚖️")
         gr.Markdown("Use a **Correção Geral (IA)** para feedback amplo ou a **Correção Baseada no Padrão OAB** para avaliação específica.")
 
         with gr.Tabs() as main_tabs:
-            # --- ABA: Correção Geral (IA) ---
             with gr.TabItem("🤖 Correção Geral (IA)", id=0):
                 with gr.Row():
                     with gr.Column(scale=1):
@@ -905,7 +928,6 @@ def interface_gradio():
                 processing_message_geral = gr.Markdown("", visible=False)
                 with gr.Tabs():
                      with gr.TabItem("📊 Análise IA"): resultado_analise_output_geral = gr.Markdown()
-                     # with gr.TabItem("💎 Critérios IA"): avaliacao_criterios_output_geral = gr.Markdown() # Optional tab
                      with gr.TabItem("✨ Correção IA"): resultado_peticao_corrigida_output_geral = gr.Textbox(lines=15, show_copy_button=True)
                      with gr.TabItem("⚖️ Diff"):
                           with gr.Row():
@@ -913,7 +935,6 @@ def interface_gradio():
                                with gr.Column(scale=3): visualizacao_detalhada_output_geral = gr.HTML()
                           criticas_output_geral = gr.Markdown()
 
-            # --- ABA: Correção Padrão OAB ---
             with gr.TabItem("📄 Correção Padrão OAB", id=1):
                 gr.Markdown("Forneça o material oficial (PDF ou texto) e sua peça.")
                 with gr.Row():
@@ -941,13 +962,11 @@ def interface_gradio():
 
         clear_btn = gr.Button("🗑️ Limpar Tudo")
 
-        # --- Lógica Eventos ---
         caso_btn_geral.click(gerar_caso_ficticio, [tipo_peticao_input_geral], [caso_input_geral])
         peticao_texto_input_geral.change(fornecer_dica_contextual, [peticao_texto_input_geral, tipo_peticao_input_geral], dica_contextual_output_geral)
         tipo_peticao_input_geral.change(fornecer_dica_contextual, [peticao_texto_input_geral, tipo_peticao_input_geral], dica_contextual_output_geral)
         peticao_arquivo_input_geral.change(process_file, [peticao_arquivo_input_geral], arquivo_preview_output_geral)
 
-        # Correção Geral IA (Texto)
         btn_corrigir_geral.click( lambda: gr.update(value="⏳ Processando IA...", visible=True), None, processing_message_geral).then(
             corrigir_peticao_geral_ia,
             [caso_input_geral, tipo_peticao_input_geral, peticao_texto_input_geral, gr.State(None), modo_treinamento_geral],
@@ -955,7 +974,6 @@ def interface_gradio():
              resumo_alteracoes_output_geral, criticas_output_geral, gr.Markdown(visible=False), visualizacao_detalhada_output_geral]
         ).then( lambda: gr.update(visible=False), None, processing_message_geral )
 
-        # Correção Geral IA (Arquivo)
         btn_corrigir_arquivo_geral.click( lambda: gr.update(value="⏳ Processando IA...", visible=True), None, processing_message_geral).then(
             corrigir_peticao_geral_ia,
             [caso_input_geral, tipo_peticao_input_geral, gr.State(None), peticao_arquivo_input_geral, modo_treinamento_geral],
@@ -963,7 +981,6 @@ def interface_gradio():
              resumo_alteracoes_output_geral, criticas_output_geral, gr.Markdown(visible=False), visualizacao_detalhada_output_geral]
         ).then( lambda: gr.update(visible=False), None, processing_message_geral )
 
-        # Salvar/Carregar Geral
         salvar_btn_geral.click( lambda: gr.update(value="⏳ Salvando...", visible=True), None, processing_message_geral).then(
              salvar_progresso, [caso_input_geral, peticao_texto_input_geral, tipo_peticao_input_geral],
              [gr.Textbox(visible=False), config_status_geral, progresso_download_file_geral]
@@ -974,7 +991,6 @@ def interface_gradio():
              [tipo_peticao_input_geral, caso_input_geral, peticao_texto_input_geral, config_status_geral]
          ).then( lambda: gr.update(visible=False), None, processing_message_geral )
 
-        # Padrão OAB
         oab_pdf_input.change(parse_oab_pdf, [oab_pdf_input], [oab_enunciado_input, oab_gabarito_input, oab_distribuicao_input, parse_status_output])
         btn_gerar_modelo_oab.click( lambda: gr.update(value="⏳ Gerando Modelo...", visible=True), None, processing_message_oab).then(
             gerar_peticao_modelo_oab, [oab_enunciado_input, oab_gabarito_input, oab_distribuicao_input], [resultado_modelo_oab_output]
@@ -983,10 +999,9 @@ def interface_gradio():
              corrigir_pela_oab, [peticao_usuario_oab_input, oab_enunciado_input, oab_gabarito_input, oab_distribuicao_input], [resultado_correcao_oab_output]
          ).then( lambda: gr.update(visible=False), None, processing_message_oab )
 
-        # Limpar Tudo
-        clear_btn.click( fn=lambda: [ # Reset values for all relevant components across tabs
-                "Petição Inicial", False, None, None, None, None, "", "", "", "", "", "", "", None, None, # Geral IA Tab
-                None, "", None, None, None, "", "", "", "" # OAB Tab
+        clear_btn.click( fn=lambda: [
+                "Petição Inicial", False, None, None, None, None, "", "", "", "", "", "", "", None, None,
+                None, "", None, None, None, "", "", "", ""
             ], inputs=None, outputs=[
                 tipo_peticao_input_geral, modo_treinamento_geral, caso_input_geral, peticao_texto_input_geral, peticao_arquivo_input_geral,
                 arquivo_preview_output_geral, processing_message_geral, resultado_analise_output_geral,
@@ -999,7 +1014,6 @@ def interface_gradio():
 
     return demo
 
-# --- Execução ---
 if __name__ == "__main__":
     app = interface_gradio()
     app.launch()