main.py

"""
Simulation model skeleton code for you to modify when 
following the tutorial.

Author: www.youtube.com/@TheOpenScienceNerd
Year: 2025
"""

import pandas as pd

from newsvendor import (
    Experiment,
    multiple_replications,
    create_profit_histogram,
    DEFAULT_PATH_DAY_TYPE,
    DEFAULT_PATH_DEMAND,
)

def main():
    purchase_cost = 33.0
    sale_price = 50.0
    salvage_price = 5.0

    # Operational parameters
    order_qty = 70
    n_days = 20
    n_reps = 1000

    df_daytype = pd.read_csv(DEFAULT_PATH_DAY_TYPE)
    df_demand = pd.read_csv(DEFAULT_PATH_DEMAND)
       
    # Run the simulation with the progress callback
    exp = Experiment(
        order_quantity=order_qty,
        periods=n_days,
        purchase_cost=purchase_cost,
        sale_price=sale_price,
        salvage_price=salvage_price,
        # strip formatting from distribution dataframes
        day_type=df_daytype,
        demand_prob=df_demand,
    )


    results = multiple_replications(
        exp, n_reps
    )

    # Display summary statistics formatted as currency
    print(
        results[["profit"]].describe().iloc[1:].map(lambda x: f"£{x/100:.2f}")
    )

    # fig = create_profit_histogram(results.profit / 100)
    # fig.show

if __name__ == "__main__":
    main()