Update app.py

app.py

@@ -6,6 +6,7 @@ import PyPDF2
 import docx
 import io
 import time
+import tempfile
 from dotenv import load_dotenv
 from pdfplumber import open as pdfplumber_open
 from PIL import Image
@@ -546,25 +547,33 @@ 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
         logger.debug("Gerando JSON content...")
         json_content = json.dumps({"caso": caso, "peticao": peticao}, ensure_ascii=False, indent=4)
         logger.debug(f"JSON content gerado: {json_content[:100]}...")
 
-        # Mensagem de sucesso com instruções para salvar manualmente
+        # Cria um arquivo temporário para salvar o JSON
+        logger.debug("Criando arquivo temporário...")
+        with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.json', prefix='progresso_', encoding='utf-8') as temp_file:
+            temp_file.write(json_content)
+            temp_file_path = temp_file.name
+        logger.debug(f"Arquivo temporário criado: {temp_file_path}")
+
+        # Mensagem de sucesso com instruções para download ou fallback
         success_message = (
-            "✅ Progresso salvo! Copie o conteúdo abaixo e cole em um arquivo com extensão `.json` "
-            "(ex.: `progresso.json`). Para isso, crie um novo arquivo de texto, cole o conteúdo, "
-            "salve com o nome `progresso.json` e use a opção 'Carregar Progresso' para restaurar."
+            "✅ Progresso salvo! O download do arquivo `progresso.json` deve iniciar automaticamente. "
+            "Se o download não iniciar, copie o conteúdo abaixo e cole em um arquivo com extensão `.json` "
+            "(ex.: `progresso.json`). No celular, você pode copiar o texto, abrir um app de notas, colar o conteúdo, "
+            "e salvar como `progresso.json`. Use a opção 'Carregar Progresso' para restaurar."
         )
 
-        logger.info("Progresso salvo com sucesso como string.")
-        return json_content, success_message
+        logger.info("Progresso salvo com sucesso.")
+        return json_content, success_message, temp_file_path
     except Exception as e:
         logger.error(f"Erro ao salvar progresso: {str(e)}", exc_info=True)
-        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...")
@@ -645,8 +654,9 @@ def interface_gradio():
 
         resultado_output = gr.Markdown(label="📊 Correção Oficial e Nota", show_copy_button=True)
 
-        # Componente para exibir o progresso salvo
+        # Componentes para exibir o progresso salvo
         progresso_json_output = gr.Textbox(label="Progresso Salvo (JSON)", lines=5, interactive=False, 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)
@@ -695,11 +705,11 @@ def interface_gradio():
             outputs=[processing_message]
         )
         clear_btn.click(
-            fn=lambda: [None, None, None, None, "", "", "", "", "", 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, processing_message]
+                     visualizacao_detalhada_output, progresso_json_output, progresso_download_file, processing_message]
         )
         comparar_btn.click(
             fn=comparar_peticao_melhorada,
@@ -714,7 +724,7 @@ def interface_gradio():
         ).then(
             fn=lambda caso, peticao: salvar_progresso(caso, peticao),
             inputs=[caso_input, peticao_texto],
-            outputs=[progresso_json_output, resultado_output]
+            outputs=[progresso_json_output, resultado_output, progresso_download_file]
         ).then(
             fn=lambda: gr.update(value="", visible=False),
             inputs=None,
@@ -722,7 +732,7 @@ def interface_gradio():
         ).then(
             fn=lambda: gr.update(visible=True),
             inputs=None,
-            outputs=[progresso_json_output]
+            outputs=[progresso_json_output, progresso_download_file]
         )
         carregar_btn.click(
             fn=lambda arquivo: carregar_progresso(arquivo.name),