fixing searcg_enigne problem

Dockerfile

@@ -1,3 +1,4 @@
+# First stage: Build and load the dataset
 FROM python:3.9 as core
 
 # Set the working directory
@@ -29,9 +30,11 @@ RUN pip install -r ./api/requirements.txt
 # Copy the API files
 COPY ./api ./api
 
-# Copy the serialized engine from the first stage
+# Copy the core files to the second stage to ensure search_engine is available
+COPY ./core ./core
+
+# Copy the serialized engine from the first stage to the API directory
 COPY --from=core /app/core/engine.pickle /app/api/engine.pickle
-COPY --from=core /app/core/search_engine /app/api/search_engine
 
 # Set the PYTHONPATH to include the /app directory
 ENV PYTHONPATH="/app"

api/web_server.py

@@ -1,7 +1,7 @@
 import dill
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
-from search_engine import PromptSearchEngine
+from core.search_engine import PromptSearchEngine
 
 
 class Query(BaseModel):