myavana-bot-test-core 1.0.3 → 1.0.5

package/.env

@@ -1,8 +1,10 @@
 # PostgreSQL
 DB_HOST=34.55.140.120
+MYAVANA_DB_HOST=34.60.246.147
 DB_USER=myavana
 DB_PASSWORD=higibertigibet
 DB_NAME=sienna-naturals-custom-bot
+MYAVANA_DB_NAME=myavana-dashboard-bot
 DB_PORT=5432
 
 # Redis

package/index.js

@@ -21,6 +21,7 @@ module.exports = {
     postResponseProcessing: require('./src/postResponseProcessing').postResponseProcessing,
     saveHairIssue: require('./src/hairIssues').saveHairIssue,
     getHairIssuesForUser: require('./src/hairIssues').getHairIssuesForUser,
+    postResponseProductCheck: require('./src/postResponseProcessing').postResponseProductCheck,
   
     // Utilities
     getAllProducts: require('./src/utils').getAllProducts,

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "myavana-bot-test-core",
-    "version": "1.0.3",
+    "version": "1.0.5",
     "description": "Shared bot functionality",
     "main": "index.js",
     "scripts": {

package/src/config.js

@@ -9,6 +9,13 @@ module.exports = {
         database: process.env.DB_NAME,
         port: parseInt(process.env.DB_PORT, 10) || 5432, // Default PostgreSQL port
     },
+    myavanaDbConfig: {
+        host: process.env.MYAVANA_DB_HOST,
+        user: process.env.DB_USER,
+        password: process.env.DB_PASSWORD,
+        database: process.env.MYAVANA_DB_NAME,
+        port: parseInt(process.env.DB_PORT, 10) || 5432, // Default PostgreSQL port
+    },
     redisConfig: {
         username: process.env.REDIS_USERNAME,
         password: process.env.REDIS_PASSWORD,

package/src/postResponseProcessing.js

@@ -5,10 +5,11 @@ const DatabaseSessionStore = require("./session");
 const { pgClient, redisClient } = require('./database');
 const {generateAndSaveSummary, saveConversationSummary, saveConversation} = require("./conversation");
 const {saveHairIssue} = require("./hairIssues");
+const {getProductByName, addProductRecommendation} = require("./utils");
 
 const postProcessingPrompt = (chatHistory, message) => {
     return `
-    You are a Sienna Naturals expert AI assistant designed to analyze user conversations and user's name and extract structured information related to hair care.
+    You are a hair expert AI assistant designed to analyze user conversations and user's name and extract structured information related to hair care.
 
     **Task:** Analyze the following conversation history and the user's latest message to:
 
@@ -222,7 +223,7 @@ const postProcessConversation = async (chatHistory, message) => {
 
 
     const extractionPrompt = await postProcessingPrompt(chatHistory, message);
-    console.log('Extraction prompt: ', extractionPrompt);
+    //console.log('Extraction prompt: ', extractionPrompt);
     try {
         const chat = ai.chat({
             model: gemini15Flash,
@@ -232,7 +233,7 @@ const postProcessConversation = async (chatHistory, message) => {
 
         });
         const { output } = await chat.send(extractionPrompt);
-        console.log("Raw post-processing output:", JSON.stringify(output));
+        //console.log("Raw post-processing output:", JSON.stringify(output));
 
         if (!output) {
             console.warn("Post-processing output is empty. Returning default empty JSON.");
@@ -313,10 +314,9 @@ const updateUserProfile = async (userId, updates) => {
     }
 };
 
-const postResponseProcessing = async (postProcessData, userId, output, allchatsSummary, conversationId, chatHistory )=>{
+const postResponseProcessing = async (postProcessData, userId, output, allchatsSummary, conversationId, chatHistory, botMessage )=>{
     if (postProcessData.user_profile_updates) {
-        // TODO: Update user profile in database based on postProcessData.user_profile_updates
-        console.log("User profile updates to be applied:", postProcessData.user_profile_updates);
+        //console.log("User profile updates to be applied:", postProcessData.user_profile_updates);
         await updateUserProfile(userId, postProcessData.user_profile_updates);
     }
 
@@ -331,14 +331,94 @@ const postResponseProcessing = async (postProcessData, userId, output, allchatsS
     if (!allchatsSummary) {
         await generateAndSaveSummary(userId, conversationId, chatHistory);
     }
+    const recommendedProductsData = await postResponseProductCheck(chatHistory, botMessage); // Get array of product objects
+    //console.log("Recommended Products (with reasons):", recommendedProductsData);
+
+    if (recommendedProductsData && recommendedProductsData.length > 0) {
+        // --- Loop through recommended products and save to DB (Corrected for reasons) ---
+        for (const productRecommendation of recommendedProductsData) { // Loop through objects
+            const { productName, reason } = productRecommendation; // Extract productName and reason
+            const product = await getProductByName(productName); // Lookup product by name
+
+            if (product) {
+                await addProductRecommendation(userId, conversationId, product.product_id, reason); // Pass the reason to addProductRecommendation
+            } else {
+                console.warn(`Product "${productName}" not found in database, cannot save recommendation.`);
+            }
+        }
+    }
 
     await saveConversationSummary(conversationId, userId, allchatsSummary);
     return await saveConversation(conversationId, userId, chatHistory, allchatsSummary);
 };
 
+const postResponseProductCheckPrompt = async (chatHistory, message) => {
+    return `
+    Analyze the following chatbot conversation history and the bot's last response to identify if any specific products were recommended to the user.
+
+    Chat History:
+    ${JSON.stringify(chatHistory, null, 2)}
+
+    Bot's Last Response:
+    ${JSON.stringify(message, null, 2)}
+
+    Identify and list the names of the products recommended in the bot's last response AND provide a brief reason for each recommendation based on the conversation.
+    If no products were explicitly recommended, return an empty list.
+
+    Response Format:
+    [
+      { "productName": "Product Name 1", "reason": "Reason for recommendation 1" },
+      { "productName": "Product Name 2", "reason": "Reason for recommendation 2" },
+      ...
+    ] or [] if no products recommended.
+    `;
+};
+
+const postResponseProductCheck = async (chatHistory, message) => {
+    const extractionPrompt = await postResponseProductCheckPrompt(chatHistory, message);
+    //console.log('Extraction prompt for product check (with reason): ', extractionPrompt); // Updated log message
+    try {
+        const chat = ai.chat({
+            model: gemini15Flash, // or gemini20FlashExp
+            config: {
+                temperature: 0.9,
+            },
+        });
+        const { output } = await chat.send(extractionPrompt);
+        //console.log("Raw post-processing output for product check (with reason):", JSON.stringify(output)); // Updated log
+
+        if (!output) {
+            console.warn("Post-processing output for product check (with reason) is empty. Returning default empty array."); // Updated log
+            return [];
+        }
+
+        try {
+            // Attempt to parse the output as JSON (array of product objects)
+            const recommendedProductsData = JSON.parse(output); // Renamed variable
+            if (Array.isArray(recommendedProductsData)) {
+                return recommendedProductsData; // Return the array of product objects
+            } else {
+                console.warn("Post-processing output is not an array. Returning empty array.");
+                return [];
+            }
+        } catch (jsonError) {
+            console.error("Error parsing post-processing JSON output:", jsonError);
+            console.warn("Returning default empty array due to JSON parsing error.");
+            return [];
+        }
+
+    } catch (error) {
+        console.error("Error in postResponseProductCheck:", error);
+        return [];
+    }
+};
+
+
 module.exports = {
     postResponseProcessing,
     postProcessConversation,
     processBotResponse,
-    updateUserProfile
+    updateUserProfile,
+    postResponseProductCheckPrompt,
+    postResponseProductCheck
 }

package/src/utils.js

@@ -54,7 +54,32 @@ async function getHairIssuesForUser(userId){
   //return empty for now
   return []
 }
+async function getProductByName(productName) {
+  try {
+    const result = await pgClient.query('SELECT product_id FROM products WHERE product_name = $1', [productName]);
+    return result.rows[0]; // Returns the product object, or undefined if not found
+  } catch (error) {
+    console.error("Error fetching product by name:", error);
+    return null; // or undefined, depending on your preference
+  }
+}
 
+// Function to add product recommendation to the database
+async function addProductRecommendation(userId, conversationId, productId, reasonForRecommending) {
+  try {
+    const query = `
+            INSERT INTO product_recommendations (user_id, conversation_id, product_id, reason_for_recommending)
+            VALUES ($1, $2, $3, $4)
+            RETURNING *;
+        `;
+    const values = [userId, conversationId, productId, reasonForRecommending];
+    const result = await pgClient.query(query, values);
+    return { success: true, recommendation: result.rows[0] }; // Return success and the new recommendation
+  } catch (error) {
+    console.error("Error adding product recommendation:", error);
+    return { success: false, error: error.message };
+  }
+}
 
 module.exports = {
   getAllProducts,
@@ -62,5 +87,7 @@ module.exports = {
   getAllYoutubeVideos,
   getAllTestimonials,
   getAdditionalInstructions,
-  getHairIssuesForUser
+  getHairIssuesForUser,
+  getProductByName,
+  addProductRecommendation
 };