Update app.py

app.py

@@ -16,7 +16,7 @@ from datetime import datetime
 from ratelimiter import RateLimiter
 import difflib
 from html import escape
-import random  # Adicionado para escolher respostas troll aleatoriamente
+import random
 
 load_dotenv()
 
@@ -304,7 +304,7 @@ def chamar_api(prompt):
 def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treinamento=False):
     if not caso.strip():
         return "⚠️ Forneça o caso da banca."
-    
+
     # Verifica se há tentativa de pedir para a IA escrever a petição
     troll_triggers = [
         "escreva a petição", "faça a petição", "redija a petição",
@@ -320,7 +320,7 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
             return peticao_conteudo
     else:
         peticao_conteudo = peticao_texto or ""
-    
+
     # Verifica se o conteúdo contém um trigger de trollagem
     if peticao_conteudo.strip():
         for trigger in troll_triggers:
@@ -357,10 +357,10 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
                     "🎬 **Trailer de Filme**: 'Em breve nos cinemas: A Petição Impossível. Escrita e dirigida por você!'",
                     "🎮 **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!'"
-                
+
                 ]
                 return random.choice(respostas_troll)
-    
+
     if not peticao_conteudo.strip():
         return "⚠️ Forneça uma petição."
 
@@ -386,7 +386,7 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
 
     correcao_completa = analise + "\n\n" + peticao_corrigida
     HISTORICO.append({
-        "data": datetime.now().strftime("%Y-%m-%d %H:%:M:%S"),
+        "data": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
         "caso": caso,
         "peticao": peticao_conteudo,
         "correcao_completa": correcao_completa
@@ -584,6 +584,9 @@ def interface_gradio():
         with gr.Row():
             clear_btn = gr.Button("🗑️ Limpar Todos os Campos")
             comparar_btn = gr.Button("⚖️ Comparar com Correção")
+            salvar_btn = gr.Button("💾 Salvar Progresso")
+            carregar_btn = gr.Button("📂 Carregar Progresso")
+            arquivo_progresso = gr.File(label="Arquivo de Progresso", file_types=[".json"])
 
         resultado_output = gr.Markdown(label="📊 Correção Oficial e Nota", show_copy_button=True)
 
@@ -646,6 +649,16 @@ def interface_gradio():
             outputs=[peticao_aluno_output, peticao_corrigida_output, resumo_alteracoes_output, criticas_output,
                      visualizacao_detalhada_output]
         )
+        salvar_btn.click(
+            fn=lambda: salvar_progresso(HISTORICO, "progresso.json"),
+            inputs=None,
+            outputs=resultado_output
+        )
+        carregar_btn.click(
+            fn=lambda arquivo: carregar_progresso(arquivo.name),
+            inputs=[arquivo_progresso],
+            outputs=[resultado_output, historico_output]
+        )
 
         with gr.Accordion("📘 Critérios Detalhados", open=False):
             gr.Markdown("""
@@ -713,4 +726,4 @@ def validar_peticao_basica(peticao):
     return problemas
 
 app = interface_gradio()
-app.launch()
+app.launch()