Update app.py

app.py

@@ -17,7 +17,6 @@ from ratelimiter import RateLimiter
 import difflib
 from html import escape
 import random
-import base64
 import logging
 
 # Configura o logging para depuração
@@ -544,20 +543,21 @@ def salvar_progresso(caso, peticao):
         # Verifica se os inputs estão vazios ou None
         if not caso or not peticao:
             logger.warning("Inputs vazios detectados: caso ou peticao estão vazios.")
-            return "", "⚠️ Caso ou petição estão vazios. Preencha ambos para salvar.", ""
+            return "", "⚠️ Caso ou petição estão vazios. Preencha ambos para salvar.", None
 
         # Cria o conteúdo JSON como string
         json_content = json.dumps({"caso": caso, "peticao": peticao}, ensure_ascii=False, indent=4)
 
-        # Codifica o conteúdo em base64 para criar um link de download
-        encoded_content = base64.b64encode(json_content.encode('utf-8')).decode('utf-8')
-        download_link = f"[Clique aqui para baixar o progresso (progresso.json)](data:application/json;base64,{encoded_content})"
+        # Cria um arquivo em memória usando io.BytesIO
+        buffer = io.BytesIO()
+        buffer.write(json_content.encode('utf-8'))
+        buffer.seek(0)
 
-        logger.info("Progresso salvo com sucesso como string e link de download.")
-        return json_content, "✅ Progresso salvo! Copie o conteúdo abaixo ou use o link para baixar:", download_link
+        logger.info("Progresso salvo com sucesso como string e arquivo em memória.")
+        return json_content, "✅ Progresso salvo! Copie o conteúdo abaixo ou baixe o arquivo:", buffer
     except Exception as e:
         logger.error(f"Erro ao salvar progresso: {str(e)}")
-        return "", f"❌ Erro ao salvar o progresso: {str(e)}", ""
+        return "", f"❌ Erro ao salvar o progresso: {str(e)}", None
 
 def carregar_progresso(arquivo_entrada):
     logger.info("Iniciando carregamento de progresso...")
@@ -640,7 +640,7 @@ def interface_gradio():
 
         # Componentes para exibir o progresso salvo
         progresso_json_output = gr.Textbox(label="Progresso Salvo (JSON)", lines=5, interactive=False, visible=False)
-        progresso_download_link = gr.Markdown(label="Link para Download", visible=False)
+        progresso_download_file = gr.File(label="Download do Progresso", visible=False)
 
         with gr.Row():
             peticao_aluno_output = gr.Textbox(label="Petição do Aluno", lines=15, interactive=False)
@@ -689,11 +689,11 @@ def interface_gradio():
             outputs=[processing_message]
         )
         clear_btn.click(
-            fn=lambda: [None, None, None, None, "", "", "", "", "", gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)],
+            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, progresso_json_output, progresso_download_link, processing_message]
+                     visualizacao_detalhada_output, progresso_json_output, progresso_download_file, processing_message]
         )
         comparar_btn.click(
             fn=comparar_peticao_melhorada,
@@ -708,7 +708,7 @@ def interface_gradio():
         ).then(
             fn=lambda caso, peticao: salvar_progresso(caso, peticao),
             inputs=[caso_input, peticao_texto],
-            outputs=[progresso_json_output, resultado_output, progresso_download_link]
+            outputs=[progresso_json_output, resultado_output, progresso_download_file]
         ).then(
             fn=lambda: gr.update(value="", visible=False),
             inputs=None,
@@ -716,7 +716,7 @@ def interface_gradio():
         ).then(
             fn=lambda: gr.update(visible=True),
             inputs=None,
-            outputs=[progresso_json_output, progresso_download_link]
+            outputs=[progresso_json_output, progresso_download_file]
         )
         carregar_btn.click(
             fn=lambda arquivo: carregar_progresso(arquivo.name),