from scripts.forest_plotter import ForestPlotter

plotter = ForestPlotter()

# Generate forest plot
plot = plotter.create_plot(
    studies=["Study A", "Study B", "Study C"],
    effect_sizes=[1.2, 0.8, 1.5],
    ci_lower=[0.9, 0.5, 1.1],
    ci_upper=[1.5, 1.1, 1.9],
    overall_effect=1.15
)

fig = plotter.plot(
    data=studies_df,
    effect_col="HR",
    ci_lower_col="CI_lower",
    ci_upper_col="CI_upper",
    study_col="study_name"
)

fig = plotter.plot_with_stats(
    data,
    heterogeneity_stats={
        "I2": 45.2,
        "p_value": 0.03,
        "Q_statistic": 18.4
    },
    overall_effect={
        "estimate": 1.15,
        "ci": [0.98, 1.35],
        "p_value": 0.08
    }
)

fig = plotter.subgroup_plot(
    data,
    subgroup_col="treatment_type",
    subgroups=["Surgery", "Radiation", "Combined"]
)

fig = plotter.plot(
    data,
    style="publication",
    journal="lancet",  # or "nejm", "jama", "nature"
    color_scheme="monochrome",
    show_weights=True
)

# From CSV data
python scripts/forest_plotter.py \
  --input meta_analysis_data.csv \
  --effect-col OR \
  --output forest_plot.pdf

# With custom styling
python scripts/forest_plotter.py \
  --input data.csv \
  --style lancet \
  --width 8 --height 10