60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
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"}
|