#!/usr/bin/env bash
set -e

# إنشاء مجلدات لتخزين النماذج والـ cache
mkdir -p models /app/.cache/huggingface/hub
chown -R appuser:appuser models /app/.cache/huggingface
chmod -R 755 /app/.cache/huggingface

# تحميل ملف Mistral .gguf إذا لم يكن موجودًا مسبقًا
python - <<PY
from huggingface_hub import hf_hub_download
import os
repo_id   = "TheBloke/Mistral-7B-Instruct-v0.1-GGUF"
filename  = "mistral-7b-instruct-v0.1.Q2_K.gguf"
local_dir = "models"
if not os.path.exists(os.path.join(local_dir, filename)):
    hf_hub_download(
        repo_id=repo_id,
        filename=filename,
        local_dir=local_dir,
        local_dir_use_symlinks=False,
        force_download=False
    )
    print("✅ تم تحميل Mistral .gguf")
else:
    print("✅ ملف Mistral .gguf موجود مسبقًا")
PY

# تحميل نموذج MGZON-FLAN-T5 إذا لم يكن موجودًا مسبقًا
python - <<PY
from huggingface_hub import snapshot_download
import os
repo_id   = "MGZON/mgzon-flan-t5-base"
local_dir = "/app/.cache/huggingface/hub/models--MGZON--mgzon-flan-t5-base/snapshots"
if not os.path.exists(local_dir):
    snapshot_download(
        repo_id=repo_id,
        local_dir=local_dir,
        local_dir_use_symlinks=False,
        force_download=False
    )
    print("✅ تم تحميل MGZON-FLAN-T5")
else:
    print("✅ ملفات MGZON-FLAN-T5 موجودة مسبقًا")
PY