Update app.py

app.py

@@ -28,12 +28,12 @@ GEMINI_API_URL = f"https://generativelanguage.googleapis.com/v1beta/models/gemin
 CONFIG = {
     "max_retry_attempts": 3,
     "retry_delay": 5,
-    "timeout": 30,  # Reduzido de 60 para 30 segundos
+    "timeout": 30,
     "temperatura": 0.2,
     "max_tokens": 12882
 }
 
-rate_limiter = RateLimiter(max_calls=100, period=60)  # Ajustado de 60 para 100 chamadas por minuto
+rate_limiter = RateLimiter(max_calls=100, 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]'.",
@@ -183,7 +183,7 @@ def estruturar_prompt_analise(caso, peticao, modo_treinamento=False):
     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)
     5. Análise Específica dos Requisitos:
-       - Endereçamentos e qualificação do juízo (Art. 319, I, CPC; Art. 101, I, CDC se aplicável)
+       - 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)
@@ -300,35 +300,40 @@ def chamar_api(prompt):
         except Exception as e:
             return f"❌ Erro inesperado: {str(e)}"
 
-def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treinamento=False):
+def corrigir_peticao_com_progresso(caso, peticao_texto=None, peticao_arquivo=None, modo_treinamento=False):
+    # Função wrapper para fornecer feedback de progresso
     if not caso.strip():
-        return "⚠️ Forneça o caso da banca."
+        return "⚠️ Forneça o caso da banca.", "⚠️ Erro: Caso não fornecido."
     if peticao_arquivo is not None:
+        yield "", "⏳ Processando arquivo..."
         peticao_conteudo = process_file(peticao_arquivo)
         if peticao_conteudo.startswith("❌") or peticao_conteudo.startswith("⚠️"):
-            return peticao_conteudo
+            return peticao_conteudo, "⚠️ Erro no processamento do arquivo."
     else:
         peticao_conteudo = peticao_texto or ""
     if not peticao_conteudo.strip():
-        return "⚠️ Forneça uma petição."
+        return "⚠️ Forneça uma petição.", "⚠️ Erro: Petição não fornecida."
 
+    yield "", "⏳ Validando petição..."
     problemas = validar_peticao_basica(peticao_conteudo)
     aviso_validacao = f"### Validação Inicial\n\n" + "\n".join(problemas) + "\n\n" if problemas else ""
 
     cache_key = f"{caso[:50]}|{peticao_conteudo[:50]}|{modo_treinamento}"
     if cache_key in CORRECAO_CACHE:
-        return aviso_validacao + CORRECAO_CACHE[cache_key]
+        return aviso_validacao + CORRECAO_CACHE[cache_key], "✅ Resultado recuperado do cache."
 
+    yield "", "⏳ Analisando petição com a API..."
     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
+        return aviso_validacao + analise, "⚠️ Erro na análise da API."
     analise_formatada = formatar_correcao(analise)
 
+    yield "", "⏳ Gerando 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
+        return aviso_validacao + analise_formatada + "\n\n" + peticao_corrigida, "⚠️ Erro ao gerar petição corrigida."
 
     peticao_corrigida = pos_corrigir_peticao(peticao_corrigida)
 
@@ -342,7 +347,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
-    return aviso_validacao + analise_formatada
+    return aviso_validacao + analise_formatada, "✅ Correção concluída!"
 
 def formatar_correcao(texto):
     linhas = texto.split('\n')
@@ -367,7 +372,7 @@ def formatar_correcao(texto):
     if "Nota Final" in texto_formatado:
         import re
         nota_pattern = r"### Nota Final.*?(\d+[.,]\d+|\d+)"
-        match = re.search(nota_pattern, texto_formatado, re.DOTALL)
+        match = re.search(nota_pattern, texto_formatada, re.DOTALL)
         if match:
             nota = match.group(1)
             texto_formatado = texto_formatado.replace(f"### ⭐ Nota Final\n\n{nota}", f"### ⭐ Nota Final: {nota}/10\n\n")
@@ -507,7 +512,6 @@ def interface_gradio():
         gr.Markdown("# 📝 Corretor Avançado de Petições Iniciais (Critérios OAB)")
         gr.Markdown("⚠️ Nota: Devido a limites da API, pode haver atrasos ou erros em muitas requisições rápidas.")
         
-        # Campo de status adicionado logo no início
         estado_processamento = gr.Textbox(label="Status do Processamento", value="Aguardando...", interactive=False)
         
         with gr.Row():
@@ -593,52 +597,16 @@ def interface_gradio():
                 outputs=historico_output
             )
 
-        # Configuração do botão de correção de texto com feedback
+        # Configuração do botão de correção com progresso
         btn_texto.click(
-            fn=lambda caso, texto, treinamento: corrigir_peticao(caso, texto, None, treinamento),
-            inputs=[caso_input, peticao_texto, modo_treinamento],
-            outputs=[resultado_output, estado_processamento],
-            _js="""() => {
-                let btn = document.querySelector('button[id*="btn_texto"]');
-                btn.disabled = true;
-                btn.innerText = '⏳ Corrigindo...';
-                document.querySelector('#estado_processamento textarea').value = '⏳ Processando sua petição...';
-                return [];
-            }"""
-        ).then(
-            fn=lambda x: (x, "✅ Correção concluída!"),
-            inputs=[resultado_output],
-            outputs=[resultado_output, estado_processamento],
-            _js="""() => {
-                let btn = document.querySelector('button[id*="btn_texto"]');
-                btn.disabled = false;
-                btn.innerText = '🔍 Corrigir Petição (Texto)';
-                return [];
-            }"""
+            fn=corrigir_peticao_com_progresso,
+            inputs=[caso_input, peticao_texto, peticao_arquivo, modo_treinamento],
+            outputs=[resultado_output, estado_processamento]
         )
-
-        # Configuração do botão de correção de arquivo com feedback
         btn_arquivo.click(
-            fn=lambda caso, arquivo, treinamento: corrigir_peticao(caso, None, arquivo, treinamento),
-            inputs=[caso_input, peticao_arquivo, modo_treinamento],
-            outputs=[resultado_output, estado_processamento],
-            _js="""() => {
-                let btn = document.querySelector('button[id*="btn_arquivo"]');
-                btn.disabled = true;
-                btn.innerText = '⏳ Corrigindo...';
-                document.querySelector('#estado_processamento textarea').value = '⏳ Processando seu arquivo...';
-                return [];
-            }"""
-        ).then(
-            fn=lambda x: (x, "✅ Correção concluída!"),
-            inputs=[resultado_output],
-            outputs=[resultado_output, estado_processamento],
-            _js="""() => {
-                let btn = document.querySelector('button[id*="btn_arquivo"]');
-                btn.disabled = false;
-                btn.innerText = '🔍 Corrigir Petição (Arquivo)';
-                return [];
-            }"""
+            fn=corrigir_peticao_com_progresso,
+            inputs=[caso_input, peticao_texto, peticao_arquivo, modo_treinamento],
+            outputs=[resultado_output, estado_processamento]
         )
 
         clear_btn.click(