#!/usr/bin/env bash
# LFM3-Audio-2B lokal ausprobieren – Linux
# Automatisch erzeugt von Tokenstrom (tokenstrom.com).
# Aufruf: bash try_linux.sh
set -e
cd "$(dirname "$0")"
echo "== LFM3-Audio-2B =="
if command -v nvidia-smi >/dev/null 2>&1; then nvidia-smi --query-gpu=name,memory.total --format=csv,noheader;
else echo "No NVIDIA GPU found – running on CPU (slow)"; fi

if true; then
  echo "== Ollama =="
  if ! command -v ollama >/dev/null 2>&1; then
    command -v zstd >/dev/null 2>&1 || sudo apt-get install -y zstd 2>/dev/null || sudo dnf install -y zstd 2>/dev/null || true
    curl -fsSL https://ollama.com/install.sh | sh
  fi
  curl -s http://localhost:11434/api/tags >/dev/null 2>&1 || (ollama serve >/dev/null 2>&1 &)
  sleep 3
  MODEL="${MODEL:-}"
  MODEL="${MODEL:-lfm3-audio}"
  echo "Downloading $MODEL (the first time this takes a few minutes, depending on size) ..."
  ollama pull "$MODEL"
  echo ""
  echo "Done! Now type your question and press Enter. Quit with /bye"
  exec ollama run "$MODEL"
fi