Updated files

Dockerfile

@@ -1,21 +1,14 @@
-# Use an official Python runtime as a parent image
 FROM python:3.9-slim
-
-# Set the working directory in the container
 WORKDIR /app
-
-# Copy the current directory contents into the container at /app
+COPY requirements.txt ./requirements.txt
+RUN apt-get update \
+    && apt-get -y install libpq-dev gcc \
+    && pip install psycopg2
+# Install uvicorn
+RUN pip install uvicorn
+# Install dependencies
+RUN pip install -r requirements.txt
 COPY . /app
 
-# Install any needed packages specified in requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Make port 7860 available to the world outside this container
-EXPOSE 7860
-
-# Create a non-root user and switch to it
-RUN adduser --disabled-password --gecos '' appuser
-USER appuser
-
-# Run main_app.py when the container launches
-CMD ["python", "main_app.py"]
+ENTRYPOINT ["uvicorn", "main_app:app"]
+CMD ["--host", "0.0.0.0", "--port", "7860"]

main_app.py

@@ -21,10 +21,12 @@ class AnswerResponse(BaseModel):
     answer: str
 
 def init_rag_system():
-    pdf_path = os.environ.get("SOURCE_DATA")
-    vector_store_path = os.environ.get("VECTOR_STORE")
+    pdf_path = os.environ.get("SOURCE_DATA", "iesc111.pdf")
+    vector_store_path = os.environ.get("VECTOR_STORE", "chroma_langchain_db")
+    
     # Load embeddings
     embeddings = get_embeddings()
+    
     if os.path.exists(vector_store_path) and os.listdir(vector_store_path):
         print("Loading existing vector store...")
         vector_store = load_vector_store(embeddings)
@@ -33,6 +35,7 @@ def init_rag_system():
         documents = load_pdf(pdf_path)
         unique_ids = create_unique_ids(documents)
         vector_store = create_vector_store(documents, unique_ids, embeddings)
+    
     retriever = get_retriever(vector_store)
     model = get_model()
     rag_chain = create_rag_chain(model, retriever)
@@ -66,7 +69,6 @@ if __name__ == "__main__":
     import uvicorn
     uvicorn.run(
         app,
-        host=os.getenv("UVICORN_HOST"),
-        port=int(os.getenv("UVICORN_PORT")),
-        # reload=True
+        host=os.getenv("UVICORN_HOST", "0.0.0.0"),
+        port=int(os.getenv("UVICORN_PORT", 7860)),
     )