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

# Create pressure grid
nlev = 100
p_bot = np.log10(100.0)
p_top = np.log10(1e-6)
p_lev = np.logspace(p_bot, p_top, nlev) * bar  # 1e-6 to 100 bar in dyne/cm²

# Example parameters
params = {"T_iso": 400.0}
T_lev, T_lay = isothermal(p_lev, params)

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