Allow editing Cadence abilities and Refinement recipes in CRUD GUI

Author: DerekGooding

modules/contentManager/app.py

@@ -47,18 +47,19 @@
         st.sidebar.warning(f"**{w['type']} [{w['name']}]**: {w['msg']}")
 
 
-def edit_list(data_list, key_prefix):
+def edit_list(data_list, key_prefix, item_pool=None):
+    if item_pool is None:
+        item_pool = [it["Name"] for i in [manager.unified_data["items"]] for it in i]
+    
     to_delete = None
     for i, entry in enumerate(data_list):
         col1, col2, col3 = st.columns([3, 2, 1])
         with col1:
-            # Safely find index of current item
-            item_names = [it["Name"] for i in [manager.unified_data["items"]] for it in i]
             current_index = 0
-            if entry["Item"] in item_names:
-                current_index = item_names.index(entry["Item"])
+            if entry["Item"] in item_pool:
+                current_index = item_pool.index(entry["Item"])
 
-            new_name = st.selectbox(f"Item {i}", item_names, index=current_index, key=f"{key_prefix}_name_{i}")
+            new_name = st.selectbox(f"Item {i}", item_pool, index=current_index, key=f"{key_prefix}_name_{i}")
             entry["Item"] = new_name
 
         with col2:
@@ -73,41 +74,75 @@ def edit_list(data_list, key_prefix):
         st.rerun()
 
     if st.button("➕ Add Entry", key=f"{key_prefix}_add"):
-        data_list.append({"Item": manager.unified_data["items"][0]["Name"], "Quantity": 1})
+        data_list.append({"Item": item_pool[0], "Quantity": 1})
         st.rerun()
 
-def edit_dict_list(data_dict, key_prefix, label_name="Stat"):
+def edit_recipes(recipes, key_prefix):
     to_delete = None
-    keys = list(data_dict.keys())
-    for i, key in enumerate(keys):
-        col1, col2, col3 = st.columns([3, 2, 1])
-        with col1:
-            stat_names = [s["Name"] for s in manager.unified_data["stats"]]
-            current_index = 0
-            if key in stat_names:
-                current_index = stat_names.index(key)
-            
-            new_key = st.selectbox(f"{label_name} {i}", stat_names, index=current_index, key=f"{key_prefix}_key_{i}")
-            if new_key != key:
-                data_dict[new_key] = data_dict.pop(key)
-                st.rerun()
+    item_names = [i["Name"] for i in manager.unified_data["items"]]
+    for i, recipe in enumerate(recipes):
+        with st.container(border=True):
+            col1, col2, col3, col4, col5 = st.columns([2, 1, 2, 1, 0.5])
+            with col1:
+                recipe["InputItem"] = st.selectbox("In", item_names, index=item_names.index(recipe["InputItem"]) if recipe["InputItem"] in item_names else 0, key=f"{key_prefix}_in_{i}")
+            with col2:
+                recipe["InputQuantity"] = st.number_input("Qty", value=recipe["InputQuantity"], min_value=1, key=f"{key_prefix}_in_q_{i}")
+            with col3:
+                recipe["OutputItem"] = st.selectbox("Out", item_names, index=item_names.index(recipe["OutputItem"]) if recipe["OutputItem"] in item_names else 0, key=f"{key_prefix}_out_{i}")
+            with col4:
+                recipe["OutputQuantity"] = st.number_input("Qty", value=recipe["OutputQuantity"], min_value=1, key=f"{key_prefix}_out_q_{i}")
+            with col5:
+                if st.button("🗑️", key=f"{key_prefix}_del_{i}"):
+                    to_delete = i
+    
+    if to_delete is not None:
+        recipes.pop(to_delete)
+        st.rerun()
+    
+    if st.button("➕ Add Recipe", key=f"{key_prefix}_add"):
+        recipes.append({"InputItem": item_names[0], "InputQuantity": 1, "OutputItem": item_names[0], "OutputQuantity": 1})
+        st.rerun()
+
+def edit_cadence_abilities(abilities_list, key_prefix):
+    to_delete = None
+    all_ability_names = [a["Name"] for a in manager.unified_data["abilities"]]
+    stat_names = [s["Name"] for s in manager.unified_data["stats"]]
+
+    for i, ab_entry in enumerate(abilities_list):
+        with st.container(border=True):
+            col1, col2, col3 = st.columns([3, 2, 1])
+            with col1:
+                # The nested structure is {"Ability": {"Name": "...", "Description": "..."}, "Requirements": [...], "PrimaryStat": "..."}
+                current_ab_name = ab_entry["Ability"]["Name"]
+                new_ab_name = st.selectbox(f"Ability {i}", all_ability_names, index=all_ability_names.index(current_ab_name) if current_ab_name in all_ability_names else 0, key=f"{key_prefix}_ab_{i}")
 
