callGemini.js

import mime from 'mime-types'
import predictFolder from './ai.js';
import chalk from 'chalk';
import moveFile from './moveFile.js';

export default async function callGemini(change, path) {
    const mimeType = mime.lookup(path)
    console.log(`\nFile ${path} has been ${change}ed\nMime type : ${mimeType}`);
    if (change === 'add' || change === 'chang') {
        const result = await predictFolder(mimeType, path);
        if (result.status === 'success'){
            // console.log(`Prediction data ${result.data}`);
            const {folder, confidence, suggestedName, reason} = JSON.parse(result.data);
            console.log((`Suggested folder name ${folder} and file name ${suggestedName} with confidence ${confidence}\nReasoning:${reason}`));
            await moveFile(confidence, path, folder, suggestedName);
        }
        else
            console.log(chalk.red(`Invalid file type : ${result.data}`));
    }
}