Update server.py

server.py

@@ -10,7 +10,8 @@ from typing import Any, Dict, List, Optional, Tuple
 from flask import Flask, jsonify, request, send_from_directory
 
 APP_ROOT = Path(__file__).parent.resolve()
-DATA_FILE = APP_ROOT / "web_invoice_store.json"
+DATA_DIR = Path(os.environ.get("DATA_DIR", APP_ROOT))
+DATA_FILE = DATA_DIR / "web_invoice_store.json"
 INVOICE_HISTORY_LIMIT = 200
 
 VAT_RATES: Dict[str, Optional[Decimal]] = {
@@ -362,3 +363,6 @@ def api_invoices() -> Any:
 if __name__ == "__main__":
     port = int(os.environ.get("PORT", "5000"))
     app.run(host="0.0.0.0", port=port, debug=True)
+
+    DATA_DIR = Path(os.environ.get("DATA_DIR", APP_ROOT))
+DATA_FILE = DATA_DIR / "web_invoice_store.json"