improved dockerfile

Dockerfile

@@ -1,44 +1,24 @@
-# Stage 1: Build stage with Python and pip
-FROM python:3.9 AS builder
+FROM python:3.9
 
-# Create a non-root user
 RUN useradd -m -u 1000 user
 USER user
+ENV PATH="/home/user/.local/bin:$PATH"
 
-# Set the working directory
 WORKDIR /app
 
-# Copy requirements file and install dependencies
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
+RUN apt-get update && apt-get install git -y && apt-get install curl -y
+RUN curl -fsSL https://ollama.com/install.sh | sh
 
-# Stage 2: Runtime stage with Ollama
-FROM ollama/ollama:latest
-
-# Create the same non-root user for consistency
-RUN useradd -m -u 1000 user
-USER user
-
-# Set the environment PATH
-ENV PATH="/home/user/.local/bin:$PATH"
-
-# Set the working directory
-WORKDIR /app
-
-# Copy Python dependencies from the build stage
-COPY --from=builder /usr/local/lib/python3.9 /usr/local/lib/python3.9
-COPY --from=builder /usr/local/bin /usr/local/bin
+# Ensure `ollama` is installed
+RUN python -m pip show ollama
 
 # Copy the application code
 COPY --chown=user . /app
 
-# Expose the port for Streamlit
-EXPOSE 7860
-
-# Add and set up the startup script
 COPY --chown=user ./startup.sh /app/startup.sh
 RUN chmod +x /app/startup.sh
 
-# Command to run the application
-CMD ["/app/startup.sh"]
+CMD ["/app/startup.sh"]