#!/bin/bash
# QUICK START SCRIPT
# Run this after uploading your PDFs

echo "=================================="
echo "MEDICAL TRANSCRIPT ANALYSIS"
echo "Quick Setup & Execution"
echo "=================================="
echo ""

# Step 1: Check if Ollama is installed
echo "Step 1: Checking Ollama installation..."
if command -v ollama &> /dev/null; then
    echo "✓ Ollama is installed"
else
    echo "✗ Ollama not found. Installing..."
    curl -fsSL https://ollama.com/install.sh | sh
fi
echo ""

# Step 2: Check if model is downloaded
echo "Step 2: Checking for Llama 3.1 model..."
if ollama list | grep -q "llama3.1:8b"; then
    echo "✓ Llama 3.1 8B is downloaded"
else
    echo "Downloading Llama 3.1 8B (4.7GB)..."
    echo "This will take 5-15 minutes depending on your internet speed"
    ollama pull llama3.1:8b
fi
echo ""

# Step 3: Install Python packages
echo "Step 3: Installing Python packages..."
pip install --break-system-packages PyPDF2 pandas openpyxl requests numpy 2>/dev/null || \
pip install PyPDF2 pandas openpyxl requests numpy
echo ""

# Step 4: Start Ollama server in background
echo "Step 4: Starting Ollama server..."
if pgrep -x "ollama" > /dev/null; then
    echo "✓ Ollama is already running"
else
    echo "Starting Ollama in background..."
    nohup ollama serve > /tmp/ollama.log 2>&1 &
    sleep 3
    echo "✓ Ollama started"
fi
echo ""

# Step 5: Test connection
echo "Step 5: Testing Ollama connection..."
if curl -s http://localhost:11434/api/tags > /dev/null; then
    echo "✓ Ollama is responding"
else
    echo "✗ Ollama is not responding. Check /tmp/ollama.log"
    exit 1
fi
echo ""

# Step 6: Check for PDFs
echo "Step 6: Checking for PDF files..."
PDF_COUNT=$(find /mnt/user-data/uploads -name "*.pdf" 2>/dev/null | wc -l)
if [ "$PDF_COUNT" -gt 0 ]; then
    echo "✓ Found $PDF_COUNT PDF files"
else
    echo "✗ No PDF files found in /mnt/user-data/uploads"
    echo "Please upload your PDF files first"
    exit 1
fi
echo ""

# Step 7: Run analysis
echo "=================================="
echo "READY TO START ANALYSIS"
echo "=================================="
echo ""
echo "Total PDFs to process: $PDF_COUNT"
echo "Estimated time: $(($PDF_COUNT * 2)) minutes"
echo ""
read -p "Press Enter to start analysis (or Ctrl+C to cancel)..."
echo ""

# Run the analysis
python3 /home/claude/run_analysis.py

echo ""
echo "=================================="
echo "ANALYSIS COMPLETE!"
echo "=================================="
echo ""
echo "Output files saved to: /mnt/user-data/outputs/"
echo ""
echo "Main deliverable: ANALYSIS_TABULATION_*.xlsx"
echo ""