"""
Reset the UAV arm AtomizerField test study

Clears all optimization results and logs to start fresh.
Preserves the setup and model files.
"""

from pathlib import Path
import shutil

def reset_study():
    """Reset study to clean state."""

    study_dir = Path(__file__).parent

    # Remove results directory
    results_dir = study_dir / "2_results"
    if results_dir.exists():
        print(f"Removing {results_dir}...")
        shutil.rmtree(results_dir)

    # Remove training data export directory if exists
    training_data_dir = Path("atomizer_field_training_data/uav_arm_test")
    if training_data_dir.exists():
        print(f"Removing {training_data_dir}...")
        shutil.rmtree(training_data_dir)

    # Remove any Python cache
    pycache = study_dir / "__pycache__"
    if pycache.exists():
        print(f"Removing {pycache}...")
        shutil.rmtree(pycache)

    print("Study reset complete!")
    print("You can now run: python run_optimization.py")

if __name__ == "__main__":
    reset_study()