import os
import zipfile
import shutil
from pathlib import Path

# Optional: Gradio import if using UI
import gradio as gr

# === CONFIGURATION ===
BUNDLE_ZIP = "CodexReflex_HF_SpaceBundle_20260104.zip"
EXTRACT_DIR = "CodexReflexSystem"
BOOT_SCRIPT = "main.py"  # Replace if different

# === EXTRACT SYSTEM BUNDLE ===
def extract_system_bundle():
    if not Path(EXTRACT_DIR).exists():
        with zipfile.ZipFile(BUNDLE_ZIP, 'r') as zip_ref:
            zip_ref.extractall(EXTRACT_DIR)
        print(f"[+] Extracted system into: {EXTRACT_DIR}")
    else:
        print(f"[i] System already extracted.")

# === RUN THE SYSTEM ===
def run_system():
    os.chdir(EXTRACT_DIR)
    os.system(f"python {BOOT_SCRIPT}")

# === ENTRY POINT ===
if __name__ == "__main__":
    extract_system_bundle()
    run_system()