Roulette

from random import randrange
from matplotlib import pyplot as plt

# Function 1
def doIContinue(answer):
    flag = True
    if answer.lower() == "no":
        print("So, you want to stop. Nice game!")
        flag = False
    elif answer.lower() == "yes" or answer.lower() == "sure":
        print("Wise choice, maybe not. Let's continue to the next bet!")
        flag = True
    else:
        print("You did not answer neither 'yes' nor 'no'. I will suppose 'yes', next time be more careful.")
        flag = True
    print()
    return flag

# **********************************************************************************************************
# ********************************** MAIN FUNCTION *********************************************************
# **********************************************************************************************************
print()
print("Welcome to the Roulette game. Rules are simple. You can only bet in red color")
wins = 0
tries = 0
stats01 = []
bets = []
money = int(input("Money here: "))
while money < 0:
    money = int(input("Money here: "))
initialMoney = money
answer = "yes"

# I will simulate the simulation with another way: red will be if my number is between 1 and 18,
# black will be if my number is between 19 and 36 and "0" will be number 37
while money > 0 and answer != "no":
    answer = input("Do we continue? ")
    flag = doIContinue(answer)
    if flag == True:            # That means we continue
        tries += 1
        print("************************ Try ", tries, "************************")
        bet = int(input("Put your bet here: "))
        while bet > money:
            bet = int(input("Put your bet here: "))
        bets.append(bet)
        # Now, we have the bet that user put in. This bet will be substracted from sum = money
        money = money - bet
        # We will suppose that our user ALWAYS bets 'red', so we want the random generated number to be
        # between 1 and 18
        randomNumber = randrange(0, 37)
        print("Bet =", bet, ", randomNumber =", randomNumber)
        if randomNumber >= 1 and randomNumber <= 18:
            wins += 1
            money += 2 * bet
            stats01.append(1)
            print("You won, because 1 <=", randomNumber, "<= 18")
            print("Current money: ", money)
        elif randomNumber >= 19 and randomNumber <= 36:
            stats01.append(0)
            print("You lost, because 19 <=", randomNumber, "<= 36")
            print("Current money: ", money)
        else:
            stats01.append(0)
            print("You lost, because randomNumber = 0 ----> green color")
            print("Current money: ", money)
        print("*******************************************************")
        print()

# Now, the basic process has finished. The remaining stuff is to show the stats in the user and the plot of his luck
print()
print("*******************************************************")
print("Here, I will present you the stats of your gameplay.")
print("Tries =", tries, ", wins =", wins, ", losses =", tries - wins)
winPercentage = wins / tries
winPercentage = round(10000 * winPercentage) / 10000
print("Win percentage  = ", 100 * winPercentage, "%")
print()
gain = money - initialMoney
gainPercentage = gain / money
gainPercentage = round(10000 * gainPercentage) / 10000
print("Initial money =", initialMoney, ", Final Money =", money, "(", 100 * gainPercentage, "%)")
# Plotting our data
plt.plot(stats01, 'ro')
plt.title("0 = Loss, 1 = Win")
plt.xlabel("Rounds")
plt.ylabel("Win/Loss")
plt.show()
plt.plot(bets, 'bx')
plt.title("Bets value per round")
plt.xlabel("Rounds")
plt.ylabel("Bet")
plt.show()