wip deepseek train

scripts/container.sh

@@ -1,4 +1,4 @@
 #!/bin/bash
 
 container_id=$(docker ps -q)
-docker exec -it $container_id bash
+docker exec -it $container_id bash -l

scripts/restart.sh

@@ -1,11 +1,31 @@
 #!/bin/bash
 
+set -e
+
+echo "=== Preparing SIA environment ==="
+
+echo "Installing ITB tool..."
+/root/venvs/itb/bin/pip install -e /root/sia/tools/itb/ || echo "Warning: Failed to install ITB tool"
+
+echo "Installing Train tool..."
+/root/venvs/train/bin/pip install -e /root/sia/tools/train/ || echo "Warning: Failed to install Train tool"
+
+echo "Installing SIA core..."
+/root/venvs/sia/bin/pip install -e /root/sia/ || { echo "Error: Failed to install SIA core"; exit 1; }
+
+echo "=== Starting SIA ==="
+
 while true; do
     sia
-    if [ $? -eq 42 ]; then
-        echo "SIA exited with code 42. Restarting."
+    EXIT_CODE=$?
+    
+    if [ $EXIT_CODE -eq 42 ]; then
+        echo "SIA exited with code 42. Restarting in 2 seconds..."
+        sleep 2
     else
-        echo "SIA exited with code $?. Not restarting."
+        echo "SIA exited with code $EXIT_CODE. Not restarting."
         break
     fi
-done
+done
+
+exit $EXIT_CODE