import os
import yaml
import geopandas as gpd

def load_config():
    config_path = os.path.join(os.path.dirname(__file__), 'config.yaml')
    with open(config_path, 'r') as f:
        return yaml.safe_load(f)

def main():
    config = load_config()
    feature_stack_shp = os.path.join(os.path.dirname(__file__), config['paths']['rock_glacier_boundary_shp'])
    output_dir = os.path.join(os.path.dirname(__file__), config['paths']['output_dir'], 'individual_polygons')
    
    os.makedirs(output_dir, exist_ok=True)
    
    print(f"Loading feature stack shapefile: {feature_stack_shp}")
    try:
        gdf = gpd.read_file(feature_stack_shp)
    except Exception as e:
        print(f"Error loading {feature_stack_shp}: {e}")
        print("Please ensure you have updated the 'feature_stack_shp' path in config.yaml.")
        return

    print(f"Separating {len(gdf)} feature stack polygons...")
    
    for idx, row in gdf.iterrows():
        single_gdf = gpd.GeoDataFrame([row], crs=gdf.crs)
        out_json = os.path.join(output_dir, f"polygon_{idx:04d}.geojson")
        single_gdf.to_file(out_json, driver="GeoJSON")
        
    print(f"Done! Saved {len(gdf)} GeoJSON files to {output_dir}")

if __name__ == "__main__":
    main()