Update Dockerfile

Dockerfile

@@ -24,16 +24,71 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy all app code
 COPY . /app/
 
-# Handle entrypoint (rename if needed, then make executable)
+# Debug: List what's in scripts directory
+RUN ls -la /app/scripts/ || echo "Scripts directory not found"
+
+# Handle entrypoint - check both possible names and create if needed
 RUN if [ -f /app/scripts/entrypoint_hf.sh ]; then \
-        mv /app/scripts/entrypoint_hf.sh /app/scripts/entrypoint.sh; \
-    fi && \
-    if [ -f /app/scripts/entrypoint.sh ]; then \
+        mv /app/scripts/entrypoint_hf.sh /app/scripts/entrypoint.sh && \
+        chmod +x /app/scripts/entrypoint.sh; \
+    elif [ -f /app/scripts/entrypoint.sh ]; then \
+        chmod +x /app/scripts/entrypoint.sh; \
+    else \
+        echo "Creating entrypoint.sh from template..." && \
+        mkdir -p /app/scripts && \
+        cat > /app/scripts/entrypoint.sh << 'EOF' \
+#!/bin/sh
+set -eu
+
+# Download Asripa model if not present
+ASRIPA_MODEL_DIR="/app/models/fine_tuned_vit"
+ASRIPA_MODEL_ID="${ASRIPA_MODEL_ID:-HimAJ/asripa-emotion-detection}"
+
+if [ -n "$ASRIPA_MODEL_ID" ] && [ ! -f "$ASRIPA_MODEL_DIR/model.safetensors" ]; then
+  echo "📥 Downloading Asripa model from HuggingFace..."
+  echo "   Model ID: $ASRIPA_MODEL_ID"
+  mkdir -p "$ASRIPA_MODEL_DIR"
+  
+  python3 -c "
+from huggingface_hub import snapshot_download
+import os
+import sys
+try:
+    snapshot_download(
+        repo_id='$ASRIPA_MODEL_ID',
+        local_dir='$ASRIPA_MODEL_DIR',
+        local_dir_use_symlinks=False
+    )
+    print('✅ Asripa model downloaded successfully!')
+except Exception as e:
+    print(f'⚠️  Failed to download Asripa model: {e}')
+    print('   App will use base model only')
+    import shutil
+    if os.path.exists('$ASRIPA_MODEL_DIR'):
+        shutil.rmtree('$ASRIPA_MODEL_DIR')
+    sys.exit(0)
+" || {
+    echo "⚠️  Asripa model download skipped"
+    rm -rf "$ASRIPA_MODEL_DIR" 2>/dev/null || true
+  }
+elif [ -f "$ASRIPA_MODEL_DIR/model.safetensors" ]; then
+  echo "✅ Asripa model already present"
+fi
+
+# Start app on port 7860
+PORT="${PORT:-7860}"
+echo "Starting gunicorn on 0.0.0.0:${PORT}"
+export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
+exec gunicorn main:app --bind 0.0.0.0:"${PORT}" --workers 1 --threads 1 --timeout 120 --worker-class gthread
+EOF
         chmod +x /app/scripts/entrypoint.sh; \
     fi
 
+# Verify entrypoint exists
+RUN ls -la /app/scripts/entrypoint.sh || (echo "ERROR: entrypoint.sh not found!" && exit 1)
+
 # Hugging Face Spaces uses port 7860
 EXPOSE 7860
 
-# Use entrypoint script - this will download models and start the app
-ENTRYPOINT ["/app/scripts/entrypoint.sh"]
+# Use entrypoint script
+ENTRYPOINT ["/app/scripts/entrypoint.sh"]