app/config.py

import os

#########################
# General Configuration #
#########################

DB_NAME = "database.db"
DB_PATH = f"/data/{DB_NAME}" if os.path.isdir("/data") else DB_NAME # If /data available => means local storage is enabled => let's use it!
if os.path.isdir("/data"):
    print("Local storage is enabled")
else:
    print("Local storage is disabled")

# LLM Models Configuration
AVAILABLE_MODELS = {
    "GPT-4": "gpt4",
    "Claude-3": "claude3",
    "Gemini-Pro": "gemini",
    "Mixtral": "mixtral",
    "Llama-2": "llama2",
    # Add more models as needed
}

# General Configuration
MAX_PROMPT_LENGTH = 2000
MIN_PROMPT_LENGTH = 10

# Sync settings
SYNC_DB = True
DB_DATASET_ID = os.getenv('DATASET_ID')
SPACE_ID = os.getenv('SPACE_ID')