diff --git a/setup.sh b/setup.sh
index 1ba256e..d3586fe 100644
--- a/setup.sh
+++ b/setup.sh
@@ -1,60 +1,74 @@
 #!/bin/bash
 
+# ==========================================
+# Farbdefinitionen
+# ==========================================
+C_DEF=$'\e[0m'          # Default / Reset
+C_BOLD=$'\e[1m'         # Fett
+C_CYAN=$'\e[1;36m'      # Cyan
+C_BLUE=$'\e[1;34m'      # Blau
+C_GREEN=$'\e[1;32m'     # Grün
+C_YELLOW=$'\e[1;33m'    # Gelb
+C_RED=$'\e[1;31m'       # Rot
+
 REPO_URL="https://git.pi-farm.de/pi-farm/PiDoBot.git"
 
-echo "=========================================="
-echo ">>> J.A.R.V.I.S. - Professional Setup  <<<"
-echo "=========================================="
+echo -e "${C_CYAN}${C_BOLD}==========================================${C_DEF}"
+echo -e "${C_CYAN}${C_BOLD}>>> J.A.R.V.I.S. - Professional Setup  <<<${C_DEF}"
+echo -e "${C_CYAN}${C_BOLD}==========================================${C_DEF}"
 
 # 0. Installationsverzeichnis abfragen
-# WICHTIG: Nutze einen Unterordner, nicht direkt /home/pi/
-read -p "Installationsverzeichnis (Standard: /home/pi/jarvis-ai): " input_dir </dev/tty
+read -p "${C_CYAN}Installationsverzeichnis (Standard: ${C_YELLOW}/home/pi/jarvis-ai${C_CYAN}): ${C_DEF}" input_dir </dev/tty
 INSTALL_DIR=${input_dir:-/home/pi/jarvis-ai}
 
-# Pfad normalisieren (entfernt doppelte Slashes und Tilde)
+# Pfad normalisieren
 INSTALL_DIR=$(realpath -m "$INSTALL_DIR")
 
-echo -e "\nZielverzeichnis: $INSTALL_DIR"
+echo -e "\n${C_GREEN}Zielverzeichnis: ${C_BOLD}$INSTALL_DIR${C_DEF}"
 mkdir -p "$INSTALL_DIR"
 cd "$INSTALL_DIR" || exit
 
 # 1. System-Abhängigkeiten
-echo -e "\n--- 1. System-Pakete..."
+echo -e "\n${C_BLUE}${C_BOLD}--- 1. Prüfe und installiere System-Pakete...${C_DEF}"
 sudo apt-get update
 sudo apt-get install -y git wget sshpass python3-pip python3-venv iproute2
 
 # 2. Repository klonen
-echo -e "\n--- 2. Quellcode..."
+echo -e "\n${C_BLUE}${C_BOLD}--- 2. Hole Quellcode...${C_DEF}"
 if [ ! -d ".git" ]; then
     git clone --branch dev --single-branch "$REPO_URL" .
 else
-    echo "Repo bereits vorhanden, aktualisiere..."
+    echo -e "${C_YELLOW}Repo bereits vorhanden, aktualisiere...${C_DEF}"
     git pull
 fi
 
-# 3. Virtual Environment (Hier lag der Fehler!)
-echo -e "\n--- 3. Python-Umgebung..."
+# 3. Virtual Environment
+echo -e "\n${C_BLUE}${C_BOLD}--- 3. Richte Python-Umgebung ein...${C_DEF}"
 if [ ! -d "venv" ]; then
     python3 -m venv venv
 fi
-# Nutze den direkten Pfad zum Pip im Venv (sicherer als 'source')
+# Nutze den direkten Pfad zum Pip im Venv
 ./venv/bin/pip install --upgrade pip
 if [ -f "source/requirements.txt" ]; then
     ./venv/bin/pip install -r source/requirements.txt
 else
