Updated start.sh for detailed debugging and robust file saving

start.sh

@@ -1,4 +1,5 @@
 #!/bin/bash
+set -eux # This makes the script exit immediately on any command failure and prints commands as they execute. Very useful for debugging!
 
 # Set environment variables for a single GPU on Hugging Face Spaces
 export CUDA_VISIBLE_DEVICES="0"
@@ -7,15 +8,17 @@ export PYTHONPATH="/app"
 
 echo "Starting GEN3C application on A100 Large 80GB GPU..."
 
-# Create the /data directory if it doesn't exist (it should be mounted, but good practice)
+# Create the /data/output directory for persistent storage
 mkdir -p /data/output
 
-# Run your Python script, saving the output video to /data/output/
-# IMPORTANT: Ensure your script can handle an absolute path for --video_save_name
+# Define a temporary output path for the video within the /app directory (container's working dir)
+TEMP_VIDEO_PATH="/app/temp_test_single_image.mp4"
+
+# Run your Python script. It will now try to save the video to the temporary path.
 python cosmos_predict1/diffusion/inference/gen3c_single_image.py \
   --checkpoint_dir checkpoints \
   --input_image_path assets/diffusion/000000.png \
-  --video_save_name /data/output/test_single_image.mp4 \ # <-- CHANGED PATH HERE
+  --video_save_name "$TEMP_VIDEO_PATH" \ 
   --guidance 1 \
   --foreground_masking \
   --offload_diffusion_transformer \
@@ -26,5 +29,17 @@ python cosmos_predict1/diffusion/inference/gen3c_single_image.py \
   --disable_guardrail \
   --disable_prompt_encoder
 
-# IMPORTANT: Your Space will still stop after this script finishes,
-# but the generated video will now be saved in /data/output/ and persist.
+# --- Post-execution checks ---
+# Check if the temporary video file was actually created by the Python script
+if [ -f "$TEMP_VIDEO_PATH" ]; then
+    echo "SUCCESS: Temporary video file was created at: $TEMP_VIDEO_PATH"
+    # If created, move it to the persistent storage location
+    mv "$TEMP_VIDEO_PATH" /data/output/test_single_image.mp4
+    echo "SUCCESS: Video moved to /data/output/test_single_image.mp4"
+else
+    echo "ERROR: Temporary video file NOT found at $TEMP_VIDEO_PATH after Python script execution."
+    echo "This indicates the Python script likely failed to generate the video."
+    exit 1 # Exit with an error code to make the Space red if the video isn't produced
+fi
+
+echo "Script finished successfully. Generated video is in /data/output/. Container is exiting."