myavana-bot-test-core 1.0.2 → 1.0.4

package/.env

@@ -0,0 +1,19 @@
+# PostgreSQL
+DB_HOST=34.55.140.120
+DB_USER=myavana
+DB_PASSWORD=higibertigibet
+DB_NAME=sienna-naturals-custom-bot
+DB_PORT=5432
+
+# Redis
+REDIS_HOST=redis-15330.c1.us-central1-2.gce.redns.redis-cloud.com
+REDIS_PORT=15330
+REDIS_PASSWORD=OyLRjkTPuCGzQ2mNzA3Uhx5HIqWza8QC
+REDIS_USERNAME=default
+
+#GENKIT
+GENKIT_API_KEY=AIzaSyBB5ZYwktOFI3R3j_vs8U7CxwKgS3XNgM0
+
+#OpenAI
+OPENAI_API_KEY=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
+

package/index.js

@@ -21,8 +21,12 @@ 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,
     getAllFaqs: require('./src/utils').getAllFaqs,
+    getAllYoutubeVideos: require('./src/utils').getAllYoutubeVideos,
+    getAllTestimonials: require('./src/utils').getAllTestimonials,
+    getAdditionalInstructions: require('./src/utils').getAdditionalInstructions
   };

package/package.json

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

package/src/postResponseProcessing.js

@@ -5,6 +5,7 @@ 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 `
@@ -313,7 +314,7 @@ 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);
@@ -331,14 +332,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

@@ -1,4 +1,5 @@
-const { pgClient } = require('./database'); // Import pgClient from core's database.js
+// app/api/utils.js
+const { pgClient } = require('./database'); // Or your DB connection
 
 async function getAllProducts() {
   try {
@@ -20,13 +21,73 @@ async function getAllFaqs() {
   }
 }
 
-async function getHairIssuesForUser(userId) {
-  // Return empty for now (or implement logic later)
-  return [];
+async function getAllYoutubeVideos() { // New function to get ALL YouTube Videos
+  try {
+    const result = await pgClient.query('SELECT * FROM youtube_videos');
+    return result.rows;
+  } catch (error) {
+    console.error("Error fetching all YouTube videos:", error);
+    return [];
+  }
+}
+
+async function getAllTestimonials() { // New function to get ALL Testimonials
+  try {
+    const result = await pgClient.query('SELECT * FROM testimonials');
+    return result.rows;
+  } catch (error) {
+    console.error("Error fetching all testimonials:", error);
+    return [];
+  }
+}
+
+async function getAdditionalInstructions() { // New function to get ALL Bot Instructions
+  try {
+    const result = await pgClient.query('SELECT * FROM bot_instructions');
+    return result.rows;
+  } catch (error) {
+    console.error("Error fetching bot instructions:", error);
+    return [];
+  }
+}
+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,
   getAllFaqs,
+  getAllYoutubeVideos,
+  getAllTestimonials,
+  getAdditionalInstructions,
   getHairIssuesForUser,
+  getProductByName,
+  addProductRecommendation
 };