Update app.py

app.py

@@ -17,12 +17,15 @@ if not os.path.exists(unzip_dir):
     print("Extraction complete.")
 
 # Step 2: Parse folders to dynamically populate dropdowns
+
+model_root = os.path.join(unzip_dir, 'Models')  # Adjust this if your ZIP structure is different
+
 activations = []
-seeds_dict = dict()  # activation -> list of seeds
-neurons_dict = dict()  # (activation, seed) -> list of neuron counts
+seeds_dict = dict()
+neurons_dict = dict()
 
-for act in os.listdir(os.path.join(unzip_dir, 'Models')):
-    act_path = os.path.join(os.path.join(unzip_dir, 'Models'), act)
+for act in os.listdir(model_root):
+    act_path = os.path.join(model_root, act)
     if os.path.isdir(act_path) and not act.startswith("linear_models"):
         activations.append(act)
         seeds = []
@@ -40,6 +43,13 @@ for act in os.listdir(os.path.join(unzip_dir, 'Models')):
 
 activations = sorted(activations)
 
+def update_seeds(activation):
+    return {"choices": seeds_dict[activation], "value": seeds_dict[activation][0]}
+
+def update_neurons(activation, seed):
+    neurons = neurons_dict[(activation, seed)]
+    return {"choices": neurons, "value": neurons[0]}
+
 # Step 3: Load linear models
 # linear_rgb_path = os.path.join(unzip_dir, "linear_models/linear_rgb.joblib")
 # linear_grey_path = os.path.join(unzip_dir, "linear_models/linear_grey.joblib")