const { DataTypes } = require('sequelize');
const sequelize = require('../database/database.js');

const Training_Project = sequelize.define('LabelStudioProject', {
  project_id: {
    type: DataTypes.INTEGER,
    primaryKey: true,
    unique: true,
    allowNull: false,
    autoIncrement: true,
  },
  title:{
    type: DataTypes.STRING,
    allowNull: false,
  },
  description: {
    type: DataTypes.STRING,
  },
  classes: {
    type: DataTypes.JSON,
    allowNull: false,
  },
  project_image: {
    type: DataTypes.BLOB,
  },
  project_image_type: {
    type: DataTypes.STRING,
    allowNull: true,
  }

}, {
  tableName: 'training_project',
  timestamps: false,
});

module.exports = Training_Project;