Spaces:
Sleeping
Sleeping
File size: 588 Bytes
e9ce2a7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import os
from dotenv import load_dotenv
load_dotenv()
class Config:
GEMINI_API_KEY = os.environ.get('GEMINI_API_KEY') #os.getenv('GEMINI_API_KEY')
SECRET_KEY = os.environ.get('SECRET_KEY', 'your-secret-key-here')
UPLOAD_FOLDER = 'uploads'
VECTOR_DB_PATH = 'vector_db'
MAX_CONTENT_LENGTH = 16 * 1024 * 1024
EMBEDDING_MODEL = 'sentence-transformers/all-MiniLM-L6-v2'
CHUNK_SIZE = 1000
CHUNK_OVERLAP = 200
TOP_K = 5
ALLOWED_EXTENSIONS = {'pdf'}
os.makedirs(Config.UPLOAD_FOLDER, exist_ok=True)
os.makedirs(Config.VECTOR_DB_PATH, exist_ok=True)
|