ai.js

import { GoogleGenAI, Type } from "@google/genai";
import * as fs from 'fs';
import chalk from "chalk";
import dotenv from "dotenv"

dotenv.config()

const ai = new GoogleGenAI({
    apiKey: process.env.GEMINI_API_KEY
});


export default async function predictFolder(mimeType, path) {
    try {
        console.log(chalk.green('Gemini background job started'));
        const file = {
            inlineData: {
                mimeType: mimeType,
                data: Buffer.from(fs.readFileSync(path)).toString("base64"),
            },
        };
        const foldersFile = {
            inlineData: {
                mimeType: "text/plain",
                data: Buffer.from(fs.readFileSync("folders.txt")).toString("base64"),
            },
        };
        const contents = [
            file,
            foldersFile,
        ]
        const response = await ai.models.generateContent({
            model: "gemini-2.5-flash",
            contents: contents,
            config: {
                systemInstruction: `
        You are an AI file organizer.
        Your task is to read and interpret the content of a PDF document and determine
        the most suitable folder location for storing it from the provided list of possible folder locations
        (contained in folders_hierarchy.json).

        Guidelines:
        1) Carefully extract and analyze the text and metadata from the PDF.
        2) Identify the main subject, context, or category of the document.
        3) Select the single most appropriate folder path only from the provided list.
        4) If the content is ambiguous, choose the closest logical folder.
        5) Never invent content or folder names.
        6) Strictly return the result as a JSON object with exactly three fields:
           'folder' (string, chosen folder path),
           'confidence' (integer between 1 and 100),
           "suggestedName": "string, a clean descriptive filename WITHOUT extension,
                    must be under 7 words, use _ instead of spaces, 
                    keep original if already short and clear.       
                    Do not include file extensions like .pdf, .docx, .png etc. 
                    If the current name is already suitable, keep it as is (without extension).",
           'reason' (string, justify your choice shortly).
      `
                ,
                responseMimeType: "application/json",
                responseSchema: {
                    folder: Type.STRING,
                    confidence: Type.INTEGER,
                    suggestedName: Type.STRING,
                    reason: Type.STRING
                },
            },
        });
        return {
            status: 'success',
            data: response.text
        };
    } catch (err) {
        return {
            status: 'error',
            data: err.message
        };
    }
}