Update start.sh

start.sh

@@ -1,51 +1,66 @@
 #!/bin/bash
 
+# Stop the script if any command fails
+set -e
+
 # --- CONFIGURATION ---
-# Default to CODER mode if not set
 MODE="${MODEL_TYPE:-CODER}"
 
-# Define the models
-# Coder: 14B Q4_K_M
+echo "------------------------------------------------"
+echo "🛠️ Starting Space in [$MODE] mode"
+echo "------------------------------------------------"
+
+# Define Models
 CODER_REPO="Qwen/Qwen2.5-Coder-14B-Instruct-GGUF"
 CODER_FILE="qwen2.5-coder-14b-instruct-q4_k_m.gguf"
 
-# Vision: 7B Q4_K_M
 VISION_REPO="Qwen/Qwen2.5-VL-7B-Instruct-GGUF"
 VISION_FILE="qwen2.5-vl-7b-instruct-q4_k_m.gguf"
 
-# --- DOWNLOAD HELPER FUNCTION ---
-download_model() {
-    repo=$1
-    file=$2
-    path="/app/$file"
-    
-    if [ ! -f "$path" ]; then
-        echo "⬇️  Downloading $file..."
-        python3 -c "from huggingface_hub import hf_hub_download; hf_hub_download(repo_id='$repo', filename='$file', local_dir='/app')"
-        echo "✅ Download complete."
+# Function to download
+download_if_needed() {
+    REPO_ID=$1
+    FILENAME=$2
+    FILE_PATH="/app/$FILENAME"
+
+    if [ -f "$FILE_PATH" ]; then
+        echo "✅ Found existing model: $FILENAME"
     else
-        echo "✅ File $file already exists. Skipping download."
+        echo "⬇️  Model not found. Downloading $FILENAME..."
+        python3 -c "
+from huggingface_hub import hf_hub_download
+import sys
+
+try:
+    hf_hub_download(repo_id='$REPO_ID', filename='$FILENAME', local_dir='/app')
+    print('Download successful.')
+except Exception as e:
+    print(f'Error: {e}')
+    sys.exit(1)
+"
     fi
 }
 
 # --- SELECTION LOGIC ---
 if [ "$MODE" = "VISION" ]; then
-    echo "🚀 Booting in VISION mode..."
-    download_model "$VISION_REPO" "$VISION_FILE"
-    
+    download_if_needed "$VISION_REPO" "$VISION_FILE"
     MODEL_PATH="/app/$VISION_FILE"
     CONTEXT=4096
-    
 else
-    echo "🚀 Booting in CODER mode..."
-    download_model "$CODER_REPO" "$CODER_FILE"
-    
+    # Default to Coder
+    download_if_needed "$CODER_REPO" "$CODER_FILE"
     MODEL_PATH="/app/$CODER_FILE"
     CONTEXT=8192
 fi
 
-# --- START SERVER ---
-echo "🔥 Starting Llama.cpp Server..."
+echo "------------------------------------------------"
+echo "🚀 Launching Llama.cpp Server..."
+echo "Model: $MODEL_PATH"
+echo "Context: $CONTEXT"
+echo "------------------------------------------------"
+
+# Run Server
+# We use 'exec' to ensure the server process takes over
 exec llama-server \
     -m "$MODEL_PATH" \
     --host 0.0.0.0 \