update cli.py to use create_byte_stream

1 files changed, 2 insertions, 0 deletions

diff --git a/cli.py b/cli.py
index b1d4000..f2c8771 100644
--- a/cli.py
+++ b/cli.py
@@ -3,6 +3,7 @@ import json
 from obfuscator.csv_reader import CSVReader
 from obfuscator.obfuscate import obfuscate
 from obfuscator.logger import get_logger
+from obfuscator.csv_writer import create_byte_stream
 
 # Create the logger
 logger = get_logger("CLI")
@@ -44,6 +45,7 @@ def main():
     obfuscated_data = obfuscate(data, args.pii)
     # For debug purposes, log the obfuscated data as JSON for readability
     logger.debug("Obfuscated data (JSON): " + json.dumps(obfuscated_data, indent=4))
+    return create_byte_stream(obfuscated_data)
 
 
 # If the script is run directly (as it should be), call the main function