renami/settings.py at main · Circloud/renami · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
import json
import os
import sys

class Settings:
    def __init__(self):
        try:
            # If the application is run as a bundle
            if getattr(sys, 'frozen', False):
                application_dir = sys._MEIPASS  # sys.MEIPASS points to the working directory of the exe file
                config_path = os.path.join(application_dir, 'config.json')
                config_template_path = os.path.join(application_dir, 'config_template.json')

                if os.path.exists(config_path):
                    self.config_file = config_path # for personal config, NOT pushed/packaged
                elif os.path.exists(config_template_path):
                    self.config_file = config_template_path # for default config, pushed/packaged
                else:
                    # Create default config_template.json if neither exists
                    self.config_file = config_template_path

                    config_template = {
                        "llm_provider": "openai",
                        "openai_api_key": "",
                        "openai_api_base_url": "https://api.openai.com/v1",
                        "openai_model": "gpt-4o-mini",
                        "gemini_api_key": "",
                        "gemini_api_base_url": "https://generativelanguage.googleapis.com/v1beta/openai",
                        "gemini_model": "gemini-2.0-flash-lite-preview-02-05",
                        "doubao_api_key": "",
                        "doubao_api_base_url": "https://ark.cn-beijing.volces.com/api/v3/",
                        "doubao_model": "doubao-1-5-vision-pro-32k-250115",
                        "openai_compatible_api_key": "",
                        "openai_compatible_api_base_url": "",
                        "openai_compatible_model": "",
                        "naming_language": "en",
                        "naming_convention": "with-spaces",
                        "custom_instruction": ""
                    }
                    with open(self.config_file, 'w') as f:
                        json.dump(config_template, f, indent=4)

            # If the application is run as a script
            else:
                if os.path.exists('config.json'):
                    self.config_file = 'config.json'  # for personal config, NOT pushed/packaged
                else:
                    self.config_file = 'config_template.json'  # for default config, pushed/packaged

        except Exception as e:
            print(f"Error initializing settings: {e}")
            return {}

    def load(self):
        """Load settings from config file"""
        with open(self.config_file, 'r') as f:
            settings = json.load(f)

            # Get the current provider's base URL
            llm_provider = settings.get('llm_provider')
            api_base_url = settings.get(f"{llm_provider}_api_base_url")

            # Format the API base URL if it exists
            if api_base_url:

                # Add https:// if no protocol is specified
                if not api_base_url.startswith(('http://', 'https://')):
                    api_base_url = 'https://' + api_base_url

                if llm_provider == 'openai':
                    api_base_url = api_base_url.rstrip('/')

                if llm_provider == 'gemini':
                    api_base_url = api_base_url.rstrip('/')

                if llm_provider == 'doubao':
                    api_base_url = api_base_url.rstrip('/')

                if llm_provider == 'openai_compatible':
                    api_base_url = api_base_url.rstrip('/')

            return settings

    def save(self, settings):
        """Save settings to config file"""
        with open(self.config_file, 'w') as f:
            json.dump(settings, f, indent=4)

    def get(self, key, default=None):
        """Get a setting value"""
        return self.load().get(key, default)

    def update(self, new_settings):
        """Update settings with new values"""
        current_settings = self.load()
        current_settings.update(new_settings)
        self.save(current_settings)