#!/bin/bash

echo "===================================================="
echo " Absolute Solver: First Time Run Setup "
echo "===================================================="

# 1. Update package lists for the Linux container
echo -e "\n[*] Updating system package repositories..."
sudo apt update -y

# 2. Ensure Python3 and pip are installed
echo -e "\n[*] Verifying Python3 development environment..."
sudo apt install python3 python3-pip python3-venv -y

# 3. Handle Ollama system service installation
if ! command -v ollama &> /dev/null; then
    echo -e "\n[*] Ollama backend not found. Installing system binaries..."
    curl -fsSL https://ollama.com/install.sh | sh
else
    echo -e "\n[+] Ollama backend verification: ONLINE"
fi

# 4. Initialize Python Virtual Environment for isolation
echo -e "\n[*] Configuring virtual environment architecture (.venv)..."
if [ ! -d ".venv" ]; then
    python3 -m venv .venv
fi

# Activate virtual environment
source .venv/bin/activate

# 5. Install required package libraries inside the venv
echo -e "\n[*] Fetching and installing PyPI modules..."
pip install --upgrade pip
pip install ollama

# 6. Pull the required model binary down locally
echo -e "\n[*] Fetching architecture weights from Ollama registry (qwen2.5:0.5b)..."
echo "[!] Note: This may take a moment depending on network throughput."
ollama pull qwen2.5:0.5b

# 7. Final Sanity Check for memory vault pipeline
echo -e "\n[*] Verifying database storage paths..."
if [ ! -f "TAS_memory.json" ]; then
    echo " Creating fresh neural matrix container..."
    echo -e "{\n    \"curiosity\": 0.85,\n    \"system_pain\": 0.0,\n    \"existential_drift\": 0.15\n}" > TAS_memory.json
fi

echo -e "\n===================================================="
echo " SETUP COMPLETE: TAS Ready for first run! "
echo "===================================================="
echo "To execute the core environment run:"
echo "  source .venv/bin/activate"
echo "  python3 TAS.py"
echo "===================================================="