add master talents

app.py

@@ -183,6 +183,34 @@ def return_hero_stat(df0, hero_name, lb_choice="None", costume_choice="None"):
                     f'{prefix} Health':'health'})
     return df_ret
 
+    
+def return_talent_choice(key="default_key"):
+    talent_list = ['None', 'Sword+20', 'Shield+20', 'Health+20', 'Sword+25', 'Shield+25', 'Health+20 Shield+5','Sword+20 Shield+5','Shield+20 Sword+5']
+    
+    talent_choice = st.selectbox(label='Approx. Talent (stat are all cruded approximated. Visit heroplan.io for exact calculation.) :', 
+                                 options=talent_list, 
+                                 index=0, 
+                                 key=key)
+    
+    talent_tp, talent_attack, talent_defense, talent_health = 0, 0, 0, 0
+    if talent_choice == 'Sword+20':
+        talent_tp, talent_attack, talent_defense, talent_health = 100, 150, 50, 100
+    elif talent_choice == 'Shield+20':
+        talent_tp, talent_attack, talent_defense, talent_health = 100, 50, 150, 100
+    elif talent_choice == 'Health+20':
+        talent_tp, talent_attack, talent_defense, talent_health = 100, 75, 75, 200
+    if talent_choice == 'Sword+25':
+        talent_tp, talent_attack, talent_defense, talent_health = 125, 300, 50, 200
+    elif talent_choice == 'Shield+25':
+        talent_tp, talent_attack, talent_defense, talent_health = 125, 50, 330, 200
+    elif talent_choice == 'Health+20 Shield+5': # choose shield-path on 5 master nodes
+        talent_tp, talent_attack, talent_defense, talent_health = 125, 75, 255, 300
+    elif talent_choice == 'Sword+20 Shield+5':
+        talent_tp, talent_attack, talent_defense, talent_health = 125, 150, 230, 200
+    elif talent_choice == 'Shield+20 Sword+5':
+        talent_tp, talent_attack, talent_defense, talent_health = 125, 200, 150, 200
+    return talent_tp, talent_attack, talent_defense, talent_health
+
 #########################################
 ## Load the main file (TODO: caching)=
 st.set_page_config(layout="wide")
@@ -327,8 +355,15 @@ elif genre == "Team Simulation":
         costume_choice = st.selectbox(label='Costume:', options=costume_list, index=0, key=key+"_costume")
         lb_list = ['None', 'LB1', 'LB2']
         lb_choice = st.selectbox(label='Limit Break:', options=lb_list, index=0, key=key+"_lb")
-    
+
+        talent_tp, talent_attack, talent_defense, talent_health = return_talent_choice(key=key+"_talent")
+        
         df_ret = return_hero_stat(df_extra, name_choice, lb_choice=lb_choice, costume_choice=costume_choice)
+        df_ret.power.values[0] += talent_tp
+        df_ret.attack.values[0] += talent_attack
+        df_ret.defense.values[0] += talent_defense
+        df_ret.health.values[0] += talent_health
+        
         return df_ret
 
     def write_short_description(df_hero):
@@ -393,24 +428,10 @@ else:
     
     lb_list = ['None', 'LB1', 'LB2']
     lb_choice = st.selectbox(label='Limit Break:', options=lb_list, index=0)
-
-    talent_list = ['None', 'Sword', 'Shield', 'Health']
-    # talent_choice = st.radio(  
-    #     "Approx. Talent",
-    #     talent_list,
-    #     captions = ["Default", "Talents stat are all cruded approximated. Visit heroplan.io for exact calculation."]
-    # )
-    talent_choice = st.selectbox(label='Approx. Talent (stat are all cruded approximated. Visit heroplan.io for exact calculation.) :', options=talent_list, index=0)
-    talent_tp, talent_attack, talent_defense, talent_health = 0, 0, 0, 0
-    if talent_choice == 'Sword':
-        talent_tp, talent_attack, talent_defense, talent_health = 100, 150, 50, 100
-    elif talent_choice == 'Shield':
-        talent_tp, talent_attack, talent_defense, talent_health = 100, 50, 150, 100
-    elif talent_choice == 'Health':
-        talent_tp, talent_attack, talent_defense, talent_health = 100, 75, 75, 200
         
-    
+    talent_tp, talent_attack, talent_defense, talent_health = return_talent_choice()
     df_ret = return_hero_stat(df_extra, name_choice, lb_choice=lb_choice, costume_choice=costume_choice)
+    
     df_ret.power.values[0] += talent_tp
     df_ret.attack.values[0] += talent_attack
     df_ret.defense.values[0] += talent_defense