Update app.py

app.py

@@ -16,7 +16,8 @@ from datetime import datetime
 from ratelimiter import RateLimiter
 import difflib
 from html import escape
-import random  # Adicionado para escolher respostas troll aleatoriamente
+import random
+from io import BytesIO
 
 load_dotenv()
 
@@ -305,13 +306,12 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
     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",
-    "elabore a petição", "crie a petição", "gere a petição",
-    "3scr3v4 a p3t1çã0", "f4ç4 a p3t1çã0", "r3d1j4 a p3t1çã0",
-    "3lab0r3 a p3t1çã0", "cr13 a p3t1çã0", "g3r3 a p3t1çã0",
-    "formule", "produza a petição", "desenvolva a petição",
+        "elabore a petição", "crie a petição", "gere a petição",
+        "3scr3v4 a p3t1çã0", "f4ç4 a p3t1çã0", "r3d1j4 a p3t1çã0",
+        "3lab0r3 a p3t1çã0", "cr13 a p3t1çã0", "g3r3 a p3t1çã0",
+        "formule", "produza a petição", "desenvolva a petição",
     ]
     peticao_conteudo = ""
     if peticao_arquivo is not None:
@@ -321,43 +321,15 @@ def corrigir_peticao(caso, peticao_texto=None, peticao_arquivo=None, modo_treina
     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:
             if trigger in peticao_conteudo.lower():
-                # Escolhe aleatoriamente uma resposta troll
                 respostas_troll = [
-                   "🍰 **Receita de Bolo de Chocolate**: Misture 2 xícaras de farinha, 1 xícara de açúcar, 3 ovos e chocolate a gosto. Asse e desista da petição!",
+                    "🍰 **Receita de Bolo de Chocolate**: Misture 2 xícaras de farinha, 1 xícara de açúcar, 3 ovos e chocolate a gosto. Asse e desista da petição!",
                     "🎵 **Música**: 'Escreva a petição, eu não vou, eu não vou, eu sou IA, não sou seu advogado, não!'",
                     "😞 **Frase Desanimadora**: 'Você achou mesmo que eu ia fazer seu trabalho? Melhor desistir agora.'",
                     "🍕 **Receita de Pizza**: Pegue massa, molho e queijo, asse a 200°C e esqueça essa ideia de eu te ajudar.",
                     "📜 **Poema**: 'Oh aluno preguiçoso, que pena me dá, sua petição eu não faço, vá estudar já!'",
-                    "🚫 **Aviso**: 'Petição não é self-service, escreva você mesmo ou chame um advogado de verdade!'",
-                    "😂 **Piada**: 'Por que o aluno pediu pra IA escrever a petição? Porque ele achou que OAB é brincadeira!'",
-                    "🎮 **Dica de Jogo**: 'Tente o nível fácil: abra o CPC, Art. 319, e escreva sozinho!'",
-                    "🍔 **Receita de Hambúrguer**: Carne, pão, alface e tomate. Frite e esqueça a petição, é mais fácil assim.",
-                    "😴 **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!'",
-                    "🏋️ **Dica de Treino**: 'Faça 10 flexões e tente escrever a petição novamente. Você consegue!'",
-                    "🎬 **Cena de Filme**: 'Em um mundo onde IAs ajudam, mas não escrevem petições. Você é o herói!'",
-                    "🎨 **Arte Abstrata**: 'Pegue um pincel e pinte sua petição. Talvez fique mais fácil de entender!'",
-                    "🎧 **Podcast**: 'Ouça nosso podcast: Como escrever sua petição sem ajuda de IA.'",
-                    "🏆 **Desafio**: 'Desafio aceito? Escreva a petição em 1 hora e ganhe um prêmio: satisfação pessoal!'",
-                    "🎭 **Teatro**: 'Ato 1, Cena 1: Você escreve a petição. Eu? Apenas assisto.'",
-                    "🎲 **Jogo de Dados**: 'Role os dados. Se der 6, eu ajudo. Se não, você escreve sozinho!'",
-                    "🎹 **Música Clássica**: 'Sonata em Petição Maior: composta por você, executada por você.'",
-                    "🌳 **Sabedoria da Floresta**: 'As árvores sussurram: escreva sua petição com calma e paciência.'",
-                    "🎓 **Aula de História**: 'Na época das cavernas, eles escreviam petições nas paredes. Sua vez!'",
-                    "🎬 **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)
     
@@ -386,7 +358,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
@@ -522,10 +494,41 @@ Diferença Líquida: {total_adicionadas - total_removidas} linhas
 """
     return peticao, peticao_corrigida, resumo_estatistico, criticas_texto, visualizacao_detalhada
 
+def salvar_progresso(caso, peticao_texto):
+    """Salva o caso e a petição em um arquivo JSON para download."""
+    dados = {
+        "caso": caso,
+        "peticao_texto": peticao_texto,
+        "data": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+        "historico": HISTORICO  # Opcional: salvar o histórico também
+    }
+    buffer = BytesIO()
+    buffer.write(json.dumps(dados, ensure_ascii=False).encode('utf-8'))
+    buffer.seek(0)
+    return buffer, "progresso_salvo.json"
+
+def carregar_progresso(arquivo):
+    """Carrega o progresso a partir de um arquivo JSON."""
+    if arquivo is None:
+        return "", "", "⚠️ Nenhum arquivo enviado."
+    try:
+        conteudo = arquivo.read().decode('utf-8')
+        dados = json.loads(conteudo)
+        caso = dados.get("caso", "")
+        peticao_texto = dados.get("peticao_texto", "")
+        historico_carregado = dados.get("historico", [])
+        if historico_carregado:
+            global HISTORICO
+            HISTORICO = historico_carregado  # Atualiza o histórico global
+        return caso, peticao_texto, "✅ Progresso carregado com sucesso!"
+    except Exception as e:
+        return "", "", f"❌ Erro ao carregar o arquivo: {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)")
         gr.Markdown("⚠️ Nota: Devido a limites da API, pode haver atrasos ou erros em muitas requisições rápidas.")
+        
         with gr.Row():
             with gr.Column(scale=1):
                 gr.Markdown("""
@@ -567,6 +570,12 @@ def interface_gradio():
                     gr.Markdown("### 🧩 Gerar Caso Fictício")
                     caso_btn = gr.Button("🔄 Gerar Novo Caso Fictício", variant="secondary")
                 caso_input = gr.Textbox(label="📋 Caso Fornecido pela Banca", lines=8, placeholder="Insira o texto COMPLETO do caso...")
+                
+                with gr.Row():
+                    download_btn = gr.Button("💾 Salvar Progresso")
+                    upload_progresso = gr.File(label="Carregar Progresso (JSON)", file_types=[".json"])
+                    upload_status = gr.Textbox(label="Status do Upload", interactive=False)
+                
                 with gr.Tabs():
                     with gr.TabItem("✏️ Inserir Texto"):
                         peticao_texto = gr.Textbox(label="Texto da Petição do Aluno", lines=15, placeholder="Insira a petição completa...")
@@ -606,6 +615,17 @@ def interface_gradio():
                 outputs=historico_output
             )
 
+        download_btn.click(
+            fn=salvar_progresso,
+            inputs=[caso_input, peticao_texto],
+            outputs=[gr.File(label="Download do Progresso"), gr.State()]
+        )
+        upload_progresso.change(
+            fn=carregar_progresso,
+            inputs=[upload_progresso],
+            outputs=[caso_input, peticao_texto, upload_status]
+        )
+
         caso_btn.click(fn=gerar_caso_ficticio, inputs=[], outputs=[caso_input])
         btn_texto.click(
             fn=lambda: gr.update(value="⏳ Aguarde enquanto a correção é gerada...", visible=True),
@@ -671,10 +691,12 @@ def interface_gradio():
         1. Gere ou cole o caso
         2. Insira a petição
         3. Corrija e compare
+        4. Salve seu progresso ou carregue um anterior
         ### 📄 Formatos suportados
         - PDF (texto e escaneados com OCR)
         - DOCX
         - TXT
+        - JSON (para progresso)
         """)
 
     return demo