#!/usr/bin/env python3
import os
import subprocess
import sys

# Set up clean compilation environment
os.environ['CC'] = '/usr/bin/gcc'
os.environ['CXX'] = '/usr/bin/g++'

# Unset Intel compiler variables that may interfere
for var in ['ICC_PATH', 'INTEL_COMPILER_HOME', 'INTEL_LICENSE_FILE']:
    if var in os.environ:
        del os.environ[var]

# Set MuJoCo environment
os.environ['MUJOCO_PY_MUJOCO_PATH'] = '/home/hk-project-robolear/ys1087/.mujoco/mujoco210'
os.environ['LD_LIBRARY_PATH'] = os.environ.get('LD_LIBRARY_PATH', '') + ':/home/hk-project-robolear/ys1087/.mujoco/mujoco210/bin:/usr/lib/nvidia'
os.environ['MUJOCO_GL'] = 'egl'

# Clear any cached compilation
import shutil
cache_dir = '.venv/lib/python3.10/site-packages/mujoco_py/generated'
if os.path.exists(cache_dir):
    shutil.rmtree(cache_dir)
    print(f"Cleared cache: {cache_dir}")

print("Environment configured for GCC compilation")
print(f"CC: {os.environ.get('CC')}")
print(f"CXX: {os.environ.get('CXX')}")

# Test import
try:
    import mujoco_py
    print("SUCCESS: mujoco_py imported successfully!")
except Exception as e:
    print(f"FAILED: {e}")
    sys.exit(1)