Update Dockerfile

Dockerfile

@@ -1,54 +1,54 @@
 FROM ubuntu:24.04
 
-# Install everything we need
-RUN apt-get update && apt-get install -y \
-    python3 \
-    python3-pip \
-    curl \
-    bash \
-    wget \
+# Install all dependencies in one layer
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3 python3-pip python3-dev \
+    curl wget bash ca-certificates \
+    zstd \
     && rm -rf /var/lib/apt/lists/*
 
-# Download and install Ollama
+# Install Ollama
 RUN curl -fsSL https://ollama.ai/install.sh | sh
 
 # Set working directory
 WORKDIR /app
 
-# Copy requirements
+# Copy and install Python dependencies
 COPY requirements.txt .
-
-# Install Python dependencies
 RUN pip3 install --break-system-packages --no-cache-dir -r requirements.txt
 
-# Copy application
+# Copy application code
 COPY app.py .
 
-# Expose ports
-EXPOSE 7860 11434
-
-# Environment setup
-ENV OLLAMA_HOST=0.0.0.0:11434
-ENV PORT=7860
-
-# Simple startup script
-RUN cat > /app/run.sh << 'EOF'
+# Create startup script
+RUN cat > /app/entrypoint.sh << 'SCRIPT'
 #!/bin/bash
 set -e
 
-echo "Starting Ollama..."
+echo "🚀 Starting Ollama server..."
 /usr/bin/ollama serve > /tmp/ollama.log 2>&1 &
 OLLAMA_PID=$!
 
-echo "Waiting for Ollama to start..."
-sleep 3
+echo "⏳ Waiting for Ollama to initialize..."
+sleep 5
 
-echo "Starting FastAPI..."
+echo "📥 Pulling Mistral model (this may take a few minutes on first run)..."
+timeout 600 /usr/bin/ollama pull mistral || echo "⚠️  Model pull timeout - will download on first request"
+
+echo "🌐 Starting FastAPI server..."
 exec python3 /app/app.py
-EOF
+SCRIPT
+
+RUN chmod +x /app/entrypoint.sh
 
-RUN chmod +x /app/run.sh
+# Expose ports
+EXPOSE 7860 11434
+
+# Environment variables
+ENV OLLAMA_HOST=0.0.0.0:11434
+ENV PORT=7860
+ENV PYTHONUNBUFFERED=1
 
-# Override entrypoint completely
+# Run entrypoint
 ENTRYPOINT []
-CMD ["/bin/bash", "/app/run.sh"]
+CMD ["/app/entrypoint.sh"]