Add hyperparameter overriding functionality in _prepare_script of app.py to allow safer and more flexible adjustments of learning rate, network dimension, seed, max training epochs, and save frequency. This enhances configurability for training scripts.

app.py

@@ -345,6 +345,27 @@ def _prepare_script(
     if override_seed is not None:
         txt = re.sub(r"--seed\s+\d+", f"--seed {override_seed}", txt)
 
+    # Prefer overriding variable definitions at top of script (safer than CLI regex)
+    def _set_var(name: str, value: str) -> None:
+        nonlocal txt
+        pattern = rf"(?m)^\s*{name}\s*=.*$"
+        replacement = f'{name}="{value}"' if not str(value).isdigit() else f'{name}={value}'
+        if re.search(pattern, txt):
+            txt = re.sub(pattern, replacement, txt)
+        else:
+            txt = f"{replacement}\n" + txt
+
+    if override_learning_rate:
+        _set_var('LEARNING_RATE', override_learning_rate)
+    if override_network_dim is not None:
+        _set_var('NETWORK_DIM', str(override_network_dim))
+    if override_seed is not None:
+        _set_var('SEED', str(override_seed))
+    if override_max_epochs is not None and override_max_epochs > 0:
+        _set_var('MAX_TRAIN_EPOCHS', str(override_max_epochs))
+    if override_save_every is not None and override_save_every > 0:
+        _set_var('SAVE_EVERY_N_EPOCHS', str(override_save_every))
+
     # Write to a temp file alongside this repo for easier inspection
     run_dir = TRAINING_DIR / ".gradio_runs"
     run_dir.mkdir(parents=True, exist_ok=True)