from BitcoinPricePredictor import BitcoinPricePredictor

if __name__ == "__main__":
    # For daily predictions (default)
    predictor_daily = BitcoinPricePredictor(db_path='bitcoin_historical_data.db', timeframe='H')
    
    # For weekly predictions
    # predictor_weekly = BitcoinPricePredictor(db_path='bitcoin_historical_data.db', timeframe='W')
    
    # Choose which predictor to use
    predictor = predictor_daily
    
    predictor.load_and_prepare_data()
    predictor.train_model()
    predictor.evaluate_model()
    predictor.plot_history()