import matplotlib.pyplot as plt
import matplotlib.colors as mcolors
from matplotlib.patches import Patch
import numpy as np
from scipy.interpolate import PchipInterpolator

# 1. PASTE YOUR DATA HERE Fathead_minnow_toxicity (Separated by tabs, including headers)
datos_crudos = """Bin Center	% Frequency
0	0.339702760084926
0.05	11.0403397027601
0.1	10.5307855626327
0.15	6.83651804670913
0.2	6.41188959660297
0.25	5.60509554140127
0.3	4.37367303609342
0.35	4.24628450106157
0.4	3.82165605095541
0.45	3.86411889596603
0.5	4.24628450106157
0.55	3.60934182590234
0.6	3.90658174097665
0.65	3.73673036093418
0.7	4.24628450106157
0.75	4.54352441613588
0.8	5.18046709129512
0.85	5.01061571125265
0.9	4.45859872611465
0.95	3.86411889596603
1	0.127388535031847"""

# 2. AUTOMATIC PROCESSING
lineas = datos_crudos.strip().split('\n')[1:] 
bins_array = []
freq_array = []

for linea in lineas:
    b, f = linea.strip().split() 
    bins_array.append(float(b))
    freq_array.append(float(f))

bins = np.array(bins_array)
freq = np.array(freq_array)

# Statistical Calculations
mean_val = np.average(bins, weights=freq)

# PCHIP interpolation
interpolator = PchipInterpolator(bins, freq)
x_fit = np.linspace(min(bins), max(bins), 500)
y_fit = interpolator(x_fit)
y_fit = np.clip(y_fit, 0, None)

# Color Function
def get_colors(b_array):
    return ['#008000' if b < 0.4 else '#FFD700' if b <= 0.7 else '#B22222' for b in b_array]

colors_hex = get_colors(bins)
face_colors = [mcolors.to_rgba(c, alpha=0.60) for c in colors_hex]
edge_colors = [mcolors.to_rgba(c, alpha=0.90) for c in colors_hex]

# 3. CREATION OF THE GRAPH
plt.figure(figsize=(7, 6))

plt.bar(bins, freq, width=0.04, color=face_colors, edgecolor=edge_colors, linewidth=1.5, zorder=2)

# 4. LABELS AND TITLES
plt.xlabel('Fathead Minnow Toxicity Probability', fontsize=12)
plt.ylabel('% Frequency', fontsize=12)
plt.title('Environmental Hazard: Aquatic Toxicity (Fathead Minnow)', fontsize=14)

legend_elements = [
    Patch(facecolor=mcolors.to_rgba('#008000', 0.6), edgecolor='#008000', label='Low Toxicity / Inert (< 0.4)'),
    Patch(facecolor=mcolors.to_rgba('#FFD700', 0.6), edgecolor='#FFD700', label='Moderate Hazard (0.4 - 0.7)'),
    Patch(facecolor=mcolors.to_rgba('#B22222', 0.6), edgecolor='#B22222', label='High Lethality Risk (> 0.7)')
]
plt.legend(handles=legend_elements, loc='upper right', framealpha=0.95, fontsize=10)

plt.grid(axis='y', linestyle=':', alpha=0.7, zorder=0)
plt.xlim(-0.05, 1.05)
plt.ylim(0, max(freq) * 1.2) 
plt.tight_layout()

plt.show()