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,
+    "timeout": 60,
     "temperatura": 0.2,
     "max_tokens": 12882
 }
 
-rate_limiter = RateLimiter(max_calls=100, period=60)
+rate_limiter = RateLimiter(max_calls=60, 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]'.",
@@ -300,40 +300,35 @@ def chamar_api(prompt):
         except Exception as e:
             return f"❌ Erro inesperado: {str(e)}"
 
-def corrigir_peticao_com_progresso(caso, peticao_texto=None, peticao_arquivo=None, modo_treinamento=False):
-    # Função wrapper para fornecer feedback de progresso
+def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treinamento=False):
     if not caso.strip():
-        return "⚠️ Forneça o caso da banca.", "⚠️ Erro: Caso não fornecido."
+        return "⚠️ Forneça o caso da banca."
     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, "⚠️ Erro no processamento do arquivo."
+            return peticao_conteudo
     else:
         peticao_conteudo = peticao_texto or ""
     if not peticao_conteudo.strip():
-        return "⚠️ Forneça uma petição.", "⚠️ Erro: Petição não fornecida."
+        return "⚠️ Forneça uma petição."
 
-    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], "✅ Resultado recuperado do cache."
+        return aviso_validacao + CORRECAO_CACHE[cache_key]
 
-    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, "⚠️ Erro na análise da API."
+        return aviso_validacao + analise
     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, "⚠️ Erro ao gerar petição corrigida."
+        return aviso_validacao + analise_formatada + "\n\n" + peticao_corrigida
 
     peticao_corrigida = pos_corrigir_peticao(peticao_corrigida)
 
@@ -347,7 +342,7 @@ def corrigir_peticao_com_progresso(caso, peticao_texto=None, peticao_arquivo=Non
     if len(HISTORICO) > 10:
         HISTORICO.pop(0)
     CORRECAO_CACHE[cache_key] = analise_formatada
-    return aviso_validacao + analise_formatada, "✅ Correção concluída!"
+    return aviso_validacao + analise_formatada
 
 def formatar_correcao(texto):
     linhas = texto.split('\n')
@@ -372,7 +367,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_formatada, re.DOTALL)
+        match = re.search(nota_pattern, texto_formatado, 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")
@@ -511,9 +506,6 @@ 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.")
-        
-        estado_processamento = gr.Textbox(label="Status do Processamento", value="Aguardando...", interactive=False)
-        
         with gr.Row():
             with gr.Column(scale=1):
                 gr.Markdown("""
@@ -597,24 +589,23 @@ def interface_gradio():
                 outputs=historico_output
             )
 
-        # Configuração do botão de correção com progresso
+        caso_btn.click(fn=gerar_caso_ficticio, inputs=[], outputs=[caso_input])
         btn_texto.click(
-            fn=corrigir_peticao_com_progresso,
-            inputs=[caso_input, peticao_texto, peticao_arquivo, modo_treinamento],
-            outputs=[resultado_output, estado_processamento]
+            fn=lambda caso, texto, treinamento: corrigir_peticao(caso, texto, None, treinamento),
+            inputs=[caso_input, peticao_texto, modo_treinamento],
+            outputs=[resultado_output]
         )
         btn_arquivo.click(
-            fn=corrigir_peticao_com_progresso,
-            inputs=[caso_input, peticao_texto, peticao_arquivo, modo_treinamento],
-            outputs=[resultado_output, estado_processamento]
+            fn=lambda caso, arquivo, treinamento: corrigir_peticao(caso, None, arquivo, treinamento),
+            inputs=[caso_input, peticao_arquivo, modo_treinamento],
+            outputs=[resultado_output]
         )
-
         clear_btn.click(
-            fn=lambda: [None, None, None, None, "", "", "", "", "", gr.update(visible=False), gr.update(visible=False), "Aguardando..."],
+            fn=lambda: [None, None, None, None, "", "", "", "", "", gr.update(visible=False), gr.update(visible=False)],
             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, estado_processamento]
+                     visualizacao_detalhada_output, download_btn, download_file]
         )
         comparar_btn.click(
             fn=comparar_peticao_melhorada,