Uploaded

server.js

@@ -50,18 +50,15 @@ async function updateSheetCache() {
             console.log('No data found in spreadsheet.');
             return;
         }
-
         rawSheetCache = rows;
         const headers = rows[0];
         const dataRows = rows.slice(1);
         let currentLocation = '';
         let currentOfficeExt = 'N/A';
         const allStaff = [];
-
         dataRows.forEach((row, rowIndex) => {
             const rowData = {};
             headers.forEach((header, index) => rowData[header] = row[index] || '');
-
             if (isValidName(rowData.Location)) {
                 currentLocation = rowData.Location;
                 if (currentLocation.toLowerCase().includes('liveoak')) {
@@ -73,27 +70,16 @@ async function updateSheetCache() {
             }
             rowData.Location = currentLocation;
             rowData.OfficeExtension = currentOfficeExt;
-
             const baseId = `r${rowIndex}`;
-            
-            if (isValidName(rowData.Provider)) {
-                allStaff.push({ ...rowData, id: `${baseId}-p`, role: 'Provider' });
-            }
-            if (isValidName(rowData.MA)) {
-                allStaff.push({ ...rowData, id: `${baseId}-ma`, role: 'Medical Assistant', Extension: rowData.Extension });
-            }
-            if (isValidName(rowData.VA)) {
-                allStaff.push({ ...rowData, id: `${baseId}-va`, role: 'Virtual Assistant', Extension: rowData.Ext });
-            }
-            if (isValidName(rowData['Team Leads/Manager'])) {
-                 allStaff.push({ ...rowData, id: `${baseId}-tl`, role: 'Team Lead / Manager', Extension: rowData['Ext/Phone#'] });
-            }
+            if (isValidName(rowData.Provider)) allStaff.push({ ...rowData, id: `${baseId}-p`, role: 'Provider' });
+            if (isValidName(rowData.MA)) allStaff.push({ ...rowData, id: `${baseId}-ma`, role: 'Medical Assistant', Extension: rowData.Extension });
+            if (isValidName(rowData.VA)) allStaff.push({ ...rowData, id: `${baseId}-va`, role: 'Virtual Assistant', Extension: rowData.Ext });
+            if (isValidName(rowData['Team Leads/Manager'])) allStaff.push({ ...rowData, id: `${baseId}-tl`, role: 'Team Lead / Manager', Extension: rowData['Ext/Phone#'] });
             if (isValidName(rowData['Ext/Phone#']) && /^[a-zA-Z]+-\s*\d+$/.test(rowData['Ext/Phone#'])) {
                 const [name, ext] = rowData['Ext/Phone#'].split('-');
                 allStaff.push({ ...rowData, id: `${baseId}-other`, role: 'Other Staff', OtherStaffName: name.trim(), Extension: ext.trim() });
             }
         });
-        
         processedStaffCache = allStaff;
         console.log('Cache updated. Processed records:', processedStaffCache.length);
     } catch (error) {
@@ -120,21 +106,15 @@ async function generateWithRetry(model, prompt, maxRetries = 3) {
     }
 }
 
-app.get('/', (req, res) => {
-    res.sendFile(path.join(__dirname, 'index.html'));
-});
+app.get('/', (req, res) => res.sendFile(path.join(__dirname, 'index.html')));
 
 app.get('/api/data', (req, res) => {
-    if (!processedStaffCache) {
-        return res.status(503).json({ message: 'Data is being fetched.' });
-    }
+    if (!processedStaffCache) return res.status(503).json({ message: 'Data is being fetched.' });
     res.json(processedStaffCache);
 });
 
 app.get('/api/raw-data', (req, res) => {
-    if (!rawSheetCache) {
-        return res.status(503).json({ message: 'Data is being fetched.' });
-    }
+    if (!rawSheetCache) return res.status(503).json({ message: 'Data is being fetched.' });
     res.json(rawSheetCache);
 });
 
@@ -147,6 +127,7 @@ app.get('/api/ask-luis', async (req, res) => {
     res.setHeader('Cache-Control', 'no-cache');
     res.setHeader('Connection', 'keep-alive');
     res.flushHeaders();
+
     const prompt = `
       You are "Luis", an AI data analyst for the Hillside Medical Group staff directory.
       Your primary function is to answer questions about staff based ONLY on the JSON data provided.
@@ -165,18 +146,28 @@ app.get('/api/ask-luis', async (req, res) => {
       **User's Question:**
       "${question}"
     `;
+
+    const AI_TIMEOUT = 20000; // 20 seconds
+    const timeoutPromise = new Promise((_, reject) => 
+        setTimeout(() => reject(new Error('AI_TIMEOUT')), AI_TIMEOUT)
+    );
+
     try {
-        const result = await generateWithRetry(model, prompt);
+        const resultPromise = generateWithRetry(model, prompt);
+        const result = await Promise.race([resultPromise, timeoutPromise]);
+        
         for await (const chunk of result.stream) {
             if (chunk.text()) {
                 res.write(`data: ${JSON.stringify({ chunk: chunk.text() })}\n\n`);
             }
         }
     } catch (error) {
-        const errorMessage = error.status === 503 
-            ? "The directory service is currently busy. Please try again in a few moments."
-            : "I'm having a bit of trouble connecting right now.";
-            
+        let errorMessage = "I'm having a bit of trouble connecting to my brain right now. Please try again in a moment.";
+        if (error.message === 'AI_TIMEOUT') {
+            errorMessage = "Sorry, my connection to the AI service timed out. The service may be busy. Please try your question again in a few moments.";
+        } else if (error.status === 503) {
+            errorMessage = "The directory AI service is currently busy. Please try again.";
+        }
         res.write(`data: ${JSON.stringify({ error: errorMessage })}\n\n`);
     } finally {
         res.write('data: [DONE]\n\n');