change Dockerfile

Dockerfile

@@ -2,10 +2,21 @@ FROM python:3.9
 
 WORKDIR /code
 
+# Create backend directory first
+RUN mkdir -p /code/backend
+
+# Copy all required files
 COPY ./requirements.txt /code/requirements.txt
-COPY ./backend /code/backend
+COPY ./backend/__init__.py /code/backend/
+COPY ./backend/main.py /code/backend/
+COPY ./backend/text_utils.py /code/backend/
+COPY ./backend/openai_utils.py /code/backend/
+COPY ./backend/vector_store.py /code/backend/
 COPY ./frontend/dist /code/frontend/dist
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
+# Set Python path to include the backend directory
+ENV PYTHONPATH=/code
+
 CMD ["uvicorn", "backend.main:app", "--host", "0.0.0.0", "--port", "7860"]

backend/__init__.py

@@ -0,0 +1 @@
+# This can be empty 

backend/main.py

@@ -13,10 +13,10 @@ import logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Import our local utilities instead of aimakerspace
-from text_utils import CharacterTextSplitter, TextFileLoader, PDFLoader
-from openai_utils import SystemRolePrompt, UserRolePrompt, ChatOpenAI
-from vector_store import VectorDatabase
+# Update imports to use relative paths
+from .text_utils import CharacterTextSplitter, TextFileLoader, PDFLoader
+from .openai_utils import SystemRolePrompt, UserRolePrompt, ChatOpenAI
+from .vector_store import VectorDatabase
 
 load_dotenv()