Fix import paths and user permissions for Streamlit

Dockerfile

@@ -2,23 +2,26 @@ FROM python:3.10-slim-bookworm
 
 # Install apt packages
 COPY apt.txt /tmp/apt.txt
-RUN apt-get update && apt-get upgrade -y && xargs -r -a /tmp/apt.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get upgrade -y && \
+    xargs -r -a /tmp/apt.txt apt-get install -y && \
+    rm -rf /var/lib/apt/lists/*
 
-# Set working directory
-WORKDIR /app
+# Create non-root user and set home
+RUN useradd -m -u 1000 streamlituser
+USER streamlituser
+WORKDIR /home/streamlituser/app
 
-# Install Python dependencies
+# Copy and install requirements
 COPY requirements.txt .
 RUN pip install --upgrade pip && pip install -r requirements.txt
 
-# Copy code
+# Copy the rest of the code
 COPY . .
 
+# Make .streamlit writable
+RUN mkdir -p /home/streamlituser/.streamlit
+
 # Expose Streamlit port
 EXPOSE 7860
 
-ENV HOME=/home/user
-ENV STREAMLIT_HOME=/home/user/.streamlit
-RUN mkdir -p $STREAMLIT_HOME
-
-CMD ["streamlit", "run", "src/streamlit_app.py", "--server.port=7860", "--server.address=0.0.0.0"]
+CMD ["streamlit", "run", "src/streamlit_app.py", "--server.port=7860", "--server.address=0.0.0.0"]

src/streamlit_app.py

@@ -15,6 +15,8 @@ from rag_engine import generate_answer, summarize_document
 
 from pdf_parser import parse_pdf
 
+from pdf_parser import parse_pdf
+
 load_dotenv()
 
 st.set_page_config(page_title="DocSight-RAG 👁️📄", page_icon="📚")