#!/usr/bin/env bash
set -euo pipefail

ROOT=/root/projects/website-builder-backend-clean
cd "$ROOT"

python3 <<'PY'
from pathlib import Path

p = Path("backend/main.py")
lines = p.read_text().splitlines()

keep = "from backend.models.gateway import ModelGateway"
remove = "from backend.llm.gateway import ModelGateway"

new = []
kept = False

for line in lines:
    if line.strip() == remove:
        continue
    if line.strip() == keep:
        if kept:
            continue
        kept = True
    new.append(line)

p.write_text("\n".join(new) + "\n")
PY

echo
echo "===== Gateway imports ====="
grep -n "ModelGateway" backend/main.py

echo
python3 -m py_compile backend/main.py

echo
python3 <<'PY'
import backend.main

print("Gateway class :", type(backend.main.llm_gateway).__module__)
print("Gateway type  :", type(backend.main.llm_gateway).__name__)
print("Memory type   :", type(backend.main.memory).__name__)
print("Bus type      :", type(backend.main.bus).__name__)
print("Runtime type  :", type(backend.main.dolor3v).__name__)
PY

echo
echo "========================================"
echo "Gateway import cleaned"
echo "========================================"