Update app.js

app.js

@@ -1,7 +1,7 @@
 import express from 'express';
 import cors from 'cors';
 import dotenv from 'dotenv';
-import mongoose from 'mongoose';
+import { createClient } from '@supabase/supabase-js';
 import { BedrockRuntimeClient, ConverseStreamCommand } from "@aws-sdk/client-bedrock-runtime";
 import { NodeHttpHandler } from "@smithy/node-http-handler";
 import path from 'path';
@@ -35,15 +35,11 @@ function getBedrockModelId(modelName) {
     }
 }
 
-// --- DB & MEMORY MANAGEMENT ---
-const ChatSchema = new mongoose.Schema({
-    id: String,
-    title: String,
-    totalTokens: { type: Number, default: 0 },
-    messages: Array, // Array of { role, content, reasoning }
-    updatedAt: { type: Date, default: Date.now }
-});
-const Chat = mongoose.model('Chat', ChatSchema);
+// --- DB & MEMORY MANAGEMENT (SUPABASE) ---
+const supabase = createClient(
+    process.env.SUPABASE_URL || '',
+    process.env.SUPABASE_KEY || ''
+);
 
 let memoryChats = {}; 
 let dirtyChats = new Set(); // Track chats that need DB syncing
@@ -51,30 +47,47 @@ let dirtyChats = new Set(); // Track chats that need DB syncing
 // Hydrate from DB on startup
 async function initDB() {
     try {
-        await mongoose.connect(process.env.MONGODB_URI);
-        console.log('✅ Connected to MongoDB');
-        const dbChats = await Chat.find();
-        dbChats.forEach(c => {
-            memoryChats[c.id] = { ...c.toObject(), isGenerating: false };
-        });
-        console.log(`✅ Hydrated ${dbChats.length} chats from DB.`);
+        const { data: dbChats, error } = await supabase.from('chats').select('*');
+        if (error) throw error;
+
+        console.log('✅ Connected to Supabase');
+        if (dbChats) {
+            dbChats.forEach(c => {
+                memoryChats[c.id] = { ...c, isGenerating: false };
+            });
+            console.log(`✅ Hydrated ${dbChats.length} chats from DB.`);
+        }
 
         // Sync to DB every 15 seconds (occasional sync)
         setInterval(async () => {
             if (dirtyChats.size === 0) return;
             const toSync = Array.from(dirtyChats);
             dirtyChats.clear();
-            for (const id of toSync) {
-                if (memoryChats[id]) {
-                    memoryChats[id].updatedAt = new Date();
-                    await Chat.findOneAndUpdate({ id }, memoryChats[id], { upsert: true });
+            
+            const rowsToUpsert = toSync.map(id => {
+                const chat = memoryChats[id];
+                chat.updatedAt = new Date().toISOString();
+                return {
+                    id: chat.id,
+                    title: chat.title,
+                    totalTokens: chat.totalTokens,
+                    messages: chat.messages,
+                    updatedAt: chat.updatedAt
+                };
+            });
+
+            if (rowsToUpsert.length > 0) {
+                const { error } = await supabase.from('chats').upsert(rowsToUpsert);
+                if (error) {
+                    console.error('❌ Supabase Sync Error:', error);
+                } else {
+                    console.log(`💾 Synced ${rowsToUpsert.length} chats to Supabase.`);
                 }
             }
-            console.log(`💾 Synced ${toSync.length} chats to MongoDB.`);
         }, 15000);
 
     } catch (err) {
-        console.error('❌ MongoDB Connection Error:', err);
+        console.error('❌ Supabase Initialization Error:', err);
     }
 }
 initDB();
@@ -105,7 +118,7 @@ app.post('/api/chats', (req, res) => {
         totalTokens: 0,
         messages:[],
         isGenerating: false,
-        updatedAt: new Date()
+        updatedAt: new Date().toISOString()
     };
     dirtyChats.add(newId);
     res.json(memoryChats[newId]);
@@ -116,7 +129,8 @@ app.delete('/api/chats/:id', async (req, res) => {
     const { id } = req.params;
     delete memoryChats[id];
     dirtyChats.delete(id);
-    await Chat.deleteOne({ id });
+    
+    await supabase.from('chats').delete().eq('id', id);
     res.json({ success: true });
 });