File size: 1,316 Bytes
fe6855c
 
 
 
 
 
 
 
d9d8b61
fe6855c
 
d9d8b61
f404c1e
fe6855c
 
 
 
 
f404c1e
122ab6b
 
fe6855c
 
 
 
 
 
 
 
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
import os
from dotenv import load_dotenv

# Load environment variables from .env file
load_dotenv()


class Settings:
    ENV = os.getenv("ENV")  # Default to "local" if ENV is not set
    # API Token
    HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN") or os.environ.get("HUGGINGFACE_API_TOKEN")

    GEMENI_API_KEY_TOKEN = os.getenv("GEMENI_API_KEY") or os.environ.get("GEMENI_API_KEY")

    # Model Path
    MODEL_NAME = os.getenv("MODEL_NAME") or os.environ.get("MODEL_NAME") # Default if not set
    OCR_LANGUAGE = os.getenv("OCR_LANGUAGE", "eng")
    directory_path = os.getenv("EMAIL_DIRECTORY_PATH") or os.environ.get("EMAIL_DIRECTORY_PATH")  # Make configurable
    ALLOWED_PRIORITY_RULES_FILENAME=os.getenv("ALLOWED_PRIORITY_RULES_FILENAME") or os.environ.get("ALLOWED_PRIORITY_RULES_FILENAME")  # Make configurable
    DATA_DIRECTORY_ATTACHMENTS_PATH=os.getenv("DATA_DIRECTORY_ATTACHMENTS_PATH") or os.environ.get("DATA_DIRECTORY_ATTACHMENTS_PATH")  # Make configurable
    DATA_DIRECTORY_EMAILS_PATH=os.getenv("DATA_DIRECTORY_EMAILS_PATH") or os.environ.get("DATA_DIRECTORY_EMAILS_PATH")  # Make configurable
    # Debugging info
    print(f"Running in {ENV} mode with model path: {MODEL_NAME}")
    print(f"Running in {ENV} mode with TOKEN: {HUGGINGFACE_API_TOKEN}")

settings = Settings()