Update app.py

app.py

@@ -17,7 +17,12 @@ from ratelimiter import RateLimiter
 import difflib
 from html import escape
 import random
-import tempfile  # Adicionado para a solução
+import tempfile
+import logging
+
+# Configura o logging para depuração
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
 
 load_dotenv()
 
@@ -60,6 +65,7 @@ def preprocess_image_for_ocr(img):
         _, thresh = cv2.threshold(np_img, 150, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
         return Image.fromarray(thresh)
     except Exception as e:
+        logger.error(f"Erro no pré-processamento da imagem: {str(e)}")
         return f"❌ Erro no pré-processamento: {str(e)}"
 
 def extract_text_from_pdf(file_path_or_obj):
@@ -95,6 +101,7 @@ def extract_text_from_pdf(file_path_or_obj):
                             text += f"[Erro na página {i+1}: {str(e)}]\n"
         return text if text.strip() else "⚠️ Não foi possível extrair texto do PDF."
     except Exception as e:
+        logger.error(f"Erro ao processar o PDF: {str(e)}")
         return f"❌ Erro ao processar o PDF: {str(e)}"
 
 def extract_text_from_docx(file_path_or_obj):
@@ -122,6 +129,7 @@ def extract_text_from_docx(file_path_or_obj):
                     text += "\n"
         return text if text.strip() else "⚠️ Não foi possível extrair texto do DOCX."
     except Exception as e:
+        logger.error(f"Erro ao processar o DOCX: {str(e)}")
         return f"❌ Erro ao processar o DOCX: {str(e)}"
 
 def extract_text_from_txt(file_path_or_obj):
@@ -146,6 +154,7 @@ def extract_text_from_txt(file_path_or_obj):
                     continue
             return "❌ Não foi possível decodificar o arquivo TXT."
     except Exception as e:
+        logger.error(f"Erro ao processar o TXT: {str(e)}")
         return f"❌ Erro ao processar o TXT: {str(e)}"
 
 def process_file(file_obj):
@@ -169,6 +178,7 @@ def process_file(file_obj):
         else:
             return "❌ Erro: Objeto de arquivo inválido."
     except Exception as e:
+        logger.error(f"Erro ao processar o arquivo: {str(e)}")
         return f"❌ Erro ao processar o arquivo: {str(e)}"
 
 def estruturar_prompt_analise(caso, peticao, modo_treinamento=False):
@@ -300,10 +310,13 @@ def chamar_api(prompt):
                 continue
             return f"❌ Erro na requisição: {e}"
         except Exception as e:
+            logger.error(f"Erro inesperado na chamada da API: {str(e)}")
             return f"❌ Erro inesperado: {str(e)}"
 
 def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treinamento=False):
+    logger.info("Iniciando correção de petição...")
     if not caso.strip():
+        logger.warning("Caso não fornecido.")
         return "⚠️ Forneça o caso da banca."
 
     troll_triggers = [
@@ -317,6 +330,7 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
     if peticao_arquivo is not None:
         peticao_conteudo = process_file(peticao_arquivo)
         if peticao_conteudo.startswith("❌") or peticao_conteudo.startswith("⚠️"):
+            logger.error(f"Erro ao processar arquivo: {peticao_conteudo}")
             return peticao_conteudo
     else:
         peticao_conteudo = peticao_texto or ""
@@ -354,9 +368,11 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
                     "🎮 **Nível Bônus**: 'Desbloqueie o nível bônus: escreva a petição sem ajuda e ganhe experiência!'",
                     "🎧 **Música de Elevador**: 'Enquanto você escreve, eu toco uma música relaxante. Boa sorte!'"
                 ]
+                logger.info("Detectado pedido para escrever petição. Retornando resposta troll.")
                 return random.choice(respostas_troll)
 
     if not peticao_conteudo.strip():
+        logger.warning("Petição não fornecida.")
         return "⚠️ Forneça uma petição."
 
     problemas = validar_peticao_basica(peticao_conteudo)
@@ -364,17 +380,20 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
 
     cache_key = f"{caso[:50]}|{peticao_conteudo[:50]}|{modo_treinamento}"
     if cache_key in CORRECAO_CACHE:
+        logger.info("Correção encontrada no cache.")
         return aviso_validacao + CORRECAO_CACHE[cache_key]
 
     prompt_analise = estruturar_prompt_analise(caso, peticao_conteudo, modo_treinamento)
     analise = chamar_api(prompt_analise)
     if "❌" in analise or "⏱️" in analise:
+        logger.error(f"Erro na análise da petição: {analise}")
         return aviso_validacao + analise
     analise_formatada = formatar_correcao(analise)
 
     prompt_peticao = estruturar_prompt_peticao_corrigida(caso, peticao_conteudo)
     peticao_corrigida = chamar_api(prompt_peticao)
     if "❌" in peticao_corrigida or "⏱️" in peticao_corrigida:
