import numpy as np
import matplotlib.pyplot as plt
from exo_skryer.vert_Tp import Milne
from exo_skryer.data_constants import bar

# Create pressure grid
nlev = 100
p_bot = np.log10(1000.0)
p_top = np.log10(1e-5)
p_lev = np.logspace(p_bot, p_top, nlev) * bar

# Example parameters for a brown dwarf
params = {
    "T_int": 1000.0,
    "log_10_g": 4.5,
    "log_10_k_ir": -2.0
}
T_lev, T_lay = Milne(p_lev, params)

# Plot
fig, ax = plt.subplots(figsize=(10, 5))
ax.semilogy(T_lev, p_lev/bar, c='forestgreen')
ax.set_xlabel('Temperature [K]', fontsize=16)
ax.set_ylabel('pressure [bar]', fontsize=16)
ax.set_title('Milne T-p Profile', fontsize=14)
ax.tick_params(labelsize=14)
ax.invert_yaxis()
plt.tight_layout()