import os
import pathlib

def cleanup():
    """Delete legacy files from the repository root."""
    legacy_files = [
        "agent.py",
        "chunker.py",
        "cli.py",
        "config.py",
        "init_loader.py",
        "qdrant_store.py",
        "rag_tools.py",
        "tools.py",
        "vector_store.py",
        "__init__.py",
    ]
    for filename in legacy_files:
        try:
            os.remove(filename)
            print(f"Deleted {filename}")
        except FileNotFoundError:
            pass
        except Exception as e:
            print(f"Could not delete {filename}: {e}")
    # Optionally delete this cleanup script itself
    try:
        script_path = pathlib.Path(__file__).resolve()
        script_path.unlink()
        print("Deleted cleanup.py")
    except Exception as e:
        print(f"Could not delete cleanup.py: {e}")

if __name__ == "__main__":
    cleanup()