Update start_server.sh

start_server.sh

@@ -3,12 +3,13 @@ JUPYTER_TOKEN="${JUPYTER_TOKEN:=huggingface}"
 
 # Define directories
 NOTEBOOK_DIR="/data"
-ORIGINAL_NOTEBOOKS="/home/user/app/notebooks"  # This is the correct path where notebooks are located
+ORIGINAL_NOTEBOOKS="/home/user/app/notebooks"  # Path where notebooks are located
+ORIGINAL_IMAGES="/home/user/app/notebooks/images"  # Path where images are located
 
-echo "Starting notebook setup process..."
-echo "Looking for notebooks in: $ORIGINAL_NOTEBOOKS"
+echo "Starting notebook and images setup process..."
 
-# List all notebook files in the source directory
+# Set up notebooks
+echo "Looking for notebooks in: $ORIGINAL_NOTEBOOKS"
 find "$ORIGINAL_NOTEBOOKS" -name "*.ipynb" -type f 2>/dev/null || echo "Cannot find notebooks in $ORIGINAL_NOTEBOOKS"
 
 # Check if we can find the notebooks
@@ -22,18 +23,45 @@ if ls "$ORIGINAL_NOTEBOOKS"/*.ipynb 1> /dev/null 2>&1; then
     
     # Copy notebooks to data directory
     cp -v "$ORIGINAL_NOTEBOOKS"/*.ipynb "$NOTEBOOK_DIR/" || echo "Failed to copy notebooks"
-    
-    # Set proper permissions
-    chmod -R 755 "$NOTEBOOK_DIR"
-    chown -R $(whoami) "$NOTEBOOK_DIR" 2>/dev/null || echo "Could not change ownership"
 else
     echo "No notebooks found in expected location. Checking all directories..."
-    find / -name "*.ipynb" -not -path "*/\.*" 2>/dev/null | head -20
+    find / -name "*.ipynb" -not -path "*/\.*" 2>/dev/null | head -10
+fi
+
+# Set up images directory
+echo "Looking for images in: $ORIGINAL_IMAGES"
+IMAGES_DIR="$NOTEBOOK_DIR/images"
+
+# Create images directory in data
+mkdir -p "$IMAGES_DIR"
+rm -f "$IMAGES_DIR"/*.png "$IMAGES_DIR"/*.jpg "$IMAGES_DIR"/*.jpeg "$IMAGES_DIR"/*.gif
+
+# Check if original images directory exists
+if [ -d "$ORIGINAL_IMAGES" ]; then
+    echo "Found images directory at $ORIGINAL_IMAGES"
+    
+    # Copy all image files to the images directory in data
+    find "$ORIGINAL_IMAGES" -type f \( -name "*.png" -o -name "*.jpg" -o -name "*.jpeg" -o -name "*.gif" \) -exec cp -v {} "$IMAGES_DIR/" \; || echo "Failed to copy images"
+else
+    echo "No images directory found at $ORIGINAL_IMAGES. Checking alternatives..."
+    # Try to find images in other possible locations
+    if [ -d "/home/user/app/images" ]; then
+        echo "Found images in /home/user/app/images"
+        find "/home/user/app/images" -type f \( -name "*.png" -o -name "*.jpg" -o -name "*.jpeg" -o -name "*.gif" \) -exec cp -v {} "$IMAGES_DIR/" \; || echo "Failed to copy images"
+    else
+        echo "No images directory found. Creating an empty one."
+    fi
 fi
 
+# Set proper permissions
+chmod -R 755 "$NOTEBOOK_DIR"
+chown -R $(whoami) "$NOTEBOOK_DIR" 2>/dev/null || echo "Could not change ownership"
+
 # Show what we've got
 echo "Contents of notebook directory ($NOTEBOOK_DIR):"
 ls -la "$NOTEBOOK_DIR"
+echo "Contents of images directory ($IMAGES_DIR):"
+ls -la "$IMAGES_DIR"
 
 # Start JupyterLab
 echo "Starting JupyterLab..."