config.py

import json
import os
import logging
from typing import Dict, Any, Optional

class Config:
    """配置管理类"""
    def __init__(self):
        """初始化配置类"""
        self.config_path = os.path.join(os.path.dirname(__file__), 'config.json')
        self._setup_logging()  # 先设置日志
        self.load_config()     # 再加载配置

    def _setup_logging(self) -> None:
        """设置日志配置"""
        log_dir = "logs"
        os.makedirs(log_dir, exist_ok=True)
        
        logging.basicConfig(
            level=logging.INFO,
            format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
            handlers=[
                logging.FileHandler(os.path.join(log_dir, 'app.log'), encoding='utf-8'),
                logging.StreamHandler()
            ]
        )
        self.logger = logging.getLogger(__name__)

    def load_config(self) -> None:
        """从config.json加载配置"""
        try:
            with open(self.config_path, 'r', encoding='utf-8') as f:
                self.config = json.load(f)
            self.logger.info("成功加载配置文件")
        except Exception as e:
            self.logger.error(f"加载配置文件失败: {e}")
            self.config = {}

    def get_model_config(self) -> Dict[str, Any]:
        """获取模型配置"""
        return self.config.get('model_config', {})

    def get_ui_config(self) -> Dict[str, Any]:
        """获取UI配置"""
        return self.config.get('ui_config', {})

    def get_system_config(self) -> Dict[str, Any]:
        """获取系统配置"""
        return self.config.get('system_config', {})

    def get_proxy_config(self) -> Dict[str, Any]:
        """获取代理配置"""
        system_config = self.get_system_config()
        return system_config.get('proxy', {
            'enabled': False,
            'http': None,
            'https': None,
            'timeout': 30,
            'retry_count': 3
        })

    def get_prompts(self) -> Dict[str, Any]:
        """获取提示词配置"""
        return self.config.get('prompts', {})

    def get_image_types(self) -> Dict[str, Dict[str, str]]:
        """获取图片类型配置"""
        prompts = self.get_prompts()
        return prompts.get('analysis_prompts', {})

    def get_image_type_prompt(self, image_type: str) -> Optional[Dict[str, str]]:
        """获取特定图片类型的提���词配置"""
        image_types = self.get_image_types()
        return image_types.get(image_type)

    def get_upload_path(self) -> str:
        """获取上传路径"""
        system_config = self.get_system_config()
        upload_path = system_config.get('upload_path', 'uploads/')
        os.makedirs(upload_path, exist_ok=True)
        return upload_path

    def get_cache_path(self) -> str:
        """获取缓存路径"""
        system_config = self.get_system_config()
        cache_path = system_config.get('cache_path', 'cache/')
        os.makedirs(cache_path, exist_ok=True)
        return cache_path

    def is_supported_image_type(self, mime_type: str) -> bool:
        """检查是否支持的图片类型"""
        system_config = self.get_system_config()
        return mime_type in system_config.get('supported_image_types', [])

    def is_supported_doc_type(self, mime_type: str) -> bool:
        """检查是否支持的文档类型"""
        system_config = self.get_system_config()
        return mime_type in system_config.get('supported_doc_types', [])

    def save_uploaded_file(self, uploaded_file) -> str:
        """
        保存上传的文件并返回保存路径
        :param uploaded_file: StreamlitUploadedFile对象
        :return: 保存后的文件路径
        """
        # 获取上传路径
        upload_path = self.get_upload_path()
        
        # 获取原始文件名
        original_filename = uploaded_file.name
        file_extension = original_filename.split('.')[-1].lower()
        
        # 构建保存路径
        save_path = os.path.join(upload_path, original_filename)
        
        # 处理文件名冲突
        if os.path.exists(save_path):
            base_name = original_filename.rsplit('.', 1)[0]
            counter = 1
            while os.path.exists(save_path):
                new_filename = f"{base_name}_{counter}.{file_extension}"
                save_path = os.path.join(upload_path, new_filename)
                counter += 1
        
        # 保存文件
        with open(save_path, "wb") as f:
            f.write(uploaded_file.getbuffer())
        
        return save_path

    def get_login_password(self):
        """获取登录密码"""
        return self.config.get("system_config", {}).get("login_config", {}).get("password", "")

# 创建全局配置实例
config = Config()