-    echo "❌ FEHLER: requirements.txt nicht gefunden!"
+    echo -e "${C_RED}❌ FEHLER: requirements.txt nicht gefunden!${C_DEF}"
     exit 1
 fi
 
 # 4. SSH-Key
-echo -e "\n--- 4. SSH-Key..."
-[ ! -f "$HOME/.ssh/id_rsa" ] && ssh-keygen -t rsa -N "" -f "$HOME/.ssh/id_rsa"
+echo -e "\n${C_BLUE}${C_BOLD}--- 4. Prüfe SSH-Schlüssel...${C_DEF}"
+if [ ! -f "$HOME/.ssh/id_rsa" ]; then
+    ssh-keygen -t rsa -N "" -f "$HOME/.ssh/id_rsa"
+    echo -e "${C_GREEN}✅ SSH-Key generiert.${C_DEF}"
+else
+    echo -e "${C_GREEN}✅ SSH-Key existiert bereits.${C_DEF}"
+fi
 
-# 5. Static Dateien (Der Grund für deine 404s)
-echo -e "\n--- 5. Static-Bibliotheken..."
+# 5. Static Dateien
+echo -e "\n${C_BLUE}${C_BOLD}--- 5. Lade Frontend-Bibliotheken...${C_DEF}"
 mkdir -p source/static
 cd source/static
-# Lade Dateien nur, wenn sie fehlen
 wget -nc -q https://cdn.jsdelivr.net/npm/gridstack@7.2.3/dist/gridstack.min.css
 wget -nc -q https://cdn.jsdelivr.net/npm/gridstack@7.2.3/dist/gridstack-all.js
 wget -nc -q https://cdn.jsdelivr.net/npm/xterm@5.1.0/css/xterm.css
@@ -63,8 +77,8 @@ wget -nc -q https://cdn.jsdelivr.net/npm/xterm-addon-fit@0.7.0/lib/xterm-addon-f
 wget -nc -q https://cdn.jsdelivr.net/npm/marked/marked.min.js
 cd "$INSTALL_DIR"
 
-# 6. .env Setup (wie besprochen)
-echo -e "\n--- 6. Konfiguration..."
+# 6. .env Setup
+echo -e "\n${C_BLUE}${C_BOLD}--- 6. Konfiguration (.env)...${C_DEF}"
 mkdir -p config
 ENV_FILE="config/.env"
 
@@ -82,13 +96,11 @@ tg_id=""
 
 # Falls die Datei existiert, alte Werte laden
 if [ -f "$ENV_FILE" ]; then
-    echo "Bestehende .env gefunden. Lade aktuelle Werte..."
-    # Lese die Datei zeilenweise ein, um die Variablen sicher zu parsen
+    echo -e "${C_YELLOW}Bestehende .env gefunden. Lade aktuelle Werte...${C_DEF}"
     set -a
     source "$ENV_FILE"
     set +a
     
-    # Überschreibe die Standardwerte mit den gefundenen Werten
     web_user="${WEB_USER_NAME:-$web_user}"
     ai_prov="${AI_PROVIDER:-$ai_prov}"
     google_key="${GOOGLE_API_KEY:-$google_key}"
@@ -101,32 +113,35 @@ if [ -f "$ENV_FILE" ]; then
     tg_id="${ALLOWED_TELEGRAM_USER_ID:-$tg_id}"
 fi
 
-echo "Bitte gib die neuen Werte ein. (Drücke ENTER, um den Wert in [Klammern] beizubehalten):"
+echo -e "\nBitte gib die neuen Werte ein. (Drücke ENTER, um den Wert in Klammern beizubehalten):"
 
-read -p "Dein Web-Benutzername [$web_user]: " input_web_user
+# Farbige Prompts mit </dev/tty
+read -p "${C_CYAN}Dein Web-Benutzername [${C_YELLOW}$web_user${C_CYAN}]: ${C_DEF}" input_web_user </dev/tty
 web_user=${input_web_user:-$web_user}
 
