requirements.txt

# --- Core ML & Data Science ---
# Pinning the original, known-good versions for local and cloud compatibility
torch==2.1.0
torchvision==0.16.0
pytorch-lightning

# Pinning Captum and its dependencies for stability
captum==0.7.0
numpy==1.26.4
scikit-learn
imbalanced-learn
pandas
# Let albumentations choose its required OpenCV version for max compatibility
opencv-python-headless
Pillow
matplotlib
scikit-image
tensorboard
# Pin protobuf to a known stable version
protobuf==3.20.0
PyYAML

# --- Image Augmentation ---
albumentations

# --- Web Frameworks & Frontend ---
streamlit
gradio

# --- LLM Integration ---
# Pinning LangChain libraries to a known stable and compatible set
langchain==0.2.5
langchain-core==0.2.10
langchain-groq==0.1.5
python-dotenv