Upload app\core\config.py with huggingface_hub

app//core//config.py

@@ -0,0 +1,75 @@
+"""
+설정 관리 모듈
+환경 변수 및 애플리케이션 설정을 관리합니다.
+"""
+
+import os
+from pathlib import Path
+from typing import Optional
+from dotenv import load_dotenv
+
+# .env 파일 로드
+load_dotenv()
+
+# 프로젝트 루트 디렉토리
+PROJECT_ROOT = Path(__file__).parent.parent.parent
+
+class Config:
+    """애플리케이션 설정 클래스"""
+    
+    # Flask 설정
+    SECRET_KEY: str = os.getenv('SECRET_KEY', 'dev-secret-key-change-in-production')
+    
+    # 데이터베이스 URI 설정
+    # PostgreSQL: postgresql://user:password@host:port/database
+    # SQLite (기본값): sqlite:///instance/finance_analysis.db
+    _database_url = os.getenv('DATABASE_URL', '')
+    if _database_url and (_database_url.startswith('postgresql://') or _database_url.startswith('postgres://')):
+        # PostgreSQL URL이 제공되면 사용
+        SQLALCHEMY_DATABASE_URI: str = _database_url
+    else:
+        # 기본값: SQLite 사용
+        if _database_url:
+            # DATABASE_URL이 설정되었지만 PostgreSQL 형식이 아닌 경우 경고
+            import warnings
+            warnings.warn(f"DATABASE_URL이 설정되었지만 PostgreSQL 형식이 아닙니다. SQLite를 사용합니다: {_database_url[:50]}...")
+        SQLALCHEMY_DATABASE_URI: str = f'sqlite:///{PROJECT_ROOT / "instance" / "finance_analysis.db"}'
+    SQLALCHEMY_TRACK_MODIFICATIONS: bool = False
+    MAX_CONTENT_LENGTH: int = 100 * 1024 * 1024  # 100MB
+    
+    # Ollama 설정
+    OLLAMA_BASE_URL: str = os.getenv('OLLAMA_BASE_URL', 'http://localhost:11434')
+    
+    # 경로 설정
+    UPLOAD_FOLDER: Path = PROJECT_ROOT / 'uploads'
+    VECTOR_DB_PATH: Path = PROJECT_ROOT / 'vector_db'
+    KNOWLEDGE_GRAPH_PATH: Path = PROJECT_ROOT / 'knowledge_graphs'
+    TEMPLATES_FOLDER: Path = PROJECT_ROOT / 'templates'
+    STATIC_FOLDER: Path = PROJECT_ROOT / 'static'
+    INSTANCE_FOLDER: Path = PROJECT_ROOT / 'instance'
+    
+    # 파일 확장자 설정
+    ALLOWED_EXTENSIONS: set[str] = {'txt', 'md', 'pdf', 'docx', 'epub'}
+    
+    # 임베딩 모델 설정
+    EMBEDDING_MODEL_NAME: str = os.getenv('EMBEDDING_MODEL_NAME', 'sentence-transformers/all-MiniLM-L6-v2')
+    RERANKER_MODEL_NAME: str = os.getenv('RERANKER_MODEL_NAME', 'BAAI/bge-reranker-base')
+    
+    # Gemini API 설정
+    GEMINI_API_KEY: Optional[str] = os.getenv('GEMINI_API_KEY', None)
+    
+    @classmethod
+    def ensure_directories(cls) -> None:
+        """필수 디렉토리 생성"""
+        cls.UPLOAD_FOLDER.mkdir(parents=True, exist_ok=True)
+        cls.VECTOR_DB_PATH.mkdir(parents=True, exist_ok=True)
+        cls.KNOWLEDGE_GRAPH_PATH.mkdir(parents=True, exist_ok=True)
+        cls.STATIC_FOLDER.mkdir(parents=True, exist_ok=True)
+        cls.INSTANCE_FOLDER.mkdir(parents=True, exist_ok=True)
+
+
+def get_config() -> Config:
+    """설정 인스턴스 반환"""
+    return Config
+
+