-read -p "Primäre KI (google, openai, nvidia, ollama) [$ai_prov]: " input_ai_prov
+read -p "${C_CYAN}Primäre KI (google, openai, nvidia, ollama) [${C_YELLOW}$ai_prov${C_CYAN}]: ${C_DEF}" input_ai_prov </dev/tty
 ai_prov=${input_ai_prov:-$ai_prov}
 
-# Keys maskieren für die Anzeige, damit sie nicht komplett im Terminal stehen
 disp_gkey=$( [ -n "$google_key" ] && echo "${google_key:0:5}***" || echo "" )
-read -p "Google Gemini API Key [$disp_gkey]: " input_google_key
+read -p "${C_CYAN}Google Gemini API Key [${C_YELLOW}$disp_gkey${C_CYAN}]: ${C_DEF}" input_google_key </dev/tty
 google_key=${input_google_key:-$google_key}
 
 disp_okey=$( [ -n "$openai_key" ] && echo "${openai_key:0:5}***" || echo "" )
-read -p "OpenAI API Key [$disp_okey]: " input_openai_key
+read -p "${C_CYAN}OpenAI API Key [${C_YELLOW}$disp_okey${C_CYAN}]: ${C_DEF}" input_openai_key </dev/tty
 openai_key=${input_openai_key:-$openai_key}
 
 disp_nkey=$( [ -n "$nvidia_key" ] && echo "${nvidia_key:0:5}***" || echo "" )
-read -p "NVIDIA API Key [$disp_nkey]: " input_nvidia_key
+read -p "${C_CYAN}NVIDIA API Key [${C_YELLOW}$disp_nkey${C_CYAN}]: ${C_DEF}" input_nvidia_key </dev/tty
 nvidia_key=${input_nvidia_key:-$nvidia_key}
 
+read -p "${C_CYAN}Ollama Base URL [${C_YELLOW}$ollama_url${C_CYAN}]: ${C_DEF}" input_ollama_url </dev/tty
+ollama_url=${input_ollama_url:-$ollama_url}
+
 disp_tgtoken=$( [ -n "$tg_token" ] && echo "${tg_token:0:8}***" || echo "" )
-read -p "Telegram Bot Token [$disp_tgtoken]: " input_tg_token
+read -p "${C_CYAN}Telegram Bot Token [${C_YELLOW}$disp_tgtoken${C_CYAN}]: ${C_DEF}" input_tg_token </dev/tty
 tg_token=${input_tg_token:-$tg_token}
 
-read -p "Erlaubte Telegram User ID [$tg_id]: " input_tg_id
+read -p "${C_CYAN}Erlaubte Telegram User ID [${C_YELLOW}$tg_id${C_CYAN}]: ${C_DEF}" input_tg_id </dev/tty
 tg_id=${input_tg_id:-$tg_id}
 
 # Neue .env schreiben
@@ -142,21 +157,35 @@ OPENAI_MODEL=$openai_mod
 TELEGRAM_BOT_TOKEN=$tg_token
 ALLOWED_TELEGRAM_USER_ID=$tg_id
 EOF
-echo "Konfiguration erfolgreich gespeichert."
+echo -e "${C_GREEN}✅ Konfiguration erfolgreich gespeichert.${C_DEF}"
 
 # 7. Port Check
-echo -e "\n--- 7. Prüfe Port 8000..."
+echo -e "\n${C_BLUE}${C_BOLD}--- 7. Prüfe Port 8000...${C_DEF}"
 if ss -tuln | grep -q ":8000 "; then
-    echo "⚠️  WARNUNG: Port 8000 wird bereits von einem anderen Prozess verwendet!"
+    echo -e "${C_RED}⚠️  WARNUNG: Port 8000 wird bereits von einem anderen Prozess verwendet!${C_DEF}"
     echo "J.A.R.V.I.S. wird möglicherweise nicht starten können. Bitte prüfe dies nach dem Setup."
