Update main.py

main.py

@@ -12,7 +12,7 @@ import os
 from datetime import datetime
 
 # Set up logging
-log_directory = "logs"
+log_directory = os.environ.get('LOG_DIR', 'logs')
 if not os.path.exists(log_directory):
     os.makedirs(log_directory)