-        with col2:
-            data_dict[new_key] = st.number_input(f"Value {i}", value=data_dict[new_key], key=f"{key_prefix}_val_{i}")
+                if new_ab_name != current_ab_name:
+                    source_ab = next(a for a in manager.unified_data["abilities"] if a["Name"] == new_ab_name)
+                    ab_entry["Ability"]["Name"] = source_ab["Name"]
+                    ab_entry["Ability"]["Description"] = source_ab["Description"]
 
-        with col3:
-            if st.button("🗑️", key=f"{key_prefix}_del_{i}"):
-                to_delete = key
-                
+            with col2:
+                ab_entry["PrimaryStat"] = st.selectbox(f"Stat {i}", stat_names, index=stat_names.index(ab_entry["PrimaryStat"]) if ab_entry["PrimaryStat"] in stat_names else 0, key=f"{key_prefix}_stat_{i}")
+            
+            with col3:
+                if st.button("🗑️ Remove Ability", key=f"{key_prefix}_del_{i}"):
+                    to_delete = i
+            
+            st.write("**Requirements**")
+            edit_list(ab_entry["Requirements"], f"{key_prefix}_req_{i}")
+
     if to_delete is not None:
-        del data_dict[to_delete]
+        abilities_list.pop(to_delete)
+        st.rerun()
+
+    if st.button("➕ Add Ability to Cadence", key=f"{key_prefix}_add"):
+        source_ab = manager.unified_data["abilities"][0]
+        abilities_list.append({
+            "Ability": {"Name": source_ab["Name"], "Description": source_ab["Description"]},
+            "Requirements": [],
+            "PrimaryStat": "Magic"
+        })
         st.rerun()
-    
-    if st.button(f"➕ Add {label_name}", key=f"{key_prefix}_add"):
-        available = [s["Name"] for s in manager.unified_data["stats"] if s["Name"] not in data_dict]
-        if available:
-            data_dict[available[0]] = 1
-            st.rerun()
 
 # --- Page Rendering ---
 
@@ -133,6 +168,11 @@ def edit_dict_list(data_dict, key_prefix, label_name="Stat"):
             new_item.update({"DurationSeconds": 10, "Type": "Single", "Requirements": [], "Rewards": [], "PrimaryStat": "Vitality", "RequiredStats": {}, "StatRewards": {}, "Requires": [], "UnlocksCadences": []})
         elif page == "Items":
             new_item.update({"ItemType": "Material", "Augments": []})
+        elif page == "Cadences":
+            new_item.update({"Abilities": []})
+        elif page == "Refinements":
+            new_item = {"Name": "New Refinement", "PrimaryStat": "Strength", "Recipes": []}
+        
         manager.unified_data[page.lower()].append(new_item)
         st.rerun()
 
@@ -142,9 +182,13 @@ def edit_dict_list(data_dict, key_prefix, label_name="Stat"):
     with st.form(f"edit_{page}_{selected_name}"):
         st.subheader(f"Editing: {selected_name}")
 
-        name = st.text_input("Name", item["Name"])
-        description = st.text_area("Description", item.get("Description", ""))
-        
+        if page != "Refinements": # Refinements use 'Ability' as name, handled in data_io
+            name = st.text_input("Name", item["Name"])
+            description = st.text_area("Description", item.get("Description", ""))
+        else:
+            name = st.selectbox("Ability Base", [a["Name"] for a in manager.unified_data["abilities"]], index=[a["Name"] for a in manager.unified_data["abilities"]].index(item["Name"]) if item["Name"] in [a["Name"] for a in manager.unified_data["abilities"]] else 0)
+            description = "" # Not used for refinements
+            
         if page == "Quests":
             col1, col2, col3 = st.columns(3)
             with col1:
