#!/bin/bash

# Activate virtual environment
source /venv/bin/activate

# Start Ollama server on port 7860
echo "Starting Ollama server on port 7860..."
ollama serve &

# Give Ollama some time to initialize
sleep 2

# Pull models
IFS=',' read -ra MODELS <<< "$model"
for m in "${MODELS[@]}"; do
    echo "Pulling $m"
    ollama pull "$m"
    sleep 5
done

# # Start the Streamlit app on port 8501
# echo "Starting Streamlit app on port 8501..."
# streamlit run /app.py --server.port 8501 --server.address 0.0.0.0 &

# Keep the container running
wait