Remove local model and add auto-download

.gitignore

@@ -0,0 +1,5 @@
+# モデルキャッシュ
+model/
+model_cache/
+
+

app.py

@@ -3,10 +3,13 @@
 FastAPI 版 LLMView Word Tree サーバー
 """
 
+import os
 import threading
+from pathlib import Path
 from typing import List, Dict, Any, Optional
 
 from fastapi import FastAPI, HTTPException
+from huggingface_hub import snapshot_download
 from pydantic import BaseModel, Field
 
 try:
@@ -22,6 +25,18 @@ adapter = None
 status_message = "モデル初期化中..."
 status_lock = threading.Lock()
 
+HF_MODEL_REPO = os.getenv("HF_MODEL_REPO", "WatNeru/LLMView-model")
+HF_MODEL_FILENAME = os.getenv(
+    "HF_MODEL_FILENAME", "llama-3.2-3b-instruct-q4_k_m.gguf"
+)
+HF_LOCAL_DIR = Path(
+    os.getenv(
+        "HF_MODEL_LOCAL_DIR",
+        path_manager.base_path / "model_cache",
+    )
+)
+HF_TOKEN = os.getenv("HF_TOKEN")
+
 
 class WordTreeRequest(BaseModel):
     prompt_text: str = Field(..., description="生成に使用するプロンプト")
@@ -41,6 +56,33 @@ def _set_status(message: str) -> None:
         status_message = message
 
 
+def ensure_model_available() -> str:
+    """モデルファイルをローカルに用意（なければHFから取得）"""
+    current_path = Path(path_manager.get_model_path())
+    if current_path.exists():
+        return str(current_path)
+
+    HF_LOCAL_DIR.mkdir(parents=True, exist_ok=True)
+    _set_status("Hugging Face からモデルをダウンロード中...")
+
+    snapshot_download(
+        repo_id=HF_MODEL_REPO,
+        allow_patterns=HF_MODEL_FILENAME,
+        local_dir=str(HF_LOCAL_DIR),
+        local_dir_use_symlinks=False,
+        token=HF_TOKEN,
+    )
+    downloaded = HF_LOCAL_DIR / HF_MODEL_FILENAME
+    if not downloaded.exists():
+        raise FileNotFoundError(
+            f"モデル {HF_MODEL_FILENAME} が {downloaded} に見つかりません"
+        )
+
+    os.environ["LLM_MODEL_PATH"] = str(downloaded)
+    path_manager.model_path = str(downloaded)
+    return str(downloaded)
+
+
 def initialize_model() -> None:
     """RustAdapter とモデルを初期化"""
     global adapter
@@ -48,7 +90,7 @@ def initialize_model() -> None:
         _set_status("モデルを読み込み中です...")
         from package.rust_adapter import RustAdapter
 
-        model_path = path_manager.get_model_path()
+        model_path = ensure_model_available()
         adapter = RustAdapter.get_instance(model_path)
         _set_status("モデル準備完了")
     except Exception as exc:  # pragma: no cover

requirements.txt

@@ -19,5 +19,8 @@ gradio>=4.38.0
 fastapi>=0.111.0
 uvicorn>=0.30.0
 
+# Model management
+huggingface-hub>=0.24.0
+
 # その他
 typing-extensions>=4.0.0