Skip to content

🎨 Palette: Add colors and emojis to trading simulation output #15

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
google-labs-jules wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

🎨 Palette: Add colors and emojis to trading simulation output #15

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
## 2025-02-23 - CLI Visual Hierarchy
**Learning:** CLI tools are often neglected in terms of UX. Adding semantic colors (Green/Red) and emojis (🟢/🔴) significantly improves the scannability of logs, acting as visual anchors for important events.
**Action:** When working on CLI tools, always check if the output stream supports ANSI colors and use them to highlight key state changes or results, ensuring a fallback or textual indicator (like +/- signs or icons) exists for accessibility.
29 changes: 20 additions & 9 deletions bitcoin_trading_simulation.py
View file
Open in desktop
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'
ENDC = '\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 @@ -45,13 +52,13 @@ def simulate_trading(signals, initial_cash=10000):
"""
Simulates trading based on signals and prints a daily ledger.
"""
portfolio = pd.DataFrame(index=signals.index).fillna(0.0)
portfolio = pd.DataFrame(index=signals.index)
portfolio['price'] = signals['price']
portfolio['cash'] = initial_cash
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"\n{Colors.HEADER}------ Daily Trading Ledger ------{Colors.ENDC}")
for i, row in signals.iterrows():
if i > 0:
portfolio.loc[i, 'cash'] = portfolio.loc[i-1, 'cash']
Expand All @@ -62,14 +69,14 @@ 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"Day {i}: {Colors.GREEN}🟢 Buy {btc_to_buy:.4f} BTC at ${row['price']:.2f}{Colors.ENDC}")

# 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"Day {i}: {Colors.RED}🔴 Sell {portfolio.loc[i, 'btc']:.4f} BTC at ${row['price']:.2f}{Colors.ENDC}")
portfolio.loc[i, 'btc'] = 0

portfolio.loc[i, 'total_value'] = portfolio.loc[i, 'cash'] + portfolio.loc[i, 'btc'] * row['price']
Expand Down Expand Up @@ -99,9 +106,13 @@ 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}------ Final Portfolio Performance ------{Colors.ENDC}")
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
profit_icon = "📈" if profit >= 0 else "📉"
print(f"Profit/Loss: {profit_color}{profit_icon} ${profit:.2f}{Colors.ENDC}")

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