Update start.sh

start.sh

@@ -9,17 +9,10 @@ echo "User: $(whoami)"
 mkdir -p /data/ai-toolkit
     
 # Sync all files including frontend (but preserve node_modules and database)
-echo "Syncing application updates..."
+echo "Syncing codebase..."
 rsync -a --exclude='database' --exclude='aitk_db.db' --exclude='datasets' \
     /app/ai-toolkit/ /data/ai-toolkit/
-echo "✓ Application synced"
-    
-# Rebuild frontend if UI files were updated
-echo "Building frontend..."
-cd /data/ai-toolkit/ui
-npm run build
-cd -
-echo "✓ Frontend built"
+echo "✓ Codebase synced"
 
 # Ensure persistent directories exist
 echo "Creating persistent directories..."
@@ -48,6 +41,11 @@ echo "Changing to writable UI directory..."
 cd /data/ai-toolkit/ui
 echo "Current directory: $(pwd)"
 
+# Run database migrations/setup if needed
+echo "=== Updating dependencies ==="
+npm install
+echo "✓ Dependencies updated"
+
 # Run database migrations/setup if needed
 echo "=== Setting up database ==="
 
@@ -61,6 +59,11 @@ echo "Creating database tables..."
 npx prisma db push --schema prisma/schema.prisma --skip-generate
 echo "✓ Database tables created"
 
+# Build the app
+echo "=== Building app ==="
+npm run build
+echo "✓ App built"
+
 # Start the application with HuggingFace cache environment variables
 echo ""
 echo "=== Starting application ==="