server.js

import process from "node:process";
import path from "node:path";
import express from "express";
import { execa } from "execa";
const app = express();
import { config } from "dotenv";
import { condaPython } from "./utils.js";

config();

const portArgIndex = process.argv.indexOf('--port');
let currentPort = 3000;

if (portArgIndex !== -1 && process.argv.length > portArgIndex + 1) {
  const parsedPort = parseInt(process.argv[portArgIndex + 1], 10);
  if (!isNaN(parsedPort)) {
    currentPort = parsedPort;
  }
}

app.use(express.json());

// Host local files
app.use("/uploads", express.static("results"));

// Host video player page
app.get("/files/:fileName", async (request, response) => {
  console.log(request.params);
  response.send(`<video controls src="/uploads/${request.params.fileName}"/>`);
});

/**
 * @typedef GenerateObject
 * @property {string} browser
 * @property {string} download
 * @property {string} fileName
 * @property {string} filePath
 */
/**
 *
 * @param {string} audio
 * @param {string} image
 * @param {number} [batchSize=2]
 * @returns {Promise<GenerateObject>}
 */
async function generate({ audio, image, batchSize = 2 }) {
  const pythonScript = "inference.py";
  const args = [
      "--driven_audio",
      audio,
      "--source_image",
      image,
      "--batch_size",
      batchSize
  ].join(' ');

  const command = condaPython('expresssadtalker', `${pythonScript} ${args}`);

  try {
      const result = await execa(command, {shell: true});

      console.log(result.stdout);
      console.log(result.stderr);
      const parts = result.stdout.split("\n");
      const relativePath = parts[3]
          .replace("The generated video is named:", "")
          .trim();
      const filePath = path.join(process.cwd(), relativePath);
      const fileName = relativePath.replace("./results\\", "");
      return {
          filePath,
          fileName,
          download: `http://127.0.0.1:${currentPort}/uploads/${fileName}`,
          browser: `http://127.0.0.1:${currentPort}/files/${fileName}`
      };
  } catch (error) {
      console.log(error);
      throw error;
  }
}


app.post("/generate", async (request, response) => {
  const { audio, image, batchSize } = request.body;
  console.log(request.body);
  try {
    const result = await generate({ audio, image, batchSize });
    response.status(201).json(result);
  } catch (error) {
    response.status(500).json({ message: error.message });
  }
});

app.listen(currentPort, async () => {
  console.log(`Server is running on port ${currentPort}`);
});