start.sh

#!/bin/bash
set -eux # Keep this for debugging and seeing command execution

export CUDA_VISIBLE_DEVICES="0"
export CUDA_HOME="/usr/local/cuda"

echo "Activating Conda environment and starting server..."

# Activate the specific conda environment
# This now works because Conda is properly installed and in PATH by the Dockerfile
source $CONDA_DIR/etc/profile.d/conda.sh # Source the conda.sh for the shell
conda activate cosmos-predict1 # Activate the environment

# Set PYTHONPATH after conda activation for your specific project modules
export PYTHONPATH="/app:/app/gui/api" 

echo "Starting GEN3C FastAPI inference server..."

export GEN3C_CKPT_PATH="/app/checkpoints" # Assuming /app/checkpoints is still correct
export GEN3C_GPU_COUNT=1

# Start the FastAPI server
exec uvicorn gui.api.server:app --host 0.0.0.0 --port 7860 --proxy-headers