Update app.py plants.csv requirements.txt

app.py

@@ -3,30 +3,59 @@ import pandas as pd
 import pulp
 import numpy as np
 
-LANGUAGE = "en"
-
 # Define the quality tiers and names for the plants
 PLANTS_TIERS = {
-    "radiant": "金品质",
-    "flourishing": "紫品质",
-    "hardy": "蓝品质",
-    "feeble": "白品质",
+    "radiant": "RADIANT",
+    "flourishing": "FLOURISHING",
+    "hardy": "HARDY",
+    "feeble": "FEEBLE",
+}
+
+PLANTS_LABLES = {
+    "fanged_geranium": "Fanged Geranium",
+    "gillyweed": "Gillyweed",
+    "rose": "Rose",
+    "puffapod": "Puffapod",
+    "nifflers_fancy": "Niffler's Fancy",
+    "ladys_mantle": "Lady's Mantle",
+    "mandrake": "Mandrake",
+    "chinese_chomping_cabbage": "Chinese Chomping Cabbage",
+    "peony": "Peony",
+    "begonia": "Begonia",
+    "mayflower": "Mayflower",
+    "hydrangea": "Hydrangea",
+    "daffodil": "Daffodil",
 }
 
-PLANTS_I18N = {
-    "fanged_geranium": "毒牙天竺葵",
-    "gillyweed": "鳃囊草",
-    "rose": "玫瑰",
-    "puffapod": "泡泡豆荚",
-    "nifflers_fancy": "嗅幻草",
-    "ladys_mantle": "羽衣草",
-    "mandrake": "曼德拉草",
-    "chinese_chomping_cabbage": "中国咬人甘蓝",
-    "peony": "牡丹",
-    "begonia": "秋海棠",
-    "mayflower": "五月花",
-    "hydrangea": "绣球",
-    "daffodil": "水仙",
+INTERFACE_TEXTS = {
+    "cn": {
+        "gold_label": "葭碧の金币预算：",
+        "strategies_label": "请选择凑单策略：",
+        "clear_btn_label": "❌清除",
+        "calculate_btn_label": "🛠计算",
+        "output_label": "计算结果：",
+        "strategy_options": [
+            ("最小化售出株数(优先出售高价植物)", "MaximizeStock"),
+            ("最大化售出株数(优先出售低价植物)", "MinimizeStock"),
+        ],
+    },
+    "en": {
+        "gold_label": "Gabby's Gold Budget:",
+        "strategies_label": "Select a strategy:",
+        "clear_btn_label": "❌Clear",
+        "calculate_btn_label": "🛠Calculate",
+        "output_label": "Output:",
+        "strategy_options": [
+            (
+                "Minimize the number of plants sold (prioritize high-priced plants)",
+                "MaximizeStock",
+            ),
+            (
+                "Maximize the number of plants sold (prioritize low-priced plants)",
+                "MinimizeStock",
+            ),
+        ],
+    },
 }
 
 # Import and process plant data
@@ -50,7 +79,7 @@ def calculator(*args):
     and inventory constraints.
 
     Args:
-        *args (tuple): A tuple where the first element is Gabby's gold budget,
+        *args (tuple): A tuple where the first element is Gabby's gold budget, the selected strategy for selling plants,
                        and the subsequent elements represent the stock levels of
                        each plant type.
 
@@ -59,15 +88,25 @@ def calculator(*args):
              the total gold earned, and the remaining inventory.
     """
     budget = args[0]  # 葭碧预算
-    stocks = np.array(args[1:])  # 植物库存
+    strategy = args[1]  # 出售策略
+    stocks = np.array([x if x else 0 for x in args[2:]])  # 植物库存
 
     # 植物名称列表
-    plants_en = [f"{row['tier']}_{row['species']}" for index, row in df.iterrows()]
+    plants_names = [
+        f"{PLANTS_TIERS[row['tier']]} {PLANTS_LABLES[row['species']]}"
+        for index, row in df.iterrows()
+    ]
 
     gold = np.array(df["gold"])  # 植物单价
 
-    # 创建问题实例，目标是最大化存货数量
-    prob = pulp.LpProblem("Maximize_Sale", pulp.LpMaximize)
+    if sum([gold[i] * stocks[i] for i in range(len(stocks))]) < budget:
+        return "No solution found.\nThe total value of the plants is lower than the budget."
+
+    # 创建问题实例，根据出售策略设定优化目标
+    if strategy == "MaximizeStock":
+        prob = pulp.LpProblem("Maximize_Stock", pulp.LpMinimize)
+    else:
+        prob = pulp.LpProblem("Maximize_Stock", pulp.LpMaximize)
 
     # 决策变量，售出每种植物的件数，必须是整数
     x = pulp.LpVariable.dicts("x", range(len(stocks)), lowBound=0, cat="Integer")
@@ -76,8 +115,8 @@ def calculator(*args):
     for i in range(len(stocks)):
         x[i].upBound = stocks[i]
 
-    # 目标函数：最大化总的存货数量
-    prob += pulp.lpSum([stocks[i] - x[i] for i in range(len(stocks))])
+    # 目标函数：最大化 / 最小化 售出株数
+    prob += pulp.lpSum([x[i] for i in range(len(stocks))])
 
     # 约束条件：每类产品售出数量乘以单价之和等于总价格
     prob += pulp.lpSum([gold[i] * x[i] for i in range(len(stocks))]) == budget
@@ -94,50 +133,91 @@ def calculator(*args):
         for i, v in x.items():
             if v.varValue:
                 if v.varValue > 0:
-                    solution.append(f"{plants_en[i]}: {int(v.varValue)}\n")
+                    solution.append(f"{plants_names[i]}: {int(v.varValue)}\n")
                     sold += int(v.varValue) * gold[i]
 
-        return f"Optimal solution found:\n\n{''.join(solution)}\nTotal price: {sold}\nStock: {pulp.value(prob.objective)}"
+        return f"Optimal solution found:\n\n{''.join(solution)}\nTotal Price: {sold}\nCount: {int(pulp.value(prob.objective))}"
+
     else:
-        return "No solution found."
+        return "No solution found.\n"
 
 
-# Create a Gradio interface with a column layout
 with gr.Blocks() as demo:
+    gr.Markdown(
+        """<center><font size=8>HP-Magic-Awakened Herbologist Toolkit👾</center>"""
+    )
+
+    # Language selection dropdown
+    # language = gr.Radio(
+    #     choices=["cn", "en"],
+    #     value="cn",
+    #     label="Language",
+    #     info="Select the interface language:",
+    #     interactive=True,
+    # )
+
+    # Create a Gradio interface with a column layout
     with gr.Column():
         # Add a row for the budget input
-        with gr.Row():
-            budget = gr.Number(
-                label="Gold Budget", value=0, minimum=0, maximum=15000, step=100
+        budget = gr.Number(
+            label="Gold",
+            info="Gabby's Gold Budget:",  # "葭碧の金币预算：",
+            value=0,
+            minimum=0,
+            maximum=15000,
+            step=100,
+        )
+
+    # Add a radio selection for the strategy
+    selected_strategy = gr.Radio(
+        [
+            (
+                "Minimize the number of plants sold (prioritize high-priced plants)",
+                "MaximizeStock",
+            ),
+            (
+                "Maximize the number of plants sold (prioritize low-priced plants)",
+                "MinimizeStock",
+            ),
+        ],
+        value="MaximizeStock",
+        label="Strategies",
+        info="Select a strategy:",
+    )
+
+    # Add a row for the plant inventory inputs
+    with gr.Row():
+        inventory = {}
+        for _, row in df.iterrows():
+            inventory[f"{row['species']}_{row['tier']}"] = gr.Number(
+                label=f"{PLANTS_LABLES[row['species']]} - {PLANTS_TIERS[row['tier']]}",
+                value=0,
+                minimum=0,
+                maximum=500,
+                step=10,
+                visible=row["gold"] != 0,
             )
 
-        # Add a row for the plant inventory inputs
-        with gr.Row():
-            inventory = {}
-            for index, row in df.iterrows():
-                inventory[f"{row['tier']}_{row['species']}"] = gr.Number(
-                    label=f"{row['tier']}_{row['species']}",
-                    value=0,
-                    minimum=0,
-                    maximum=500,
-                    step=10,
-                )
+    # Add a row for the Clear and Calculate buttons
+    with gr.Row():
+        clear_btn = gr.ClearButton(list(inventory.values()), size="sm", value="❌Clear")
 
         # Add a button to trigger the calculation
-        submit_btn = gr.Button(value="Calculate")
+        submit_btn = gr.Button(value="🛠Calculate")
 
-        # Add a row for the result textbox
-        with gr.Row():
-            result = gr.Textbox()
+    # Add a row for the result textbox
+    with gr.Row():
+        result = gr.Textbox(label="Output")
 
     # Set up the button click event to call the calculator function
     submit_btn.click(
         calculator,
-        inputs=[budget] + list(inventory.values()),
+        inputs=[budget] + [selected_strategy] + list(inventory.values()),
         outputs=[result],
         api_name=False,
     )
 
+
 # Launch the Gradio application
-if __name__ == "__main__":
-    demo.launch(show_api=False)
+demo.queue(api_open=False)
+demo.launch(max_threads=5, share=False)

plants.csv

@@ -10,11 +10,11 @@ gold,gems,species,tier,amount
 42,0,rose,radiant,0
 34,0,rose,flourishing,0
 30,0,rose,hardy,0
-,0,rose,feeble,0
+1,0,rose,feeble,0
 85,0,puffapod,radiant,0
 62,0,puffapod,flourishing,0
 55,0,puffapod,hardy,0
-,0,puffapod,feeble,0
+1,0,puffapod,feeble,0
 0,5,nifflers_fancy,radiant,0
 260,0,nifflers_fancy,flourishing,0
 234,0,nifflers_fancy,hardy,0
@@ -22,32 +22,32 @@ gold,gems,species,tier,amount
 105,0,ladys_mantle,radiant,0
 96,0,ladys_mantle,flourishing,0
 76,0,ladys_mantle,hardy,0
-,0,ladys_mantle,feeble,0
+1,0,ladys_mantle,feeble,0
 24,0,mandrake,radiant,0
 22,0,mandrake,flourishing,0
 15,0,mandrake,hardy,0
-,0,mandrake,feeble,0
+1,0,mandrake,feeble,0
 57,0,chinese_chomping_cabbage,radiant,0
 40,0,chinese_chomping_cabbage,flourishing,0
 36,0,chinese_chomping_cabbage,hardy,0
-,0,chinese_chomping_cabbage,feeble,0
+1,0,chinese_chomping_cabbage,feeble,0
 0,2,peony,radiant,0
 58,0,peony,flourishing,0
 45,0,peony,hardy,0
 1,0,peony,feeble,0
 72,0,begonia,radiant,0
 58,0,begonia,flourishing,0
-,0,begonia,hardy,0
-,0,begonia,feeble,0
+45,0,begonia,hardy,0
+1,0,begonia,feeble,0
 186,0,mayflower,radiant,0
-,0,mayflower,flourishing,0
-,0,mayflower,hardy,0
-,0,mayflower,feeble,0
+150,0,mayflower,flourishing,0
+117,0,mayflower,hardy,0
+1,0,mayflower,feeble,0
 215,0,hydrangea,radiant,0
-,0,hydrangea,flourishing,0
-,0,hydrangea,hardy,0
-,0,hydrangea,feeble,0
+150,0,hydrangea,flourishing,0
+135,0,hydrangea,hardy,0
+1,0,hydrangea,feeble,0
 0,2,daffodil,radiant,0
-,0,daffodil,flourishing,0
-,0,daffodil,hardy,0
-,0,daffodil,feeble,0
+50,0,daffodil,flourishing,0
+45,0,daffodil,hardy,0
+1,0,daffodil,feeble,0