#!/bin/bash
set -eux

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

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

# Activate the specific conda environment
# Use '. ' (dot space) instead of '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"
export GEN3C_GPU_COUNT=1

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