better organize vibeui

services/our/.dockerignore

@@ -1,6 +1,5 @@
-!dist/vibeui.pt
-
-runs
+dist
+vibeui
 venv
 src/test
 

services/our/.gitignore

@@ -1,4 +1,3 @@
-runs
 venv
 vibeui
 

services/our/Dockerfile

@@ -21,7 +21,8 @@ COPY --from=ipfs/kubo:v0.36.0 /usr/local/bin/ipfs /usr/local/bin/ipfs
 
 # Bundle the vibeui pytorch model
 RUN mkdir -p /app/dist/vibeui \
-  && wget -q https://gitea.futureporn.net/futureporn/fp/raw/branch/main/apps/vibeui/best.pt -O /app/dist/vibeui/vibeui.pt
+  && wget -q https://gitea.futureporn.net/futureporn/fp/raw/branch/main/apps/vibeui/dist/vibeui.pt -O /app/vibeui/vibeui.pt \
+  && wget -q https://gitea.futureporn.net/futureporn/fp/raw/branch/main/apps/vibeui/dist/data.yaml -O /app/vibeui/data.yaml
 
 # Copy and install dependencies
 COPY package.json package-lock.json ./

services/our/src/utils/vibeui.ts

@@ -139,7 +139,7 @@ export async function inference(videoFilePath: string): Promise<string> {
 
   // Generate a unique name based on video name + UUID
   const uniqueName = nanoid()
-  const customProjectDir = 'runs' // or any custom folder
+  const customProjectDir = 'vibeui/runs' // or any custom folder
   const outputPath = join(env.APP_DIR, customProjectDir, uniqueName)
 
   await spawn('./venv/bin/yolo', [