from openpyxl import Workbook

# 从图片人工识别到的口味与尼古丁浓度（mg）。
# 目前未录入库存数量；如需要，可再新增一列并补充。

inventory_rows = []

# 6mg
for flavor in [
    "Apple Mint","Blueberry","Citrus","Coffee","Cool Mint",
    "Mango","Peach","Spearmint","Watermelon","Wintergreen",
    "Cola","Sour Blue Razz","Ice Wintergreen"
]:
    inventory_rows.append((flavor, 6))

# 9mg
for flavor in [
    "Cool Mint","Wintergreen","Citrus","Peppermint","Spearmint",
    "Smooth","Apple Mint","Blueberry","Grape","Mango",
    "Black Cherry","Double Mint","Menthol","Ice Wintergreen",
    "Max Polar Mint","Cinnamon","Energy Drink","Orange","Peach",
    "Watermelon","Coffee","Dragon Fruit","Cola","Sour Blue Razz",
    "Strawberry Kiwi","Pineapple"
]:
    inventory_rows.append((flavor, 9))

# 12mg
for flavor in [
    "Apple Mint","Black Cherry","Blueberry","Coffee","Cool Mint",
    "Mango","Peach","Pink Lemonade","Spearmint","Wintergreen",
    "Ice Wintergreen","Grape","Watermelon"
]:
    inventory_rows.append((flavor, 12))

# 15mg
for flavor in [
    "Cool Mint","Wintergreen","Citrus","Peppermint","Spearmint",
    "Chill","Coffee","Blueberry","Dragon Fruit","Mango",
    "Apple Mint","Citrus Chill","Grape","Double Mint","Ice Wintergreen",
    "Strawberry Watermelon","Max Polar Mint","Vanilla","Pink Lemonade"
]:
    inventory_rows.append((flavor, 15))


def build_excel(output_path: str):
    wb = Workbook()

    # Sheet1: 图片转Excel（口味-浓度）
    ws1 = wb.active
    ws1.title = "图片转Excel"
    ws1.append(["Flavor", "Strength(mg)"])
    for flavor, mg in inventory_rows:
        ws1.append([flavor, mg])

    # Sheet2: 口味列表（去重）
    ws2 = wb.create_sheet(title="口味列表")
    ws2.append(["Flavor"])
    unique_flavors = sorted({f for f, _ in inventory_rows})
    for f in unique_flavors:
        ws2.append([f])

    wb.save(output_path)


if __name__ == "__main__":
    build_excel("flavors_inventory.xlsx")