#!/bin/bash
# Script to set up the local test environment

# Check if Python is installed
if ! command -v python3 &> /dev/null; then
    echo "Python 3 is not installed. Please install Python 3 and try again."
    exit 1
fi

# Create a virtual environment if it doesn't exist
if [ ! -d "venv" ]; then
    echo "Creating virtual environment..."
    python3 -m venv venv
    if [ $? -ne 0 ]; then
        echo "Failed to create virtual environment. Please check your Python installation."
        exit 1
    fi
    echo "Virtual environment created successfully."
else
    echo "Virtual environment already exists."
fi

# Activate the virtual environment
echo "Activating virtual environment..."
source venv/bin/activate
if [ $? -ne 0 ]; then
    echo "Failed to activate virtual environment."
    exit 1
fi

# Install dependencies
echo "Installing dependencies..."
pip install --upgrade pip
pip install requests python-dotenv argparse

# Create local_test.env file if it doesn't exist
if [ ! -f "local_test.env" ]; then
    echo "Creating local_test.env file from template..."
    cp local_test.env.example local_test.env
    echo "Please edit local_test.env with your server credentials."
else
    echo "local_test.env file already exists."
fi

# Make the test script executable
chmod +x test_remote_ollama.py

echo "Setup complete!"
echo "To run the test script:"
echo "1. Activate the virtual environment: source venv/bin/activate"
echo "2. Run the test script: python test_remote_ollama.py"
echo "3. To specify a different model or prompt: python test_remote_ollama.py --model llama3.3 --prompt \"Tell me about AI\""
echo "4. To specify a different Ollama URL: python test_remote_ollama.py --ollama-url http://your-server-ip:11434"