Robustly locate and activate conda environment in start.sh

README.md

@@ -1,5 +1,5 @@
 ---
-title: GEN3C Project (from DGX Station) 
+title: GEN3C Project (built from existing Docker image) 
 emoji: 🫁
 colorFrom: green
 colorTo: blue

start.sh

@@ -1,22 +1,47 @@
 #!/bin/bash
-set -eux # Keep this for debugging and seeing command execution
+set -eux # Keep this for detailed logging
 
 export CUDA_VISIBLE_DEVICES="0"
 export CUDA_HOME="/usr/local/cuda"
 
-# Source conda.sh to make 'conda' command available
-# This finds the conda.sh script, which sets up the conda command in the shell.
-# We try a few common installation paths for Miniconda/Anaconda.
-source /opt/conda/etc/profile.d/conda.sh || \
-source /usr/local/miniconda/etc/profile.d/conda.sh || \
-source /root/miniconda/etc/profile.d/conda.sh || \
-{ echo "ERROR: Could not find conda.sh script to source. Ensure Conda is installed correctly."; exit 1; }
+echo "Attempting to locate and activate Conda environment..."
+
+# 1. Try to find the 'conda' executable in the system's PATH
+CONDA_EXEC=$(which conda)
+
+if [ -z "$CONDA_EXEC" ]; then
+    echo "ERROR: 'conda' executable not found in PATH."
+    echo "This means Conda might not be installed in the base image, or it's not configured in the system's PATH."
+    echo "Please check if 'elungky/gen3c:latest' image actually contains a working Conda installation."
+    exit 1
+fi
+
+echo "Found 'conda' executable at: $CONDA_EXEC"
+
+# 2. Derive the base Conda installation path from the executable's location.
+# For example, if CONDA_EXEC is /opt/conda/bin/conda, CONDA_BASE_PATH will be /opt/conda.
+CONDA_BASE_PATH=$(dirname $(dirname "$CONDA_EXEC"))
+echo "Derived Conda base path: $CONDA_BASE_PATH"
+
+# 3. Construct the path to conda.sh script based on the derived base path
+CONDA_SH_PATH="$CONDA_BASE_PATH/etc/profile.d/conda.sh"
+
+if [ -f "$CONDA_SH_PATH" ]; then
+    echo "Found conda.sh at: $CONDA_SH_PATH"
+    # Attempt to source it. Capture potential errors from sourcing itself.
+    source "$CONDA_SH_PATH" || { echo "ERROR: Failed to source conda.sh script at $CONDA_SH_PATH. Check permissions."; exit 1; }
+else
+    echo "ERROR: conda.sh not found at expected location derived from 'conda' executable: $CONDA_SH_PATH"
+    echo "This indicates an unusual Conda installation or structure within the base image."
+    exit 1
+fi
+
+echo "Conda environment initialized successfully."
 
 # Activate the specific conda environment
-conda activate cosmos-predict1 || { echo "ERROR: Failed to activate conda environment 'cosmos-predict1'. Ensure it exists."; exit 1; }
+conda activate cosmos-predict1 || { echo "ERROR: Failed to activate conda environment 'cosmos-predict1'. Ensure it exists and is accessible for this user."; exit 1; }
 
-# Set PYTHONPATH *after* conda activation, as conda might modify PATH/PYTHONPATH itself.
-# This ensures local project paths are prioritized/available for your application's modules.
+# Set PYTHONPATH after conda activation, as conda might adjust PATH/PYTHONPATH internally.
 export PYTHONPATH="/app:/app/gui/api" 
 
 echo "Starting GEN3C FastAPI inference server..."
@@ -25,5 +50,4 @@ export GEN3C_CKPT_PATH="/app/checkpoints"
 export GEN3C_GPU_COUNT=1
 
 # Start the FastAPI server
-# It's crucial that 'uvicorn' is installed and available within the 'cosmos-predict1' environment.
 exec uvicorn gui.api.server:app --host 0.0.0.0 --port 7860 --proxy-headers