Upload 2 files

Dockerfile

@@ -43,8 +43,9 @@ WORKDIR /app
 # Copy application code
 COPY app/ ./app/
 COPY server.py ./
-# Note: app.py exists but we use server.py as entry point
-# If app.py exists, it won't interfere since CMD specifies server:app
+# Copy startup script
+COPY start.sh ./start.sh
+RUN chmod +x ./start.sh
 
 # Create necessary directories
 RUN mkdir -p /app/models
@@ -56,8 +57,7 @@ EXPOSE 7860
 ENV PYTHONUNBUFFERED=1
 ENV SAM2_BUILD_CUDA=0
 
-# Run the FastAPI application
+# Run the FastAPI application using startup script
 # Hugging Face Spaces sets PORT environment variable automatically (usually 7860)
-# The application will listen on 0.0.0.0 to accept external connections
-CMD python -c "import os; port = int(os.environ.get('PORT', 7860)); import uvicorn; uvicorn.run('server:app', host='0.0.0.0', port=port, log_level='info')"
+CMD ["./start.sh"]
 

start.sh

@@ -0,0 +1,19 @@
+#!/bin/bash
+set -e
+
+# Get port from environment variable (Hugging Face Spaces sets this)
+PORT=${PORT:-7860}
+
+echo "Starting FastAPI application on port $PORT..."
+
+# Verify critical imports before starting
+echo "Verifying imports..."
+python -c "import sys; sys.path.insert(0, '/app'); import server; print('✓ server.py imported successfully')" || {
+    echo "ERROR: Failed to import server.py"
+    exit 1
+}
+
+# Run the FastAPI application
+echo "Starting uvicorn server..."
+exec python -m uvicorn server:app --host 0.0.0.0 --port $PORT --log-level info
+