change approach to docker file

Dockerfile

@@ -6,24 +6,29 @@ WORKDIR /app
 # Copy your code
 COPY . /app
 
-# Install dependencies with --user flag
-RUN pip install --no-cache-dir --user -r requirements.txt
+# First install critical packages system-wide
+RUN pip install --no-cache-dir streamlit guardrails-ai
 
-# Ensure proper permissions
-RUN mkdir -p /app/.phoenix /tmp/.phoenix /app/.guardrails /usr/local/bin && \
-    chmod -R 777 /app/.phoenix /tmp/.phoenix /app/.guardrails /usr/local/bin && \
-    chmod -R 777 /usr/local/lib/python3.10/site-packages
+# Set proper permissions for system directories where packages get installed
+RUN mkdir -p /usr/local/bin && chmod -R 777 /usr/local/bin
+RUN chmod -R 777 /usr/local/lib/python3.10/site-packages
 
-# Create fallback implementations (keep your existing code)
+# Create necessary directories with appropriate permissions
+RUN mkdir -p /app/.phoenix /tmp/.phoenix /app/.guardrails && \
+    chmod -R 777 /app/.phoenix /tmp/.phoenix /app/.guardrails
+
+# Create fallback implementations
 RUN mkdir -p /app/guardrails_fallback
 RUN echo 'class DetectJailbreak:\n    def __init__(self, *args, **kwargs):\n        print("Using fallback DetectJailbreak implementation")\n        pass\n\nclass ToxicLanguage:\n    def __init__(self, *args, **kwargs):\n        print("Using fallback ToxicLanguage implementation")\n        pass' > /app/guardrails_fallback/__init__.py
 
+# Make the setup script executable
+RUN chmod +x /app/setup.sh
+
 # Create a non-root user
 RUN useradd -m -u 1000 appuser
 
-# Make the setup script executable and modify it to use --user flag
-RUN chmod +x /app/setup.sh
-RUN sed -i 's/pip install/pip install --user/g' /app/setup.sh
+# Install remaining dependencies (can use system or user install)
+RUN pip install --no-cache-dir -r requirements.txt
 
 # Change ownership
 RUN chown -R appuser:appuser /app /home/appuser
@@ -36,5 +41,8 @@ ENV PATH="/home/appuser/.local/bin:${PATH}"
 # Switch to non-root user
 USER appuser
 
+# Expose the port for Streamlit
+EXPOSE 7860
+
 # Launch the app
 CMD ["streamlit", "run", "/app/app.py", "--server.port", "7860", "--server.address", "0.0.0.0"]

setup.sh

@@ -1,23 +1,21 @@
 #!/bin/bash
-# setup.sh - Run once when the Streamlit app starts
 
-echo "Starting setup process..."
+echo "Setting up environment..."
 
-# Check for Guardrails token
-if [ ! -z "$GUARDRAILS_TOKEN" ]; then
-    echo "Configuring Guardrails with token..."
-    # Clean the token and configure
-    CLEAN_TOKEN=$(echo $GUARDRAILS_TOKEN | tr -d "\"' \n\t")
-    
-    # Configure guardrails
-    guardrails configure --token "$CLEAN_TOKEN" --enable-metrics --enable-remote-inferencing
-    
-    # Install validators
-    echo "Installing Guardrails validators..."
-    guardrails hub install hub://guardrails/detect_jailbreak || echo "Failed to install detect_jailbreak validator"
-    guardrails hub install hub://guardrails/toxic_language || echo "Failed to install toxic_language validator"
-else
-    echo "No GUARDRAILS_TOKEN found. Skipping Guardrails setup."
-fi
+# Configure guardrails with token
+python -c "
+import os
+import guardrails as gd
+try:
+    gd.configure(api_endpoint='https://api.guardrailsai.com', api_key=os.environ.get('GUARDRAILS_API_KEY', ''))
+    print('Guardrails configured successfully')
+except Exception as e:
+    print(f'Warning: Could not configure guardrails: {e}')
+"
+
+# Try installing validators but don't fail if they don't work
+echo "Installing Guardrails validators..."
+guardrails hub install hub://guardrails/detect_jailbreak || echo "Warning: Could not install detect_jailbreak validator"
+guardrails hub install hub://guardrails/toxic_language || echo "Warning: Could not install toxic_language validator"
 
 echo "Setup completed!"