emergency fix

package.json

@@ -7,6 +7,7 @@
     "start": "node --loader ts-node/esm src/index.mts",
     "test": "node --loader ts-node/esm src/test.mts",
     "postprod:interpolate": "node --loader ts-node/esm src/postprod/interpolate.mts",
+    "postprod:interpolate_legacy": "node --loader ts-node/esm src/postprod/interpolate_legacy.mts",
     "docker": "npm run docker:build && npm run docker:run",
     "docker:build": "docker build -t  media-server .",
     "docker:run": "docker run -it -p 7860:7860 media-server"

scripts/interpolate.sh

@@ -23,7 +23,7 @@ for FILE_PATH in $INPUT_DIR*.mp4; do
             # The first attempt has failed, retrying
             echo "Attempt 1 failed for $FILE_NAME, retrying..."
 
-            if ! npm run postprod:interpolate "$FILE_PATH" "$TEMP_FILE_PATH"
+            if ! npm run postprod:interpolate_legacy "$FILE_PATH" "$TEMP_FILE_PATH"
             then
                 # Both attempts have failed, skipping the file
                 echo "Both attempts failed for $FILE_NAME, skipping the file..."

src/postprod/InterpolateVideoCLI_legacy.mts

@@ -0,0 +1,53 @@
+import fs from 'node:fs'
+
+import puppeteer from 'puppeteer'
+
+// TODO we should use an inference endpoint instead
+export async function interpolateVideo(inputFilePath: string, outputFilePath: string) {
+  
+  const browser = await puppeteer.launch({
+    headless: true,
+    protocolTimeout: 400000,
+  })
+
+  const spaceUrl = process.env.WEBTV_INTERPOLATION_SPACE_URL
+
+  const page = await browser.newPage()
+  await page.goto(spaceUrl, { waitUntil: 'networkidle2' })
+  
+  await new Promise(r => setTimeout(r, 3000))
+
+  const fileField = await page.$('input[type=file]')
+
+  // console.log(`uploading file..`)
+  await fileField.uploadFile(inputFilePath)
+
+  // console.log('looking for the button to submit')
+  const submitButton = await page.$('button.lg')
+
+  // console.log('clicking on the button')
+  await submitButton.click()
+
+  await page.waitForSelector('a[download="interpolated_result.mp4"]', {
+    timeout: 400000, // need to be large enough in case someone else attemps to use our space
+  })
+
+  const interpolatedFileUrl = await page.$$eval('a[download="interpolated_result.mp4"]', el => el.map(x => x.getAttribute("href"))[0])
+
+  // console.log('downloading file from space..')
+  console.log(`- downloading from ${interpolatedFileUrl}`)
+
+
+  // download the video
+  const response = await fetch(interpolatedFileUrl)
+  // write it to the disk
+  const arrayBuffer = await response.arrayBuffer()
+  
+  await fs.promises.writeFile(
+    outputFilePath,
+    Buffer.from(arrayBuffer)
+  )
+  console.log('success! wrote to:', outputFilePath)
+
+  process.exit()
+}

src/postprod/interpolate.mts

@@ -1,7 +1,7 @@
 import { interpolateVideo } from "./interpolateVideoCLI.mts"
 
 // to test this file:
-//  npm run postprod:interpolate sandbox/video/1688471841394.mp4 sandbox/video/output.mp4
+//  npm run postprod:interpolate sandbox/video/raccoon.mp4 sandbox/video/output.mp4
 
 const inputFilePath = process.argv[2]
 const outputFilePath = process.argv[3]

src/postprod/interpolateVideoCLI.mts

@@ -25,7 +25,7 @@ export const interpolateVideo = async (filePath: string, outputFilePath: string)
   const result = await app.predict(1, [
     blob, 	// blob in 'parameter_5' Video component		
     1, // number (numeric value between 1 and 4) in 'Interpolation Steps' Slider component		
-    8, // string  in 'FPS output' Radio component
+    24, // string  in 'FPS output' Radio component
   ])
 
 

src/postprod/interpolate_legacy.mts

@@ -0,0 +1,12 @@
+import { interpolateVideo } from "./InterpolateVideoCLI_legacy.mts"
+
+// to test this file:
+//  npm run postprod:interpolate sandbox/video/1688471841394.mp4 sandbox/video/output.mp4
+
+const inputFilePath = process.argv[2]
+const outputFilePath = process.argv[3]
+
+console.log('input file path:', inputFilePath)
+console.log('output file path:', outputFilePath)
+
+await interpolateVideo(inputFilePath, outputFilePath)