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": 60,
+    "timeout": 30,  # Reduzido de 60 para 30 segundos
     "temperatura": 0.2,
     "max_tokens": 12882
 }
 
-rate_limiter = RateLimiter(max_calls=60, period=60)
+rate_limiter = RateLimiter(max_calls=100, period=60)  # Ajustado de 60 para 100 chamadas por minuto
 
 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çamento e qualificação do juízo (Art. 319, I, CPC; Art. 101, I, CDC se aplicável)
+       - Endereçamentos 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)
@@ -506,6 +506,10 @@ 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 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():
             with gr.Column(scale=1):
                 gr.Markdown("""
@@ -589,23 +593,60 @@ def interface_gradio():
                 outputs=historico_output
             )
 
-        caso_btn.click(fn=gerar_caso_ficticio, inputs=[], outputs=[caso_input])
+        # Configuração do botão de correção de texto com feedback
         btn_texto.click(
             fn=lambda caso, texto, treinamento: corrigir_peticao(caso, texto, None, treinamento),
             inputs=[caso_input, peticao_texto, modo_treinamento],
-            outputs=[resultado_output]
+            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 [];
+            }"""
         )
+
+        # 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]
+            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 [];
+            }"""
         )
+
         clear_btn.click(
-            fn=lambda: [None, None, None, None, "", "", "", "", "", gr.update(visible=False), gr.update(visible=False)],
+            fn=lambda: [None, None, None, None, "", "", "", "", "", gr.update(visible=False), gr.update(visible=False), "Aguardando..."],
             inputs=None,
             outputs=[caso_input, peticao_texto, peticao_arquivo, arquivo_preview, resultado_output, dica_contextual,
                      peticao_aluno_output, peticao_corrigida_output, resumo_alteracoes_output, criticas_output,
-                     visualizacao_detalhada_output, download_btn, download_file]
+                     visualizacao_detalhada_output, download_btn, download_file, estado_processamento]
         )
         comparar_btn.click(
             fn=comparar_peticao_melhorada,