import random
import matplotlib.pyplot as plt
from colorama import Fore, Style, init
init(autoreset=True)  # Auto-reset color after each print

import os
os.system('cls' if os.name == 'nt' else 'clear')  # Clear screen on start

cryptos = {
    "Bitcoin": {"price": 100.0, "sensitivity": 0.5, "volatility": 0.0, "total_supply": 1000.0, "price_history": [100.0]},
    "Ethereum": {"price": 50.0, "sensitivity": 0.3, "volatility": 0.0, "total_supply": 1500.0, "price_history": [50.0]},
}

users = {}

def available_supply(coin):
    owned = sum(user['coins'].get(coin, 0.0) for user in users.values())
    return max(0.0, cryptos[coin]["total_supply"] - owned)

def save_graph(coin):
    plt.clf()
    plt.plot(cryptos[coin]["price_history"], label=f"{coin} Price")
    plt.title(f"{coin} Price Over Time")
    plt.xlabel("Turn")
    plt.ylabel("Price")
    plt.legend()
    plt.grid(True)
    filename = f"{coin.lower()}_price.png"
    plt.savefig(filename)
    print(Fore.MAGENTA + f"📊 Graph saved as '{filename}'")

def total_value(user):
    return user["cash"] + sum(user["coins"].get(c, 0.0) * cryptos[c]["price"] for c in cryptos)

# Intro
print(Fore.YELLOW + Style.BRIGHT + "💰 Welcome to Crypto Kids Multiplayer — Multi-Coin Edition!")
print(Style.DIM + "🎮 Register users, then trade Bitcoin and Ethereum in turn-based rounds.\n")

# Register Users
while True:
    user_input = input(Fore.CYAN + "Enter username to register/login (or type 'start' to begin): ").strip()
    if user_input.lower() == "start":
        if not users:
            print(Fore.RED + "⚠️ Add at least one user to begin.")
            continue
        break
    if user_input not in users:
        users[user_input] = {"cash": 1000.0, "coins": {coin: 0.0 for coin in cryptos}}
        print(Fore.GREEN + f"🆕 User '{user_input}' registered with $1000 cash.")
    else:
        print(Fore.BLUE + f"👋 Welcome back, {user_input}!")

usernames = list(users.keys())
turn = 0

print(Fore.YELLOW + "\n🎬 Game started! Trade using +amount to buy or -amount to sell. Type 'exit' to quit.\n")

# Main Game Loop
while True:
    # Display all users' holdings before each turn
    print(Fore.LIGHTWHITE_EX + "\n📊 CURRENT STANDINGS")
    print("="*40)
    for uname, data in users.items():
        print(Fore.LIGHTCYAN_EX + f"👤 {uname}:")
        print(f"   💼 Cash: {Fore.GREEN}${data['cash']:.2f}")
        for c in cryptos:
            print(f"   🪙 {c}: {Fore.YELLOW}{data['coins'][c]:.2f}")
        print(f"   📈 Total Value: {Fore.CYAN}${total_value(data):.2f}\n")

    username = usernames[turn % len(usernames)]
    user = users[username]
    print(Fore.LIGHTWHITE_EX + "\n" + "="*40)
    print(Fore.CYAN + f"🔁 {username.upper()}'S TURN")
    print(Fore.LIGHTWHITE_EX + "="*40)
    print(f"💼 Cash: {Fore.GREEN}${user['cash']:.2f}")
    print("🪙 Your Holdings:")
    for coin in cryptos:
        print(f"  - {Fore.YELLOW}{coin}: {user['coins'][coin]:.2f}")

    coin = input(Fore.CYAN + f"\nSelect a coin to trade {list(cryptos.keys())}: ").strip()
    if coin not in cryptos:
        print(Fore.RED + "⚠️ Invalid coin selection.")
        continue

    price = cryptos[coin]["price"]
    supply = available_supply(coin)
    print(Fore.LIGHTBLUE_EX + f"\n📈 {coin} Price: ${price:.2f}")
    print(f"📊 Supply Available: {supply:.2f} / {cryptos[coin]['total_supply']:.0f}")

    user_input = input(Fore.CYAN + f"Enter trade amount for {coin} (e.g., +10 or -5), or 'exit': ").strip()
    if user_input.lower() == "exit":
        print(Fore.YELLOW + "\n📉 GAME OVER — FINAL BALANCES:\n")
        for uname, data in users.items():
            print(Fore.LIGHTWHITE_EX + f"👤 {uname}:")
            print(f"   💼 Cash: {Fore.GREEN}${data['cash']:.2f}")
            for c in cryptos:
                print(f"   🪙 {c}: {Fore.YELLOW}{data['coins'][c]:.2f}")
            print(f"   📈 Total Value: {Fore.CYAN}${total_value(data):.2f}\n")
        break

    try:
        trade_amount = int(user_input)
    except ValueError:
        print(Fore.RED + "⚠️ Invalid input format. Use +10 or -5.")
        continue

    # Buy
    if trade_amount > 0:
        if supply < trade_amount:
            print(Fore.RED + "❌ Not enough coins available.")
            continue
        cost = trade_amount * price
        if user["cash"] >= cost:
            user["cash"] -= cost
            user["coins"][coin] += trade_amount
            print(Fore.GREEN + f"✅ Bought {trade_amount} {coin} for ${cost:.2f}")
        else:
            print(Fore.RED + "❌ Not enough cash.")
            continue

    # Sell
    elif trade_amount < 0:
        sell_amt = -trade_amount
        if user["coins"][coin] >= sell_amt:
            user["coins"][coin] -= sell_amt
            user["cash"] += sell_amt * price
            print(Fore.GREEN + f"✅ Sold {sell_amt} {coin} for ${sell_amt * price:.2f}")
        else:
            print(Fore.RED + "❌ Not enough coins.")
            continue

    # Price Update
    net_demand = trade_amount
    crypto = cryptos[coin]
    price_change = net_demand * crypto["sensitivity"]
    randomness = random.uniform(-crypto["volatility"], crypto["volatility"]) * crypto["price"]
    new_price = crypto["price"] + price_change + randomness
    crypto["price"] = max(1.0, new_price)
    crypto["price_history"].append(crypto["price"])

    save_graph(coin)
    turn += 1