-    read -p "Drücke ENTER, um trotzdem fortzufahren..."
+    read -p "Drücke ENTER, um trotzdem fortzufahren..." </dev/tty
 else
-    echo "✅ Port 8000 ist frei."
+    echo -e "${C_GREEN}✅ Port 8000 ist frei.${C_DEF}"
 fi
 
-# 8. Systemd Service einrichten
-echo -e "\n--- 8. Systemdienst (Autostart) einrichten..."
-read -p "Möchtest du J.A.R.V.I.S. als Hintergrunddienst installieren? (j/N): " setup_service
+# 8. Firewall Check (NEU!)
+echo -e "\n${C_BLUE}${C_BOLD}--- 8. Prüfe Firewall-Einstellungen...${C_DEF}"
+if command -v ufw >/dev/null 2>&1; then
+    if sudo ufw status | grep -qw "active"; then
+        echo -e "${C_YELLOW}UFW Firewall ist aktiv. Öffne Port 8000...${C_DEF}"
+        sudo ufw allow 8000/tcp
+        echo -e "${C_GREEN}✅ Port 8000 (TCP) in UFW freigegeben.${C_DEF}"
+    else
+        echo -e "${C_GREEN}✅ UFW ist installiert, aber inaktiv. Keine Blockade zu erwarten.${C_DEF}"
+    fi
+else
+    echo -e "${C_GREEN}✅ Keine UFW Firewall gefunden. Überspringe Firewall-Setup.${C_DEF}"
+fi
+
+# 9. Systemd Service einrichten
+echo -e "\n${C_BLUE}${C_BOLD}--- 9. Systemdienst (Autostart) einrichten...${C_DEF}"
+read -p "${C_CYAN}Möchtest du J.A.R.V.I.S. als Hintergrunddienst installieren? (j/N): ${C_DEF}" setup_service </dev/tty
 
 if [[ "$setup_service" =~ ^[jJ]$ ]]; then
     SERVICE_FILE="/etc/systemd/system/jarvis.service"
@@ -184,18 +213,18 @@ EOF"
     sudo systemctl enable jarvis.service
     sudo systemctl restart jarvis.service
     
-    echo -e "\n=========================================="
-    echo "✅ Setup komplett! J.A.R.V.I.S. läuft als Dienst."
-    echo "Verzeichnis:   $INSTALL_DIR"
-    echo "Web-Interface: http://<deine-ip>:8000"
-    echo "Log-Ausgabe:   sudo journalctl -u jarvis -f"
-    echo "=========================================="
+    echo -e "\n${C_GREEN}${C_BOLD}==========================================${C_DEF}"
+    echo -e "${C_GREEN}✅ Setup komplett! J.A.R.V.I.S. läuft als Dienst.${C_DEF}"
+    echo -e "Verzeichnis:   ${C_BOLD}$INSTALL_DIR${C_DEF}"
+    echo -e "Web-Interface: ${C_CYAN}http://<deine-ip>:8000${C_DEF}"
+    echo -e "Log-Ausgabe:   ${C_YELLOW}sudo journalctl -u jarvis -f${C_DEF}"
+    echo -e "${C_GREEN}${C_BOLD}==========================================${C_DEF}"
 else
-    echo -e "\n=========================================="
-    echo "✅ Setup komplett! Du kannst J.A.R.V.I.S. nun manuell starten:"
+    echo -e "\n${C_GREEN}${C_BOLD}==========================================${C_DEF}"
+    echo -e "${C_GREEN}✅ Setup komplett! Du kannst J.A.R.V.I.S. nun manuell starten:${C_DEF}"
     echo "cd $INSTALL_DIR"
     echo "source venv/bin/activate"
     echo "export PYTHONPATH=\$PYTHONPATH:\$(pwd)/source"
     echo "python3 -m uvicorn source.main:app --host 0.0.0.0 --port 8000"
-    echo "=========================================="
+    echo -e "${C_GREEN}${C_BOLD}==========================================${C_DEF}"
 fi
\ No newline at end of file