Update app.py

app.py

@@ -17,7 +17,6 @@ from ratelimiter import RateLimiter
 import difflib
 from html import escape
 import random
-from io import BytesIO
 
 load_dotenv()
 
@@ -496,16 +495,20 @@ Diferença Líquida: {total_adicionadas - total_removidas} linhas
 
 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"
+    try:
+        dados = {
+            "caso": caso or "",
+            "peticao_texto": peticao_texto or "",
+            "data": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+            "historico": HISTORICO
+        }
+        caminho_temp = "progresso_salvo.json"
+        with open(caminho_temp, "w", encoding="utf-8") as f:
+            json.dump(dados, f, ensure_ascii=False)
+        return caminho_temp
+    except Exception as e:
+        print(f"Erro ao salvar progresso: {str(e)}")
+        return None
 
 def carregar_progresso(arquivo):
     """Carrega o progresso a partir de um arquivo JSON."""
@@ -519,7 +522,7 @@ def carregar_progresso(arquivo):
         historico_carregado = dados.get("historico", [])
         if historico_carregado:
             global HISTORICO
-            HISTORICO = historico_carregado  # Atualiza o histórico global
+            HISTORICO = historico_carregado
         return caso, peticao_texto, "✅ Progresso carregado com sucesso!"
     except Exception as e:
         return "", "", f"❌ Erro ao carregar o arquivo: {str(e)}"
@@ -618,7 +621,7 @@ def interface_gradio():
         download_btn.click(
             fn=salvar_progresso,
             inputs=[caso_input, peticao_texto],
-            outputs=[gr.File(label="Download do Progresso"), gr.State()]
+            outputs=gr.File(label="Download do Progresso")
         )
         upload_progresso.change(
             fn=carregar_progresso,