Transcribe audio with AI locally

# Create main directory and subdirectories
$rootDir = "D:\whisper-ai"
$modelsDir = "$rootDir\models"
$testdataDir = "$rootDir\testdata"

# Create directories if they don't exist
New-Item -ItemType Directory -Force -Path $rootDir
New-Item -ItemType Directory -Force -Path $modelsDir
New-Item -ItemType Directory -Force -Path $testdataDir

# Function to download model if it doesn't exist
function Download-WhisperModel {
    param (
        [string]$modelName,
        [string]$outputPath
    )
    
    if (-not (Test-Path $outputPath)) {
        Write-Host "Downloading $modelName model..."
        $url = "https://huggingface.co/ggerganov/whisper.cpp/resolve/main/ggml-$modelName.bin"
        Invoke-WebRequest -Uri $url -OutFile $outputPath
        Write-Host "$modelName model downloaded successfully"
    } else {
        Write-Host "$modelName model already exists"
    }
}

# Download medium and large models
Download-WhisperModel -modelName "medium" -outputPath "$modelsDir\ggml-medium.bin"
Download-WhisperModel -modelName "large-v3" -outputPath "$modelsDir\ggml-large-v3.bin"

# Function to transcribe audio
function Transcribe-Audio {
    param (
        [string]$modelPath,
        [string]$audioPath
    )
    
    # Remove existing container if it exists
    docker rm -f whisper-transcribe 2>&1 | Out-Null
    
    # Run docker with output handling
    $result = docker run `
        --name whisper-transcribe `
        -v ${modelsDir}:/app/models `
        -v ${testdataDir}:/app/testdata `
        ghcr.io/appleboy/go-whisper:latest `
        --model $modelPath `
        --audio-path $audioPath 2>&1

    # Output the result as regular output
    $result | ForEach-Object {
        Write-Host $_
    }
}

Write-Host "Setup complete! To transcribe an audio file, follow these steps:"
Write-Host "1. Place your audio file in: $testdataDir"
Write-Host "2. Run one of these commands (replace 'your-audio.wav' with your actual filename):"
Write-Host "`nFor medium model:"
Write-Host "Transcribe-Audio -modelPath '/app/models/ggml-medium.bin' -audioPath '/app/testdata/your-audio.wav'"
Write-Host "`nFor large model:"
Write-Host "Transcribe-Audio -modelPath '/app/models/ggml-large-v3.bin' -audioPath '/app/testdata/your-audio.wav'"

# Example usage (commented out)
# Transcribe-Audio -modelPath "/app/models/ggml-medium.bin" -audioPath "/app/testdata/your-audio.wav"


Transcribe-Audio -modelPath "/app/models/ggml-large-v3.bin" -audioPath "/app/testdata/your-audio.m4a"