+        logger.error(f"Erro ao gerar petição corrigida: {peticao_corrigida}")
         return aviso_validacao + analise_formatada + "\n\n" + peticao_corrigida
 
     peticao_corrigida = pos_corrigir_peticao(peticao_corrigida)
@@ -389,6 +408,7 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
     if len(HISTORICO) > 10:
         HISTORICO.pop(0)
     CORRECAO_CACHE[cache_key] = analise_formatada
+    logger.info("Correção de petição concluída com sucesso.")
     return aviso_validacao + analise_formatada
 
 def formatar_correcao(texto):
@@ -429,6 +449,7 @@ def extrair_peticao_corrigida(correcao_completa):
             return peticao_texto[:fim].strip()
         return "⚠️ Seção '### Petição Corrigida' não encontrada."
     except Exception as e:
+        logger.error(f"Erro ao extrair petição corrigida: {str(e)}")
         return f"❌ Erro ao extrair petição: {str(e)}"
 
 def fornecer_dica_contextual(texto):
@@ -518,27 +539,48 @@ Diferença Líquida: {total_adicionadas - total_removidas} linhas
     return peticao, peticao_corrigida, resumo_estatistico, criticas_texto, visualizacao_detalhada
 
 def salvar_progresso(caso, peticao):
+    logger.info("Iniciando salvamento de progresso...")
     try:
         # Verifica se os inputs estão vazios ou None
         if not caso or not peticao:
+            logger.warning("Inputs vazios detectados: caso ou peticao estão vazios.")
             return gr.File.update(visible=False), "⚠️ Caso ou petição estão vazios. Preencha ambos para salvar."
         
-        # Cria um arquivo temporário no diretório acessível
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".json", mode='w', encoding='utf-8') as temp_file:
+        logger.info("Criando arquivo temporário...")
+        # Especifica um diretório personalizado para o arquivo temporário
+        temp_dir = os.path.join(os.getcwd(), "temp_files")
+        os.makedirs(temp_dir, exist_ok=True)
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".json", dir=temp_dir, mode='w', encoding='utf-8') as temp_file:
             json.dump({"caso": caso, "peticao": peticao}, temp_file, ensure_ascii=False, indent=4)
             arquivo_saida = temp_file.name
         
+        # Verifica se o arquivo foi criado e é acessível
+        if not os.path.exists(arquivo_saida):
+            logger.error(f"Arquivo temporário não encontrado: {arquivo_saida}")
+            return gr.File.update(visible=False), f"❌ Erro: Arquivo temporário não foi criado."
+        
+        logger.info(f"Arquivo temporário criado com sucesso: {arquivo_saida}")
         # Retorna o arquivo temporário para download e uma mensagem de sucesso
         return gr.File.update(value=arquivo_saida, visible=True), "✅ Progresso salvo com sucesso!"
     except Exception as e:
-        return gr.File.update(visible=False), f"❌ Erro ao salvar o progresso: {str(e)}"
+        logger.error(f"Erro ao salvar progresso: {str(e)}")
+        # Fallback: Retorna o conteúdo como string se o arquivo não puder ser criado
+        try:
+            json_content = json.dumps({"caso": caso, "peticao": peticao}, ensure_ascii=False, indent=4)
+            return gr.File.update(visible=False), f"✅ Progresso salvo! Copie o conteúdo abaixo:\n\n```json\n{json_content}\n```"
+        except Exception as e2:
+            logger.error(f"Erro ao gerar JSON como string: {str(e2)}")
+            return gr.File.update(visible=False), f"❌ Erro ao salvar o progresso: {str(e)}"
 
 def carregar_progresso(arquivo_entrada):
+    logger.info("Iniciando carregamento de progresso...")
     try:
         with open(arquivo_entrada, 'r', encoding='utf-8') as f:
             dados = json.load(f)
+        logger.info("Progresso carregado com sucesso.")
         return dados.get("caso", ""), dados.get("peticao", ""), "Progresso carregado com sucesso."
     except Exception as e:
+        logger.error(f"Erro ao carregar o progresso: {str(e)}")
         return "", "", f"❌ Erro ao carregar o progresso: {str(e)}"
 
 def interface_gradio():
@@ -670,9 +712,17 @@ def interface_gradio():
                      visualizacao_detalhada_output]
         )
         salvar_btn.click(
+            fn=lambda: gr.update(value="⏳ Salvando progresso...", visible=True),
+            inputs=None,
+            outputs=[processing_message]
+        ).then(
             fn=lambda caso, peticao: salvar_progresso(caso, peticao),
             inputs=[caso_input, peticao_texto],
             outputs=[download_file, resultado_output]
+        ).then(
+            fn=lambda: gr.update(value="", visible=False),
+            inputs=None,
+            outputs=[processing_message]
         )
         carregar_btn.click(
             fn=lambda arquivo: carregar_progresso(arquivo.name),