Adieee5's picture
Upload 8 files
e9ce2a7 verified
raw
history blame contribute delete
588 Bytes
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)