Update Dockerfile

Dockerfile

@@ -1,36 +1,71 @@
-# استخدام الصورة الرسمية التي تدمج Open WebUI مع Ollama
-FROM ghcr.io/open-webui/open-webui:ollama
-
-# ── إعدادات المسارات الدائمة (HF Spaces يربط المخزن على /data) ──────────────
-ENV OLLAMA_MODELS=/data/ollama
-ENV DATA_DIR=/data/webui
-ENV PORT=7860
-ENV HOST=0.0.0.0
-
-# ── إبقاء النموذج في الذاكرة إلى الأبد (لا تفريغ عند الخمول) ───────────────
-ENV OLLAMA_KEEP_ALIVE=-1
-
-# ── إعدادات النموذج المراد تحميله من Hugging Face ────────────────────────────
-# غيّر هذه القيم فقط لتشغيل أي نموذج GGUF من HF دون تعديل أي شيء آخر:
-#
-#   HF_REPO   : مسار المستودع على HF  (مثال: unsloth/gemma-4-E2B-it-GGUF)
-#   HF_FILE   : اسم ملف GGUF بدقة كما يظهر في المستودع
-#   MODEL_NAME: الاسم الذي سيظهر به النموذج في واجهة Open WebUI
-#   HF_TOKEN  : (اختياري) مطلوب فقط للنماذج الخاصة/المحمية (gated models)
-ENV HF_REPO="gijl/gemma-4-E2B-it-GGUF"
-ENV HF_FILE="gemma-4-E2B-it-UD-Q5_K_XL.gguf"
-ENV MODEL_NAME="my-gemma"
-ENV HF_TOKEN=""
-
-# ── معاملات تشغيل النموذج (يمكن تعديلها حسب حجم النموذج والذاكرة المتاحة) ──
-ENV MODEL_CTX=128000
-ENV MODEL_THREADS=2
-
-# تثبيت أداة Hugging Face
-RUN pip install -U huggingface_hub
-
-# نسخ سكريبت التشغيل
-COPY run.sh /app/run.sh
-RUN chmod +x /app/run.sh
-
-CMD ["/app/run.sh"]
+#!/bin/bash
+
+# تحديد المسارات الأساسية
+MODELS_DIR="/data/models"
+MODEL_FILE="gemma-4-E2B-it-UD-Q5_K_XL.gguf"
+OLLAMA_MODEL_NAME="my-gemma"
+
+# إنشاء المجلدات المطلوبة في المخزن الدائم (بما فيها كاش Hugging Face)
+mkdir -p "$MODELS_DIR"
+mkdir -p /data/ollama
+mkdir -p /data/webui
+mkdir -p /data/.hf_cache
+
+# تشغيل خادم Ollama في الخلفية
+echo "Starting Ollama server in background..."
+OLLAMA_MODELS=/data/ollama ollama serve &
+
+# [العامل الخلفي] دالة مستقلة لمعالجة النموذج دون تعطيل إقلاع الواجهة
+launch_model_setup_async() {
+    echo "[Async Worker] Checking Ollama availability..."
+    # الانتظار الذكي حتى يستجيب خادم Ollama بالكامل
+    while [ "$(curl -s -o /dev/null -w '%{http_code}' http://localhost:11434/api/tags)" != "200" ]; do
+        sleep 3
+    done
+    echo "[Async Worker] Ollama server is ready."
+
+    # خطوة حاسمة: التحقق مما إذا كان النموذج مسجلاً مسبقاً لتوفير وقت الإقلاع اللاحق
+    if ollama list | grep -q "$OLLAMA_MODEL_NAME"; then
+        echo "[Async Worker] Model '$OLLAMA_MODEL_NAME' already registered in Ollama. Ready to chat!"
+        return 0
+    fi
+
+    # التحقق من وجود الملف الفعلي أو تحميله
+    if [ ! -f "$MODELS_DIR/$MODEL_FILE" ]; then
+        echo "[Async Worker] Model file not found. Downloading from Hugging Face..."
+        huggingface-cli download gijl/gemma-4-E2B-it-GGUF "$MODEL_FILE" \
+            --local-dir "$MODELS_DIR"
+    else
+        echo "[Async Worker] Model file detected in persistent storage."
+    fi
+
+    # تحميل ملف الرؤية (Adapter)
+    if [ ! -f "$MODELS_DIR/mmproj-BF16.gguf" ]; then
+        echo "[Async Worker] Downloading vision adapter..."
+        huggingface-cli download gijl/gemma-4-E2B-it-GGUF mmproj-BF16.gguf \
+            --local-dir "$MODELS_DIR"
+    fi
+
+    # إنشاء ملف الـ Modelfile ديناميكياً
+    echo "[Async Worker] Creating Modelfile..."
+    cat <<EOF > "$MODELS_DIR/Modelfile"
+FROM $MODELS_DIR/$MODEL_FILE
+# لتفعيل ميزة قراءة الصور لاحقاً، يمكنك إلغاء التعليق (#) عن السطر أدناه:
+# ADAPTER $MODELS_DIR/mmproj-BF16.gguf
+
+PARAMETER num_ctx 8192
+EOF
+
+    # تسجيل النموذج في Ollama (يحدث الآن في الخلفية بأمان)
+    echo "[Async Worker] Registering model with Ollama..."
+    ollama create "$OLLAMA_MODEL_NAME" -f "$MODELS_DIR/Modelfile"
+    echo "[Async Worker] Task complete! Model '$OLLAMA_MODEL_NAME' is active."
+}
+
+# إطلاق عامل الخلفية كعملية منفصلة (تفرع باكتملال &)
+launch_model_setup_async &
+
+# تشغيل واجهة Open WebUI فوراً (يتولى الصدارة لمنع انهيار الحاوية بسبب مهلة الوقت)
+echo "Launching Open WebUI interface immediately to pass host health checks..."
+cd /app/backend
+exec bash start.sh