Upload 13 files

Dockerfile

@@ -3,6 +3,12 @@ FROM python:3.9-slim
 # Set working directory
 WORKDIR /app
 
+# Create cache directories with proper permissions
+RUN mkdir -p /.cache/huggingface/hub && \
+    chmod -R 777 /.cache/huggingface && \
+    mkdir -p /app/.cache && \
+    chmod -R 777 /app/.cache
+
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     git \
@@ -24,8 +30,9 @@ RUN apt-get update && apt-get install -y \
 # Copy requirements first to leverage Docker cache
 COPY requirements.txt .
 
-# Install dependencies in specific order to avoid conflicts
+# Install dependencies with numpy compatibility fix
 RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
+    pip install --no-cache-dir --force-reinstall --no-deps numpy==1.21.6 && \
     pip install --no-cache-dir salesforce-lavis && \
     pip install --no-cache-dir torch>=1.10.0 torchvision>=0.11.0 && \
     pip install --no-cache-dir Flask==2.3.3 gunicorn==21.2.0 && \
@@ -39,13 +46,25 @@ RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
 # Copy application code
 COPY . .
 
+# Make startup script executable
+RUN chmod +x start.sh
+
 # Create necessary directories
 RUN mkdir -p static/css templates
 
-# Set environment variables for Hugging Face Spaces
+# Set environment variables for Hugging Face Spaces and caching
 ENV PYTHONPATH=/app
 ENV FLASK_APP=app.py
 ENV FLASK_ENV=production
+ENV TRANSFORMERS_CACHE=/app/.cache/transformers
+ENV HF_HOME=/app/.cache/huggingface
+ENV TORCH_HOME=/app/.cache/torch
+ENV HF_DATASETS_CACHE=/app/.cache/datasets
+ENV HUGGINGFACE_HUB_CACHE=/app/.cache/huggingface/hub
+
+# Create additional cache directories
+RUN mkdir -p /app/.cache/transformers /app/.cache/huggingface /app/.cache/torch /app/.cache/datasets && \
+    chmod -R 777 /app/.cache
 
 # Expose port for Hugging Face Spaces
 EXPOSE 7860
@@ -55,4 +74,4 @@ HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
     CMD curl -f http://localhost:7860/health || exit 1
 
 # Command to run the application
-CMD ["python", "app.py"]
+CMD ["./start.sh"]

Dockerfile.simple

@@ -0,0 +1,43 @@
+FROM python:3.9-slim
+
+# Set working directory
+WORKDIR /app
+
+# Set environment variables first
+ENV TRANSFORMERS_CACHE=/tmp/transformers
+ENV HF_HOME=/tmp/huggingface
+ENV TORCH_HOME=/tmp/torch
+ENV HUGGINGFACE_HUB_CACHE=/tmp/huggingface/hub
+ENV PYTHONPATH=/app
+ENV FLASK_APP=app.py
+ENV FLASK_ENV=production
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git curl wget build-essential \
+    libgl1-mesa-dev libglib2.0-0 \
+    libsm6 libxext6 libxrender-dev \
+    libgomp1 libgcc-s1 ffmpeg \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create cache directories with proper permissions
+RUN mkdir -p /tmp/transformers /tmp/huggingface /tmp/torch && \
+    chmod -R 777 /tmp/
+
+# Copy application files
+COPY . .
+
+# Install minimal dependencies only
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install Flask==2.3.3 pillow>=10.0.0 && \
+    pip install requests>=2.31.0 loguru
+
+# Expose port
+EXPOSE 7860
+
+# Simple health check
+HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:7860/health || exit 1
+
+# Run with simple python command
+CMD ["python", "app.py"]

app.py

@@ -1,5 +1,17 @@
 import os
 import io
