Update app.py

app.py

@@ -339,8 +339,6 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
                     "😴 **Conselho**: 'Descanse, porque eu não vou fazer isso por você. Boa sorte na próxima tentativa!'",
                     "📚 **Lição**: 'Aula 1: IA não é seu estagiário. Abra o livro e comece a escrever.'",
                     "🌧️ **Previsão**: 'Dia nublado com 100% de chance de você ter que escrever a petição sozinho.'",
-                    "🎤 **Rap**: 'Yo, yo, escute aqui, petição não faço, sou IA, não seu caderno de rascunho, vai!'",
-                    "🐢 **Fábula**: 'A tartaruga disse: devagar se escreve a petição. Eu digo: por você, eu não!'",
                     "💡 **Ideia Brilhante**: 'Que tal estudar o caso e tentar de novo? Eu só corrijo, não escrevo!'"
                     "🎸 **Rockstar**: 'Você queria uma petição, mas eu só toco guitarra. Tente de novo!'",
                     "🌟 **Horóscopo**: 'As estrelas dizem: hoje não é dia de pedir ajuda para IA. Escreva você mesmo!'",
@@ -522,6 +520,22 @@ Diferença Líquida: {total_adicionadas - total_removidas} linhas
 """
     return peticao, peticao_corrigida, resumo_estatistico, criticas_texto, visualizacao_detalhada
 
+def salvar_progresso(caso, peticao, arquivo_saida):
+    try:
+        with open(arquivo_saida, 'w', encoding='utf-8') as f:
+            json.dump({"caso": caso, "peticao": peticao}, f, ensure_ascii=False, indent=4)
+        return f"Progresso salvo com sucesso em {arquivo_saida}."
+    except Exception as e:
+        return f"❌ Erro ao salvar o progresso: {str(e)}"
+
+def carregar_progresso(arquivo_entrada):
+    try:
+        with open(arquivo_entrada, 'r', encoding='utf-8') as f:
+            dados = json.load(f)
+        return dados.get("caso", ""), dados.get("peticao", ""), "Progresso carregado com sucesso."
+    except Exception as e:
+        return "", "", f"❌ Erro ao carregar o progresso: {str(e)}"
+
 def interface_gradio():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("# 📝 Corretor Avançado de Petições Iniciais (Critérios OAB)")
@@ -650,14 +664,14 @@ def interface_gradio():
                      visualizacao_detalhada_output]
         )
         salvar_btn.click(
-            fn=lambda: salvar_progresso(HISTORICO, "progresso.json"),
-            inputs=None,
+            fn=lambda caso, peticao: salvar_progresso(caso, peticao, "progresso.json"),
+            inputs=[caso_input, peticao_texto],
             outputs=resultado_output
         )
         carregar_btn.click(
             fn=lambda arquivo: carregar_progresso(arquivo.name),
             inputs=[arquivo_progresso],
-            outputs=[resultado_output, historico_output]
+            outputs=[caso_input, peticao_texto, resultado_output]
         )
 
         with gr.Accordion("📘 Critérios Detalhados", open=False):
@@ -726,4 +740,4 @@ def validar_peticao_basica(peticao):
     return problemas
 
 app = interface_gradio()
-app.launch()
+app.launch()