@@ -158,25 +202,26 @@ def edit_dict_list(data_dict, key_prefix, label_name="Stat"):
             st.subheader("Unlocks & Dependencies")
             requires = st.multiselect("Prerequisite Quests", [q["Name"] for q in manager.unified_data["quests"] if q["Name"] != name], default=item["Requires"])
             unlocks_c = st.multiselect("Unlocks Cadences", [c["Name"] for c in manager.unified_data["cadences"]], default=item["UnlocksCadences"])
-            
-            # Complex nested data (Requirements/Rewards) are hard in standard Streamlit forms 
-            # so we'll just show them as JSON for now or implement outside form
-            st.info("Requirements and Rewards are managed below the form for technical reasons.")
 
-        if page == "Items":
+        elif page == "Items":
             i_type = st.selectbox("Type", ["Material", "Currency", "Consumable", "Spell", "KeyItem"], index=["Material", "Currency", "Consumable", "Spell", "KeyItem"].index(item["ItemType"]))
+        
+        elif page == "Refinements":
+             stat = st.selectbox("Primary Stat", [s["Name"] for s in manager.unified_data["stats"]], index=[s["Name"] for s in manager.unified_data["stats"]].index(item["PrimaryStat"]))
 
         if st.form_submit_button("Update Basic Info"):
             item["Name"] = name
-            item["Description"] = description
+            if "Description" in item: item["Description"] = description
             if page == "Quests":
                 item["DurationSeconds"] = duration
                 item["Type"] = q_type
                 item["PrimaryStat"] = stat
                 item["Requires"] = requires
                 item["UnlocksCadences"] = unlocks_c
-            if page == "Items":
+            elif page == "Items":
                 item["ItemType"] = i_type
+            elif page == "Refinements":
+                item["PrimaryStat"] = stat
             st.success("Updated basic info in memory.")
             st.rerun()
 
@@ -193,8 +238,20 @@ def edit_dict_list(data_dict, key_prefix, label_name="Stat"):
 
         st.subheader("📈 Stat Rewards")
         edit_dict_list(item["StatRewards"], "rew_stat")
+    
+    elif page == "Cadences":
+        st.subheader("⚡ Abilities")
+        edit_cadence_abilities(item["Abilities"], "cad_ab")
+
+    elif page == "Refinements":
+        st.subheader("🧪 Recipes")
+        edit_recipes(item["Recipes"], "ref_rec")
+
+    elif page == "Abilities":
+        st.info("Abilities themselves are simple Name/Description. Use the Cadence page to assign them and define requirements.")
 
     if st.button("🔥 Delete Entity", type="secondary"):
         manager.unified_data[page.lower()].remove(item)
         st.warning(f"Deleted {selected_name} from memory.")
         st.rerun()
+

modules/contentManager/data_io.py

@@ -19,7 +19,8 @@ def __init__(self, data_dir=DATA_DIR):
             "cadences": [],
             "locations": [],
             "refinements": [],
-            "stats": []
+            "stats": [],
+            "abilities": []
         }
         self.load_all()
 
@@ -36,7 +37,8 @@ def load_all(self):
             "items.json", "stat_augments.json", "quests.json", 
             "quest_details.json", "quest_unlocks.json", 
             "quest_cadence_unlocks.json", "cadences.json", 
-            "locations.json", "refinements.json", "stats.json"
+            "locations.json", "refinements.json", "stats.json",
+            "cadence_abilities.json"
         ]
         for f in files:
             self.raw_data[f] = self._load_json(f)
@@ -47,6 +49,10 @@ def load_all(self):
         self._unify_locations()
         self._unify_refinements()
         self._unify_stats()
+        self._unify_abilities()
+
+    def _unify_abilities(self):
+        self.unified_data["abilities"] = [a.copy() for a in self.raw_data["cadence_abilities.json"]]
 
     def _unify_items(self):
         augments = { a["Item"]: a["Augments"] for a in self.raw_data["stat_augments.json"] }