Skip to content

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

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

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

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

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 @@
## 2024-05-22 - [CLI Output Formatting]
**Learning:** Adding colors and emojis to CLI output significantly improves readability and user engagement, transforming a dry log into a visual narrative.
**Action:** When working on CLI tools in the future, always look for opportunities to use ANSI colors to highlight key events (like success/failure, buy/sell) and use emojis sparingly to add visual cues.
49 changes: 38 additions & 11 deletions bitcoin_trading_simulation.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,17 @@
import numpy as np
import pandas as pd

# ANSI Color Codes
class Colors:
HEADER = '\033[95m'
BLUE = '\033[94m'
GREEN = '\033[92m'
RED = '\033[91m'
YELLOW = '\033[93m'
BOLD = '\033[1m'
UNDERLINE = '\033[4m'
ENDC = '\033[0m'

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 +58,11 @@ def simulate_trading(signals, initial_cash=10000):
"""
portfolio = pd.DataFrame(index=signals.index).fillna(0.0)
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"{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,18 +73,27 @@ 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']
print(f"Day {i}: Portfolio Value: ${portfolio.loc[i, 'total_value']:.2f}, Cash: ${portfolio.loc[i, 'cash']:.2f}, BTC: {portfolio.loc[i, 'btc']:.4f}")

# Color code daily value based on previous day
val_color = Colors.ENDC
if i > 0:
if portfolio.loc[i, 'total_value'] > portfolio.loc[i-1, 'total_value']:
val_color = Colors.GREEN
elif portfolio.loc[i, 'total_value'] < portfolio.loc[i-1, 'total_value']:
val_color = Colors.RED

print(f"Day {i}: Portfolio Value: {val_color}${portfolio.loc[i, 'total_value']:.2f}{Colors.ENDC}, Cash: ${portfolio.loc[i, 'cash']:.2f}, BTC: {portfolio.loc[i, 'btc']:.4f}")

return portfolio

Expand All @@ -99,9 +119,16 @@ 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}")
print(f"Buy and Hold Strategy Value: ${buy_and_hold_value:.2f}")
print("-----------------------------------------")

final_color = Colors.GREEN if final_value >= initial_cash else Colors.RED
print(f"Final Portfolio Value: {final_color}${final_value:.2f}{Colors.ENDC}")

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

buy_hold_color = Colors.GREEN if buy_and_hold_value >= initial_cash else Colors.RED
print(f"Buy and Hold Strategy Value: {buy_hold_color}${buy_and_hold_value:.2f}{Colors.ENDC}")
print(f"{Colors.HEADER}-----------------------------------------{Colors.ENDC}")
Loading