modified filea

Dockerfile

@@ -9,6 +9,13 @@ ENV PATH="/home/user/.local/bin:$PATH"
 
 WORKDIR /app
 
+# Set environment variables for database connection
+ENV MONGO_DB_URL=mongodb+srv://amrita:PwywryiC55QmERU0@cluster0.xyssel6.mongodb.net/?retryWrites=true&w=majority&appName=Cluster0
+ENV NEO_DB_HOST=neo4j+s://bedcd9e5.databases.neo4j.io
+ENV NEO_DB_PORT=7687
+ENV NEO_DB_USER=neo4j
+ENV NEO_DB_PASSWORD=G0NJr1Y0I6H-E4hsms7XrMMN8VRqRJiUdk7aslHv7Ow
+
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 

app.py

@@ -18,13 +18,21 @@ app = FastAPI()
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Get environment variables
+MONGO_DB_URL=os.getenv('MONGO_DB_URL')
+NEO_DB_HOST = os.getenv('NEO_DB_HOST')
+NEO_DB_PORT = os.getenv('NEO_DB_PORT')
+NEO_DB_USER = os.getenv('NEO_DB_USER')
+NEO_DB_PASSWORD = os.getenv('NEO_DB_PASSWORD')
+
+
 # MongoDB setup
-mongo_client = MongoClient("mongodb+srv://amrita:PwywryiC55QmERU0@cluster0.xyssel6.mongodb.net/?retryWrites=true&w=majority&appName=Cluster0")
+mongo_client = MongoClient("MONGO_DB_URL")
 db = mongo_client["pdf_db"]
 chunks_collection = db["chunks"]
 
 # Neo4j setup
-neo4j_driver = GraphDatabase.driver("neo4j+s://bedcd9e5.databases.neo4j.io:7687", auth=("neo4j", "G0NJr1Y0I6H-E4hsms7XrMMN8VRqRJiUdk7aslHv7Ow"))
+neo4j_driver = GraphDatabase.driver("NEO_DB_HOST:NEO_DB_PORT", auth=("NEO_DB_USER", "NEO_DB_PASSWORD"))
 
 # Load pre-trained BERT model and tokenizer
 tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')