server.js

import express from "express";
import cors from "cors";

const app = express();
const PORT = process.env.PORT || 3000;

app.use(cors());
app.use(express.json({ limit: "2mb" }));
app.use(express.static("public"));

app.get("/health", (req, res) => res.json({ ok: true }));

app.post("/api/generate", async (req, res) => {
  try {
    const { prompt, duration = 5 } = req.body || {};
    if (!prompt) return res.status(400).json({ error: "Missing prompt." });

    const HF_TOKEN = process.env.HF_TOKEN;
    const HF_ENDPOINT = process.env.HF_ENDPOINT || "https://api-inference.huggingface.co/models/damo-vilab/text-to-video-ms-1.7b";

    if (!HF_TOKEN) {
      return res.status(500).json({ error: "Missing HF_TOKEN environment variable on server." });
    }

    const hfResponse = await fetch(HF_ENDPOINT, {
      method: "POST",
      headers: {
        "Authorization": `Bearer ${HF_TOKEN}`,
        "Content-Type": "application/json"
      },
      body: JSON.stringify({
        inputs: prompt,
        parameters: {
          num_frames: Math.min(64, Math.max(16, Number(duration) * 8 || 32))
        }
      })
    });

    const contentType = hfResponse.headers.get("content-type") || "";

    if (!hfResponse.ok) {
      const text = await hfResponse.text();
      return res.status(hfResponse.status).json({ error: "Hugging Face failed.", details: text });
    }

    if (contentType.includes("application/json")) {
      const data = await hfResponse.json();
      return res.json({ status: "json_response", data });
    }

    const arrayBuffer = await hfResponse.arrayBuffer();
    const buffer = Buffer.from(arrayBuffer);

    res.setHeader("Content-Type", contentType || "video/mp4");
    res.setHeader("Content-Disposition", "inline; filename=command-cinema-output.mp4");
    res.send(buffer);
  } catch (err) {
    res.status(500).json({ error: "Server error.", details: err.message });
  }
});

app.listen(PORT, () => console.log(`Command Cinema Runtime running on port ${PORT}`));