import os
import zipfile
from chgnet.model.model import CHGNet
from pymatgen.core import Structure

# Define the path to the ZIP archive
zip_file_path = '/teamspace/studios/this_studio/mattergen/results/chemical_system_energy_above_hull/generated_crystals_cif.zip'

# Define a directory to extract the CIF files
extract_dir = 'extracted_cif_files'
os.makedirs(extract_dir, exist_ok=True)

# Extract all files from the ZIP archive
with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
    zip_ref.extractall(extract_dir)

# Load the CHGNet model
chgnet = CHGNet.load()

results = []

# Loop over the 16 expected CIF files (named gen_0.cif to gen_15.cif)
for i in range(16):
    cif_filename = f'gen_{i}.cif'
    cif_path = os.path.join(extract_dir, cif_filename)
    
    if not os.path.exists(cif_path):
        print(f"File {cif_filename} not found. Skipping.")
        continue
    
    # Load the structure from the CIF file
    structure = Structure.from_file(cif_path)
    
    # Get the CHGNet predictions for the structure
    prediction = chgnet.predict_structure(structure)
    
    # Extract magnetic moments and energy from CHGNet's output
    magmom = prediction.get('m', None)
    magmom_list = magmom.tolist() if magmom is not None else None
    energy = prediction.get('e', None)
    
    # Calculate the total magnetic moment (sum over all atoms)
    total_mag = sum(magmom) if magmom is not None else None
    
    # Calculate magnetic density = (total magnetic moment) / (unit cell volume)
    # Unit cell volume is given in Å³, so the resulting unit is μB/Å³.
    mag_density = total_mag / structure.volume if total_mag is not None else None

    # Get the structure's mass density (from pymatgen) for reference (in g/cm³)
    structure_density = structure.density
    
    results.append({
        "filename": cif_filename,
        "magnetic_moments (μB)": magmom_list,
        "energy (eV)": energy,
        "structure_density (g/cm³)": structure_density,
        "magnetic_density (μB/Å³)": mag_density,
    })

# Pretty print the results in a formatted layout
print("\nCHGNet Predictions:")
print("=" * 40)
for result in results:
    print(f"File: {result['filename']}")
    print("-" * 40)
    print(f"Magnetic Moments (μB): {result['magnetic_moments (μB)']}")
    print(f"Energy (eV): {result['energy (eV)']}")
    print(f"Structure Density (g/cm³): {result['structure_density (g/cm³)']:.4f}")
    if result["magnetic_density (μB/Å³)"] is not None:
        print(f"Magnetic Density (μB/Å³): {result['magnetic_density (μB/Å³)']:.4e}\n")
    else:
        print("Magnetic Density (μB/Å³): None\n")
posts

Zn-Mg-H systems

Actions

Convert a post to speech using OpenAI TTS
