backend/requirements.txt

# Copyright 2025 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# --- Core Flask Application ---
flask
gunicorn
requests
pillow
diskcache
google-auth

# --- RAG: PDF and Text Processing ---
PyMuPDF # For parsing PDF files (fitz)
nltk # For NLTKTextSplitter
tqdm # For progress bars during knowledge base build

# --- RAG: LangChain Components ---
langchain
langchain_community
langchain-huggingface
langchain-text-splitters
chromadb # Vector store for embeddings
rank_bm25 # For the BM25 sparse retriever

# --- RAG: ML/NLP Models & Frameworks ---
# User-specified versions for reproducibility
torch==2.6.0
numpy==2.0.2

# Libraries for loading and running the embedding and NER models
sentence-transformers
transformers
accelerate
bitsandbytes
sentencepiece
stanza # For NER (Named Entity Recognition)