1
1
mirror of https://github.com/leon-ai/leon.git synced 2024-12-20 23:31:30 +03:00
leon/scripts/setup/setup.js
2021-03-02 17:02:55 +08:00

50 lines
1.3 KiB
JavaScript

import fs from 'fs'
import loader from '@/helpers/loader'
import log from '@/helpers/log'
import os from '@/helpers/os'
import train from '../train'
import setupDotenv from './setup-dotenv'
import setupCore from './setup-core'
import setupPackagesConfig from './setup-packages-config'
import setupPythonPackages from './setup-python-packages'
// Do not load ".env" file because it is not created yet
/**
* Main entry to setup Leon
*/
(async () => {
try {
const info = os.get()
// Required env vars to setup
process.env.LEON_LANG = 'en-US'
process.env.PIPENV_PIPFILE = 'bridges/python/Pipfile'
process.env.PIPENV_VENV_IN_PROJECT = 'true'
await setupDotenv()
loader.start()
await Promise.all([
setupCore(),
setupPackagesConfig()
])
await setupPythonPackages()
await train()
if (info.type === 'windows') {
log.info('Windows detected, reinjecting DeepSpeech into package.json...')
fs.unlinkSync('package.json')
fs.renameSync('package.json.backup', 'package.json')
log.success('DeepSpeech has been reinjected into package.json')
}
log.default('')
log.success('Hooray! Leon is installed and ready to go!')
loader.stop()
} catch (e) {
log.error(e)
loader.stop()
}
})()