server.js

// server.js
require("dotenv").config();
const express = require("express");
const path = require("path");
const mongoose = require("mongoose"); // NUEVO: Importar mongoose
const predictRoutes = require("./routes/predictRoutes");
const { initModel } = require("./services/tfModelService");

const PORT = process.env.PORT || 3002;

const MONGO_URI = process.env.MONGO_URI || "mongodb://localhost:27017/prediction";

const app = express();
app.use(express.json());

// Servir la carpeta del modelo TFJS
const modelDir = path.resolve(__dirname, "model");
app.use("/model", express.static(modelDir));

// Rutas
app.use("/", predictRoutes);


app.listen(PORT, async () => {
  const serverUrl = `http://localhost:${PORT}`;
  console.log(`[PREDICT] Servicio escuchando en ${serverUrl}`);

  
  try {
    await mongoose.connect(MONGO_URI);
    console.log("[PREDICT] Conectado a MongoDB correctamente");
  } catch (err) {
    console.error("[PREDICT] Error conectando a MongoDB:", err);

  }

  // Carga del modelo IA
  try {
    await initModel(serverUrl);
  } catch (err) {
    console.error("Error al inicializar modelo:", err);
    process.exit(1);
  }
});