suggestions

crypto/gogo/main.py

@@ -1,5 +1,3 @@
-# main.py
-
 import asyncio
 import torch
 import torch.nn as nn
@@ -17,8 +15,9 @@ from model.trading_model import TradingModel
 from training.rl_agent import ContinuousRLAgent, ReplayBuffer
 from training.train_historical import train_on_historical_data, load_best_checkpoint, save_candles_cache, CACHE_FILE, BEST_DIR
 from data.data_utils import get_aligned_candle_with_index, get_features_for_tf
+import argparse
 
-async def main():
+async def main_training():
     symbol = 'BTC/USDT'
     data_manager = LiveDataManager(symbol)
 
@@ -293,4 +292,11 @@ def plot_trade_history(candles, trade_history):
     plt.show()
 
 if __name__ == '__main__':
-    asyncio.run(main_backtest())
+    parser = argparse.ArgumentParser(description='Trading Bot Modes')
+    parser.add_argument('--mode', type=str, default='backtest', choices=['train', 'backtest'], help='Choose mode: train or backtest')
+    args = parser.parse_args()
+
+    if args.mode == 'train':
+        asyncio.run(main_training())
+    elif args.mode == 'backtest':
+        asyncio.run(main_backtest())