build time

App/Shaders/utils.py

@@ -1,7 +1,5 @@
 import os
-import os
 
-os.makedirs("/usr/local/lib/python3.10/site-packages/Workspace", exist_ok=True)
 os.environ["WINDOW_BACKEND"] = "headless"  # Use software rendering
 os.environ["SKIP_TORCH"] = "1"
 

Dockerfile

@@ -58,6 +58,16 @@ USER appuser
 COPY --chown=appuser . /srv
 
 # Command to run the application
-CMD uvicorn App.app:app --host 0.0.0.0 --port 7860 --workers 1
+# Create a startup script
+RUN echo '#!/bin/bash\n\
+    rm -rf /usr/local/lib/python3.10/site-packages/Workspace\n\
+    mkdir -p /usr/local/lib/python3.10/site-packages/Workspace\n\
+    ln -sf /home/admin/.local/share/BrokenSource/Broken /usr/local/lib/python3.10/site-packages/Workspace/Broken\n\
+    ln -sf /home/admin/.local/share/BrokenSource/DepthFlow /usr/local/lib/python3.10/site-packages/Workspace/DepthFlow\n\
+    uvicorn app:app --host 0.0.0.0 --port 8000\n'\
+    > /app/start.sh && chmod +x /app/start.sh
+
+# Run the startup script when the container launches
+CMD ["/app/start.sh"]
 # Expose the server port
 EXPOSE 7860