+import sys
+
+# Set cache directories before importing any ML libraries
+os.environ['TRANSFORMERS_CACHE'] = os.environ.get('TRANSFORMERS_CACHE', '/app/.cache/transformers')
+os.environ['HF_HOME'] = os.environ.get('HF_HOME', '/app/.cache/huggingface') 
+os.environ['TORCH_HOME'] = os.environ.get('TORCH_HOME', '/app/.cache/torch')
+os.environ['HF_DATASETS_CACHE'] = os.environ.get('HF_DATASETS_CACHE', '/app/.cache/datasets')
+
+# Create cache directories if they don't exist
+for cache_dir in ['/app/.cache/transformers', '/app/.cache/huggingface', '/app/.cache/torch', '/app/.cache/datasets']:
+    os.makedirs(cache_dir, exist_ok=True)
+
 import torch
 from flask import Flask, render_template, request, jsonify, url_for
 from PIL import Image
@@ -8,13 +20,23 @@ import logging
 
 # Safe import with error handling
 try:
+    import numpy as np
+    # Check numpy version compatibility
+    numpy_version = np.__version__
+    logging.info(f"NumPy version: {numpy_version}")
+    
     from transformers import AutoTokenizer
     from blip2_vicuna_instruct import Blip2VicunaInstruct
     MODEL_AVAILABLE = True
+    logging.info("All imports successful")
 except ImportError as e:
     logging.error(f"Model import failed: {e}")
     MODEL_AVAILABLE = False
     Blip2VicunaInstruct = None
+except Exception as e:
+    logging.error(f"Unexpected error during import: {e}")
+    MODEL_AVAILABLE = False
+    Blip2VicunaInstruct = None
 
 app = Flask(__name__)
 app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # 16MB max file size

start.sh

@@ -0,0 +1,64 @@
+#!/bin/bash
+set -e
+
+echo "🚀 Starting EmoVIT application..."
+
+# Create and set permissions for cache directories
+echo "📁 Setting up cache directories..."
+mkdir -p /.cache/huggingface/hub
+mkdir -p /app/.cache/transformers
+mkdir -p /app/.cache/huggingface
+mkdir -p /app/.cache/torch
+mkdir -p /app/.cache/datasets
+
+# Set permissions
+chmod -R 777 /.cache/ 2>/dev/null || true
+chmod -R 777 /app/.cache/ 2>/dev/null || true
+
+# Set environment variables
+export TRANSFORMERS_CACHE=${TRANSFORMERS_CACHE:-/app/.cache/transformers}
+export HF_HOME=${HF_HOME:-/app/.cache/huggingface}
+export TORCH_HOME=${TORCH_HOME:-/app/.cache/torch}
+export HF_DATASETS_CACHE=${HF_DATASETS_CACHE:-/app/.cache/datasets}
+export HUGGINGFACE_HUB_CACHE=${HUGGINGFACE_HUB_CACHE:-/app/.cache/huggingface/hub}
+
+echo "✅ Cache directories set up"
+echo "🔧 Environment variables:"
+echo "  TRANSFORMERS_CACHE=$TRANSFORMERS_CACHE"
+echo "  HF_HOME=$HF_HOME" 
+echo "  TORCH_HOME=$TORCH_HOME"
+
+# Test dependencies
+echo "🔍 Testing dependencies..."
+python3 -c "
+import sys
+import warnings
+warnings.filterwarnings('ignore')
+
+try:
+    import numpy
+    print(f'✅ NumPy {numpy.__version__}')
+except Exception as e:
+    print(f'❌ NumPy: {e}')
+
+try:
+    import cv2
+    print(f'✅ OpenCV {cv2.__version__}')
+except Exception as e:
+    print(f'❌ OpenCV: {e}')
+
+try:
+    import torch
+    print(f'✅ PyTorch {torch.__version__}')
+except Exception as e:
+    print(f'❌ PyTorch: {e}')
+
+try:
+    import transformers
+    print(f'✅ Transformers {transformers.__version__}')
+except Exception as e:
+    print(f'❌ Transformers: {e}')
+"
+
+echo "🌐 Starting Flask application..."
+exec python3 app.py