#!/usr/bin/env bash
set -Eeuo pipefail

cd ~/CyberBrain26
source venv-webapp/bin/activate

set -a
. ./.env
set +a

echo ""
echo "=== CyberBrain26 Simple Chat ==="
echo "Model: gemini-pro"
echo "Ctrl+C = ukončit"
echo ""

while true; do
    read -p "🧑 " user_input
    
    if [[ "$user_input" == "quit" || "$user_input" == "exit" || "$user_input" == "q" ]]; then
        echo "👋 Ukončuji."
        break
    fi
    
    if [[ -z "$user_input" ]]; then
        continue
    fi
    
    echo -n "🤖 "
    
    curl -s http://127.0.0.1:4000/v1/chat/completions \
        -H "Authorization: Bearer ${LITELLM_MASTER_KEY}" \
        -H "Content-Type: application/json" \
        -d "{\"model\":\"gemini-pro\",\"messages\":[{\"role\":\"user\",\"content\":\"$user_input\"}]}" | \
        python3 -c "import sys, json; data=json.load(sys.stdin); print(data['choices'][0]['message']['content'])"
    
    echo ""
done