|
|
| def prompt_weights(default_weights: dict) -> dict:
|
| """
|
| Interactively collect weights from the user (in %) and validate that they sum to 100.
|
| Press Enter to keep defaults.
|
| """
|
| print("\\n⚖️ Set Weights (press Enter to keep default values)")
|
| new_w = {}
|
| for key, val in default_weights.items():
|
| try:
|
| raw = input(f" {key.replace('_WEIGHT','').title()} (%), default {val}: ").strip()
|
| new_w[key] = val if raw == "" else int(raw)
|
| except ValueError:
|
| print(f" Invalid input for {key}, keeping default {val}.")
|
| new_w[key] = val
|
|
|
| total = sum(new_w.values())
|
| if total != 100:
|
| print(f" Weights sum to {total}, normalizing to 100 proportionally.")
|
| factor = 100.0 / total if total else 0
|
| for k in new_w:
|
| new_w[k] = int(round(new_w[k] * factor))
|
|
|
| drift = 100 - sum(new_w.values())
|
| if drift != 0:
|
| largest_key = max(new_w.keys(), key=new_w.get)
|
| new_w[largest_key] += drift
|
|
|
| print(" Final Weights:", new_w)
|
| return new_w |
