import os
from pathlib import Path

PROJECT_ROOT = Path(__file__).resolve().parents[1]  # /workspace
STATIC_DIR = PROJECT_ROOT / "static"

# 프롬프트 & 스키마 경로
DEFAULT_PROMPT_PATH = STATIC_DIR / "prompt" / "default_prompt_v0.1.txt"
STRUCTURED_PROMPT_PATH = STATIC_DIR / "prompt" / "structured_prompt_v0.1.txt"
I18N_PROMPT_PATH = STATIC_DIR / "prompt" / "i18n_test_prompt_kor.txt"
D6C_PROMPT_PATH = STATIC_DIR / "prompt" / "d6c_test_prompt_eng.txt"
STRUCTURED_SCHEMA_PATH = STATIC_DIR / "structured_schema.json"

# html 경로
EXTRACT_DEFAULT_PATH = STATIC_DIR / "html" / "extract_guide.html"
EXTRACT_STRUCTURED_PATH = STATIC_DIR / "html" / "extraction_structured_guide.html"
GENERAL_GUIDE_PATH = STATIC_DIR / "html" / "general_guide.html"
SCHEMA_FILE_PATH = STATIC_DIR / "html" / "schema_file_guide.html"
SUMMARY_HTML_DIR = STATIC_DIR / "html" / "generated"

# Ollama 모델 API 엔드포인트
OLLAMA_URL = [
    "http://pgn_ollama_gemma:11534/api/generate",
    "http://pgn_ollama_gpt_oss:11634/api/generate",
    "http://pgn_ollama_qwen:11734/api/generate",
]

# OCR API URL
OCR_API_URL = "http://ocr_api_8890:8890/ocr"
OCR_STATUS_PUBLIC_URL = "http://localhost:8890/ocr/progress"

# OCR Redis 기본 설정
OCR_REDIS_HOST = "ocr_redis"
OCR_REDIS_PORT = 6379
OCR_REDIS_DB = 0

# llm_gateway 서비스 Redis 설정
PGN_REDIS_HOST = "pgn_redis"
PGN_REDIS_PORT = 6379
PGN_REDIS_DB = 2

# ✅ MinIO 설정
MINIO_ENDPOINT = "172.16.10.175:9000"  # Docker 네트워크용 내부 주소
MINIO_ACCESS_KEY = "kyy"
MINIO_SECRET_KEY = "hLAk3aQfH8HTs7ELTcyR"
MINIO_BUCKET_NAME = "ocr-gateway"  # 미리 생성한 버킷명
MINIO_RESULTS_BUCKET_NAME = "ocr-gateway-results"  # 결과 저장용 버킷

# 파일 저장 경로
UPLOAD_DIR = "/workspace/temp_upload"
CACHED_PROMPT_DIR = "/workspace/cached"
LOG_DIR = "/workspace/logs"
os.makedirs(UPLOAD_DIR, exist_ok=True)
os.makedirs(SUMMARY_HTML_DIR, exist_ok=True)
os.makedirs(CACHED_PROMPT_DIR, exist_ok=True)
os.makedirs(LOG_DIR, exist_ok=True)

# 허용된 확장자 목록
ALLOWED_EXTENSIONS = {".pdf", ".docx", ".jpg", ".jpeg", ".png"}