# Numeric per-day patient counts (= the lengths of the emoji groups above). The
# figure and the day-by-day text below run on these flat Int counts so they
# compile to live WebAssembly islands — no emoji string-length work in the kernel.
patient_counts = (exponential ? [1, 2, 5, 8, 16, 32, 65, 0] : [1, 2, 5, 8, 4, 2, 1, 0])[1:len]

begin
	# A SIMPLER but faithful WebAssembly-compilable figure of the SAME data the
	# notebook is about: the 1D CONVOLUTION of the two daily signals (new pills
	# per day ⋆ new patients per day) = the stockpile of pills you must hold each
	# day, with the current day highlighted. Everything is built from flat
	# Float64 vectors with explicit loops so it compiles to a live island, and
	# the convolution is the very `simple_conv` the notebook teaches.

	# the convolution result: the stockpile required each day — the same
	# explicit kernel the whole notebook is about
	y3 = simple_conv(y1_pills, reverse(y2_patients))
	n3 = length(y3)
	xs3 = Float64[]; ys3 = Float64[]
	for i in 1:n3
		push!(xs3, Float64(i))
		push!(ys3, Float64(y3[i]))
	end

	# clamp the day slider into a valid index so NO sampled bond value can trap
	# (the differential oracle samples Slider possible_values = POSITION INDICES)
	day = k_conv
	if day < 1
		day = 1
	elseif day > n3
		day = n3
	end
	day_x = Float64(day)
	day_y = Float64(y3[day])

	# build the figure with the proven-minimal WasmMakie API (bare Axis + bare
	# lines!, exactly like the known-good figure island): the convolution curve,
	# plus a vertical stem at the current day so it stays visible as you scrub.
	f = Figure(size = (640, 380))
	ax = Axis(f[1, 1])
	lines!(ax, xs3, ys3)
	lines!(ax, [day_x, day_x], [0.0, day_y])
	f
end

begin
	# Number of time steps
	nb = 8 

	# Slider for how long the pandemic should go 
	len_slider = @bind len PlutoUI.Slider(3:1:nb-1, show_value=true, default=5)

	# Checkboxes for more options
	show_stairs = @bind stairs CheckBox(default=true)
	make_exponential = @bind exponential CheckBox()
		
end;

begin
	# Number of patients per day (First function) — flipped + left-padded to 8,
	# exactly as `patients_flipped` is, but built from flat Int counts.
	numbers_patients = append!([0 for i in 1:8-length(patient_counts)], reverse(patient_counts))
	y2_patients = append!([0 for i in 1:9], numbers_patients, [0 for i in 1:16-length(numbers_patients)])

	# Number of pills per day (Second function) — the scrubbable counts directly.
	numbers_pills = treatment_in
	y1_pills = append!([0 for i in 1:9], numbers_pills, [0 for i in 1:16-length(numbers_pills)])

end;

# (the figure is built and displayed by the single cell below — keeping all the
# WasmMakie drawing in the one cell whose VALUE is the Figure lets it ship as a
# live WebAssembly canvas island)

begin
	# plotting:
	using WasmMakie, ColorSchemes, Colors

	# widgets and layout:
	using PlutoUI
	using PlutoUI.ExperimentalLayout: grid, vbox, hbox, Div
	using HypertextLiteral
end

# (the emoji-pill PNG download is gone — WasmMakie markers are plain shapes,
# so the visualization is fully self-contained and wasm-compilable)

treatment_in = [a1_s, a2_s, a3_s, a4_s, a5_s, a6_s, a7_s, a8_s];

begin
	_rgba(c) = (Float64(red(c)), Float64(green(c)), Float64(blue(c)), 1.0)
	color_grey = (0.5, 0.5, 0.5, 1.0)
	colors = [_rgba(ColorSchemes.viridis[i]) for i in 1:40:256]
	col_length = length(colors)
	color_blue = _rgba(ColorSchemes.viridis[116])
end;

sidebar = Div([
	@htl("""
	<header>
	<span class="sidebar-toggle open-sidebar">🕹</span>
	<span class="sidebar-toggle closed-sidebar">🕹</span>
	Interactive Sliders
	</header>
	"""),
	md"""
	Here are all interactive bits of the notebook at one place. Feel free to change them!
	"""
], class="plutoui-sidebar aside")

sidebar2 = Div([
	md""" **Choose a day:**""",
	k_slider,
	md"""**Choose how long the pandemic is:**""",
	len_slider, 
	md"""**Change the treatment:**""",
	md""" $(a1) $(a2) $(a3) $(a4) $(a5) $(a6) $(a7) $(a8)""",
	md"""**Make the pandemic exponential:** $(make_exponential)""",
	md"""**Show the stairs** $(show_stairs)"""
], class="plutoui-sidebar aside third")

html"""
<style>
	div.plutoui-sidebar.aside {
		position: fixed;
		right: 1rem;
		top: 10rem;
		width: min(80vw, 300px);
		padding: 10px;
		border: 3px solid rgba(0, 0, 0, 0.15);
		border-radius: 10px;
		box-shadow: 0 0 11px 0px #00000010;
		max-height: calc(100vh - 5rem - 56px);
		overflow: auto;
		z-index: 40;
		background: white;
		transition: transform 300ms cubic-bezier(0.18, 0.89, 0.45, 1.12);
		color: var(--pluto-output-color);
		background-color: var(--main-bg-color);
	}

	.second {
		top: 17.5rem !important;
	}

	.third {
		top: 17.5rem !important;
	}
	
	div.plutoui-sidebar.aside.hide {
		transform: translateX(calc(100% - 28px));
	}
	
	.plutoui-sidebar header {
		display: block;
		font-size: 1.5em;
		margin-top: -0.1em;
		margin-bottom: 0.4em;
		padding-bottom: 0.4em;
		margin-left: 0;
		margin-right: 0;
		font-weight: bold;
		border-bottom: 2px solid rgba(0, 0, 0, 0.15);
	}
	
	.plutoui-sidebar.aside.hide .open-sidebar, .plutoui-sidebar.aside:not(.hide) .closed-sidebar, .plutoui-sidebar:not(.aside) .closed-sidebar {
		display: none;
	}

	.sidebar-toggle {
		cursor: pointer;
	}
	
</style>
<script>
	let listener = event => {
		if (event.target.classList.contains("sidebar-toggle")) {
			document.querySelectorAll('.plutoui-sidebar').forEach(function(el) {
				el.classList.toggle("hide");
			});
		}
	}
	document.addEventListener('click', listener);
	invalidation.then(() => {
		document.removeEventListener('click', listener);
	})
</script>
"""