Update app.py

app.py

@@ -17,7 +17,7 @@ from ratelimiter import RateLimiter
 import difflib
 from html import escape
 import random
-import tempfile
+import base64
 import logging
 
 # Configura o logging para depuração
@@ -544,33 +544,20 @@ 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 gr.File.update(visible=False), "⚠️ Caso ou petição estão vazios. Preencha ambos para salvar."
-        
-        logger.info("Criando arquivo temporário...")
-        # Especifica um diretório personalizado para o arquivo temporário
-        temp_dir = os.path.join(os.getcwd(), "temp_files")
-        os.makedirs(temp_dir, exist_ok=True)
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".json", dir=temp_dir, mode='w', encoding='utf-8') as temp_file:
-            json.dump({"caso": caso, "peticao": peticao}, temp_file, ensure_ascii=False, indent=4)
-            arquivo_saida = temp_file.name
-        
-        # Verifica se o arquivo foi criado e é acessível
-        if not os.path.exists(arquivo_saida):
-            logger.error(f"Arquivo temporário não encontrado: {arquivo_saida}")
-            return gr.File.update(visible=False), f"❌ Erro: Arquivo temporário não foi criado."
-        
-        logger.info(f"Arquivo temporário criado com sucesso: {arquivo_saida}")
-        # Retorna o arquivo temporário para download e uma mensagem de sucesso
-        return gr.File.update(value=arquivo_saida, visible=True), "✅ Progresso salvo com sucesso!"
+            return "", "⚠️ Caso ou petição estão vazios. Preencha ambos para salvar.", ""
+
+        # 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})"
+
+        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
     except Exception as e:
         logger.error(f"Erro ao salvar progresso: {str(e)}")
-        # Fallback: Retorna o conteúdo como string se o arquivo não puder ser criado
-        try:
-            json_content = json.dumps({"caso": caso, "peticao": peticao}, ensure_ascii=False, indent=4)
-            return gr.File.update(visible=False), f"✅ Progresso salvo! Copie o conteúdo abaixo:\n\n```json\n{json_content}\n```"
-        except Exception as e2:
-            logger.error(f"Erro ao gerar JSON como string: {str(e2)}")
-            return gr.File.update(visible=False), f"❌ Erro ao salvar o progresso: {str(e)}"
+        return "", f"❌ Erro ao salvar o progresso: {str(e)}", ""
 
 def carregar_progresso(arquivo_entrada):
     logger.info("Iniciando carregamento de progresso...")
@@ -648,10 +635,13 @@ def interface_gradio():
             salvar_btn = gr.Button("💾 Salvar Progresso")
             carregar_btn = gr.Button("📂 Carregar Progresso")
             arquivo_progresso = gr.File(label="Arquivo de Progresso", file_types=[".json"])
-            download_file = gr.File(label="Download do Progresso", interactive=False)
 
         resultado_output = gr.Markdown(label="📊 Correção Oficial e Nota", show_copy_button=True)
 
+        # 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)
+
         with gr.Row():
             peticao_aluno_output = gr.Textbox(label="Petição do Aluno", lines=15, interactive=False)
             peticao_corrigida_output = gr.Textbox(label="Petição Corrigida", lines=15, interactive=False)
@@ -699,11 +689,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), 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, processing_message]
+                     visualizacao_detalhada_output, progresso_json_output, progresso_download_link, processing_message]
         )
         comparar_btn.click(
             fn=comparar_peticao_melhorada,
@@ -718,11 +708,15 @@ def interface_gradio():
         ).then(
             fn=lambda caso, peticao: salvar_progresso(caso, peticao),
             inputs=[caso_input, peticao_texto],
-            outputs=[download_file, resultado_output]
+            outputs=[progresso_json_output, resultado_output, progresso_download_link]
         ).then(
             fn=lambda: gr.update(value="", visible=False),
             inputs=None,
             outputs=[processing_message]
+        ).then(
+            fn=lambda: gr.update(visible=True),
+            inputs=None,
+            outputs=[progresso_json_output, progresso_download_link]
         )
         carregar_btn.click(
             fn=lambda arquivo: carregar_progresso(arquivo.name),