Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .Jules/palette.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
## 2024-05-22 - CLI Visual Enhancement
**Learning:** CLI tools often suffer from "wall of text" syndrome. Simple color coding (Green=Good/Buy, Red=Bad/Sell) dramatically improves scanability.
**Action:** When working on CLI tools, always check if ANSI colors can be added to highlight key events/metrics.
29 changes: 21 additions & 8 deletions bitcoin_trading_simulation.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,13 @@
import numpy as np
import pandas as pd

class Colors:
HEADER = '\033[95m'
GREEN = '\033[92m'
RED = '\033[91m'
RESET = '\033[0m'
BOLD = '\033[1m'

def simulate_bitcoin_prices(days=60, initial_price=50000, volatility=0.02):
"""
Simulates Bitcoin prices for a given number of days using Geometric Brownian Motion.
Expand Down Expand Up @@ -47,11 +54,12 @@ def simulate_trading(signals, initial_cash=10000):
"""
portfolio = pd.DataFrame(index=signals.index).fillna(0.0)
portfolio['price'] = signals['price']
portfolio['cash'] = initial_cash
# Explicitly cast to float to avoid FutureWarning
portfolio['cash'] = float(initial_cash)
portfolio['btc'] = 0.0
portfolio['total_value'] = portfolio['cash']
portfolio['total_value'] = float(initial_cash)

print("------ Daily Trading Ledger ------")
print(f"{Colors.HEADER}{Colors.BOLD}------ πŸ“’ Daily Trading Ledger ------{Colors.RESET}")
for i, row in signals.iterrows():
if i > 0:
portfolio.loc[i, 'cash'] = portfolio.loc[i-1, 'cash']
Expand All @@ -62,17 +70,19 @@ def simulate_trading(signals, initial_cash=10000):
btc_to_buy = portfolio.loc[i, 'cash'] / row['price']
portfolio.loc[i, 'btc'] += btc_to_buy
portfolio.loc[i, 'cash'] -= btc_to_buy * row['price']
print(f"Day {i}: Buy {btc_to_buy:.4f} BTC at ${row['price']:.2f}")
print(f"{Colors.GREEN}Day {i}: 🟒 Buy {btc_to_buy:.4f} BTC at ${row['price']:.2f}{Colors.RESET}")

# Sell signal
elif row['positions'] == -2.0:
if portfolio.loc[i, 'btc'] > 0:
cash_received = portfolio.loc[i, 'btc'] * row['price']
portfolio.loc[i, 'cash'] += cash_received
print(f"Day {i}: Sell {portfolio.loc[i, 'btc']:.4f} BTC at ${row['price']:.2f}")
print(f"{Colors.RED}Day {i}: πŸ”΄ Sell {portfolio.loc[i, 'btc']:.4f} BTC at ${row['price']:.2f}{Colors.RESET}")
portfolio.loc[i, 'btc'] = 0

portfolio.loc[i, 'total_value'] = portfolio.loc[i, 'cash'] + portfolio.loc[i, 'btc'] * row['price']

# Optional: Gray out normal days? No, let's keep it simple.
print(f"Day {i}: Portfolio Value: ${portfolio.loc[i, 'total_value']:.2f}, Cash: ${portfolio.loc[i, 'cash']:.2f}, BTC: {portfolio.loc[i, 'btc']:.4f}")

return portfolio
Expand All @@ -99,9 +109,12 @@ def simulate_trading(signals, initial_cash=10000):
buy_and_hold_btc = initial_cash / prices.iloc[0]
buy_and_hold_value = buy_and_hold_btc * prices.iloc[-1]

print("\n------ Final Portfolio Performance ------")
print(f"\n{Colors.HEADER}{Colors.BOLD}------ πŸ“Š Final Portfolio Performance ------{Colors.RESET}")
print(f"Initial Cash: ${initial_cash:.2f}")
print(f"Final Portfolio Value: ${final_value:.2f}")
print(f"Profit/Loss: ${profit:.2f}")

profit_color = Colors.GREEN if profit >= 0 else Colors.RED
print(f"Profit/Loss: {profit_color}${profit:.2f}{Colors.RESET}")

print(f"Buy and Hold Strategy Value: ${buy_and_hold_value:.2f}")
print("-----------------------------------------")
print(f"{Colors.HEADER}-----------------------------------------{Colors.RESET}")
Loading