Update app.py

app.py

@@ -18,7 +18,7 @@ import difflib
 from html import escape
 import random
 import logging
-import re # Importado para uso em funções como calcular_progresso e gerar_caso_ficticio
+import re
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
@@ -101,7 +101,6 @@ def extract_text_from_pdf(file_path_or_obj):
             else:
                 logger.warning(f"Página {i+1} sem texto extraível, tentando OCR.")
                 try:
-                    # Supondo que Tesseract esteja instalado via packages.txt no HF
                     img = page.to_image(resolution=300)
                     preprocessed_img = preprocess_image_for_ocr(img.original)
                     page_ocr_text = pytesseract.image_to_string(preprocessed_img, lang='por')
@@ -725,7 +724,7 @@ def chamar_api(prompt):
     return "❌ Falha ao chamar a API após todas as tentativas."
 
 def corrigir_peticao(caso, tipo_peticao, peticao_texto=None, peticao_arquivo=None, modo_treinamento=False):
-    logger.info(f"Iniciando correção para tipo: {tipo_peticao}")
+    logger.info(f"Iniciando correção para tipo: {tipo_peticao}, Modo Treinamento: {modo_treinamento}")
     start_time = time.time()
 
     if not caso or not caso.strip():
@@ -739,6 +738,7 @@ def corrigir_peticao(caso, tipo_peticao, peticao_texto=None, peticao_arquivo=Non
     peticao_conteudo = ""
     if peticao_arquivo is not None:
         logger.info("Processando arquivo de petição...")
+        # O objeto peticao_arquivo aqui é o próprio arquivo temporário do Gradio
         peticao_conteudo = process_file(peticao_arquivo)
         if peticao_conteudo.startswith("❌") or peticao_conteudo.startswith("⚠️"):
             logger.error(f"Erro ao processar arquivo: {peticao_conteudo}")
@@ -778,7 +778,6 @@ def corrigir_peticao(caso, tipo_peticao, peticao_texto=None, peticao_arquivo=Non
     if cache_key in CORRECAO_CACHE:
         logger.info("Correção encontrada no cache.")
         cached_data = CORRECAO_CACHE[cache_key]
-        # Recalcular diff_html se não foi cacheado
         diff_html_cache, _, _ = calcular_diff_interno(cached_data["peticao_original"], cached_data["peticao_corrigida"])
         return cached_data["analise_formatada"], cached_data["peticao_original"], cached_data["peticao_corrigida"], cached_data["resumo_diff"], cached_data["criticas_diff"], diff_html_cache
 
@@ -1212,6 +1211,7 @@ def interface_gradio():
                     choices=TIPOS_DE_PETICAO,
                     value="Petição Inicial"
                 )
+                # Definindo o componente checkbox
                 modo_treinamento = gr.Checkbox(label="🎓 Modo Treinamento (Feedback Detalhado)", value=False)
 
                 with gr.Accordion("⚙️ Configurações Avançadas", open=False):
@@ -1281,6 +1281,23 @@ def interface_gradio():
 
         clear_btn = gr.Button("🗑️ Limpar Tudo")
 
+        # --- FUNÇÕES WRAPPER ---
+        def corrigir_peticao_wrapper_texto(caso, tipo_peticao, peticao_texto, peticao_arquivo_state):
+            # Acessa o valor atual do componente checkbox
+            is_treinamento = modo_treinamento.value
+            # Chama a função original passando o valor booleano
+            return corrigir_peticao(caso, tipo_peticao, peticao_texto, peticao_arquivo_state, is_treinamento)
+
+        def corrigir_peticao_wrapper_arquivo(caso, tipo_peticao, peticao_texto_state, peticao_arquivo):
+            # Acessa o valor atual do componente checkbox
+            is_treinamento = modo_treinamento.value
+            # Chama a função original passando o valor booleano
+            return corrigir_peticao(caso, tipo_peticao, peticao_texto_state, peticao_arquivo, is_treinamento)
+        # --- FIM DAS FUNÇÕES WRAPPER ---
+
+
+        # --- LÓGICA DOS BOTÕES E EVENTOS (COM CORREÇÃO) ---
+
         caso_btn.click(
              fn=gerar_caso_ficticio,
              inputs=[tipo_peticao_input],
@@ -1304,13 +1321,16 @@ def interface_gradio():
              outputs=[arquivo_preview_output]
         )
 
+        # Botão Corrigir (Texto) - MODIFICADO PARA USAR WRAPPER
         btn_corrigir.click(
             fn=lambda: gr.update(value="⏳ Processando... Gerando análise e correção. Isso pode levar um minuto...", visible=True),
             inputs=None,
             outputs=[processing_message]
         ).then(
-            fn=corrigir_peticao,
-            inputs=[caso_input, tipo_peticao_input, peticao_texto_input, gr.State(None), modo_treinamento],
+            fn=corrigir_peticao_wrapper_texto, # Chama a função wrapper
+            # Inputs SEM o componente modo_treinamento
+            inputs=[caso_input, tipo_peticao_input, peticao_texto_input, gr.State(None)],
+            # Outputs permanecem os mesmos
             outputs=[resultado_analise_output, gr.Textbox(visible=False), resultado_peticao_corrigida_output,
                      resumo_alteracoes_output, criticas_output, visualizacao_detalhada_output]
         ).then(
@@ -1319,13 +1339,16 @@ def interface_gradio():
             outputs=[processing_message]
         )
 
+        # Botão Corrigir (Arquivo) - MODIFICADO PARA USAR WRAPPER
         btn_corrigir_arquivo.click(
             fn=lambda: gr.update(value="⏳ Processando arquivo... Gerando análise e correção. Isso pode levar um minuto...", visible=True),
             inputs=None,
             outputs=[processing_message]
         ).then(
-            fn=corrigir_peticao,
-            inputs=[caso_input, tipo_peticao_input, gr.State(None), peticao_arquivo_input, modo_treinamento],
+            fn=corrigir_peticao_wrapper_arquivo, # Chama a função wrapper
+             # Inputs SEM o componente modo_treinamento
+            inputs=[caso_input, tipo_peticao_input, gr.State(None), peticao_arquivo_input],
+             # Outputs permanecem os mesmos
             outputs=[resultado_analise_output, gr.Textbox(visible=False), resultado_peticao_corrigida_output,
                      resumo_alteracoes_output, criticas_output, visualizacao_detalhada_output]
         ).then(