config.py

import os
import sys
from dotenv import load_dotenv

# Load environment variables
load_dotenv()

# API Keys
GROQ_API_KEY = os.getenv("GROQ_API_KEY")
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")

if not GROQ_API_KEY:
    print("Warning: GROQ_API_KEY not found in .env file", file=sys.stderr)
if not GEMINI_API_KEY:
    print("Warning: GEMINI_API_KEY not found in .env file", file=sys.stderr)

# Paths
PDF_FILES = [
    "Constitution.pdf",
    "Pakistan Penal Code.pdf"
]

CHROMA_DB_DIR = "./chroma_db_legal"

# Models
EMBEDDING_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
LLM_MODEL_NAME = "llama-3.3-70b-versatile"
GEMINI_MODEL_NAME = "gemini-2.5-flash"

# RAG Configuration
CHUNK_SIZE = 1000
CHUNK_OVERLAP = 200
RETRIEVER_K = 6