from bokeh.plotting import figure, output_file, show
from bokeh.io import output_notebook
from bokeh.models import ColumnDataSource

# Set Bokeh output to notebook
output_notebook()

import numpy as np
from bokeh.plotting import figure, show

# Tinkerbell map parameters
a, b, c, d = 0.9, -0.6013, 2.0, 0.5
num_points = 10000  # Number of iterations
x0, y0 = 0.1, 0.1  # Initial conditions

# Generate points for the Tinkerbell map
x, y = [x0], [y0]
for _ in range(num_points - 1):
    x_new = x[-1]**2 - y[-1]**2 + a * x[-1] + b * y[-1]
    y_new = 2 * x[-1] * y[-1] + c * x[-1] + d * y[-1]
    x.append(x_new)
    y.append(y_new)

# Split data into segments for multi-line
num_segments = 7
xs = np.array_split(x, num_segments)
ys = np.array_split(y, num_segments)

# Define a color palette
colors = ["#FFF7FB", "#ECE7F2", "#D0D1E6", "#A6BDDB", "#74A9CF", "#2B8CBE", "#045A8D"]

# Create the Bokeh figure
p = figure(title="Tinkerbell Map Visualization",
           background_fill_color="#f9f9f9",
           x_axis_label="X",
           y_axis_label="Y",
           match_aspect=True)

# Add the multi_line glyph
p.multi_line(xs, ys, line_color=colors, line_alpha=0.8, line_width=1.5)

# Show the plot
show(p)