#!/bin/bash
source "$HOME/.config/hypr/scripts/ai-stuff/ollama-common.sh"

FIRST_PROMPT=$(wofi --dmenu --prompt "Ask AI")
[ -z "$FIRST_PROMPT" ] && exit 0

kitty --title "AI Chat" bash -lc "
source \"$HOME/.config/hypr/scripts/ai-stuff/ollama-common.sh\"

clear
echo '╔══════════════════════════════════════╗'
echo '║              AI Chat                 ║'
echo '║      exit = beenden | clear = leeren ║'
echo '╚══════════════════════════════════════╝'
echo
echo 'Modell: $OLLAMA_MODEL'
echo
echo 'Du > $FIRST_PROMPT'
echo
echo 'AI >'
ask_ollama \"$FIRST_PROMPT\"
echo

while true; do
    read -rp 'Du > ' PROMPT

    [ -z \"\$PROMPT\" ] && continue

    case \"\$PROMPT\" in
        exit)
            break
            ;;
        clear)
            clear
            echo '╔══════════════════════════════════════╗'
            echo '║              AI Chat                 ║'
            echo '║      exit = beenden | clear = leeren ║'
            echo '╚══════════════════════════════════════╝'
            echo
            echo 'Modell: $OLLAMA_MODEL'
            echo
            continue
            ;;
    esac

    echo
    echo 'AI >'
    ask_ollama \"\$PROMPT\"
    echo
done
"