#!/bin/bash
# Prerad Setup and Launch Script
# This script creates a virtual environment and starts the Streamlit application

set -e

PROJECT_PATH="/run/media/dheena/Leave you files/prerad"
VENV_PATH="${PROJECT_PATH}/venv"

echo "🎯 Prerad Environment Setup and Launch"
echo "======================================"
echo ""

# Run Python setup script
echo "Running setup script..."
python3 "${PROJECT_PATH}/setup_venv.py"

echo ""
echo "======================================"
echo "🎉 Environment is ready!"
echo "======================================"
echo ""

# Activate virtual environment
echo "Activating virtual environment..."
source "${VENV_PATH}/bin/activate"

echo ""
echo "✅ Virtual environment activated"
echo ""

# Offer options to start services
echo "Choose what to start:"
echo "1) Streamlit application (localhost:8501)"
echo "2) Jupyter Notebook (localhost:8888)"
echo "3) Both (using Docker Compose with sudo)"
echo "4) Exit"
echo ""
read -p "Enter your choice (1-4): " choice

case $choice in
    1)
        echo ""
        echo "🚀 Starting Streamlit application..."
        streamlit run "${PROJECT_PATH}/containers/streamlit/app.py"
        ;;
    2)
        echo ""
        echo "📓 Starting Jupyter Notebook..."
        jupyter notebook --notebook-dir="${PROJECT_PATH}/volumes/notebooks"
        ;;
    3)
        echo ""
        echo "🐳 Starting services with Docker Compose..."
        cd "${PROJECT_PATH}"
        sudo docker compose up -d jupyter streamlit
        echo ""
        echo "✅ Services started!"
        echo "   Jupyter: http://localhost:8888"
        echo "   Streamlit: http://localhost:8501"
        ;;
    4)
        echo "Exiting..."
        exit 0
        ;;
    *)
        echo "Invalid choice. Exiting..."
        exit 1
        ;;
esac