Update static/script.js

static/script.js

@@ -17,11 +17,12 @@ document.addEventListener('DOMContentLoaded', () => {
     const imagePreview = document.getElementById('image-preview');
     const removeImageBtn = document.getElementById('remove-image-btn');
 
-    // Add search and filter elements (assuming they exist in HTML or create them)
-    let searchInput, clearAllBtn;
+    // Add search and filter elements
+    let searchInput, clearAllBtn, userStatsDiv;
 
     // --- State ---
     let currentChatId = null;
+    let currentUserId = null;
     let conversationsCache = {};
     let selectedImageFile = null;
     let imgbbApiKey = '';
@@ -29,10 +30,106 @@ document.addEventListener('DOMContentLoaded', () => {
     let firebaseDB = null;
     let useFirebase = false;
 
+    // --- User Management ---
+    const generateUserFingerprint = () => {
+        // Create a device fingerprint based on available browser info
+        const canvas = document.createElement('canvas');
+        const ctx = canvas.getContext('2d');
+        ctx.textBaseline = 'top';
+        ctx.font = '14px Arial';
+        ctx.fillText('Device fingerprint', 2, 2);
+        
+        const fingerprint = [
+            navigator.userAgent,
+            navigator.language,
+            navigator.platform,
+            navigator.cookieEnabled,
+            navigator.doNotTrack,
+            screen.width + 'x' + screen.height,
+            screen.colorDepth,
+            new Date().getTimezoneOffset(),
+            canvas.toDataURL(),
+            navigator.hardwareConcurrency || 'unknown',
+            navigator.deviceMemory || 'unknown'
+        ].join('|');
+        
+        // Create hash
+        return hashCode(fingerprint).toString();
+    };
+
+    const hashCode = (str) => {
+        let hash = 0;
+        for (let i = 0; i < str.length; i++) {
+            const char = str.charCodeAt(i);
+            hash = ((hash << 5) - hash) + char;
+            hash = hash & hash; // Convert to 32bit integer
+        }
+        return Math.abs(hash);
+    };
+
+    const initializeUser = async () => {
+        // Try to get existing user ID from storage
+        let userId = localStorage.getItem('easyfarms_user_id');
+        
+        if (!userId) {
+            // Generate new user ID based on device fingerprint
+            const deviceFingerprint = generateUserFingerprint();
+            const timestamp = Date.now();
+            const randomPart = Math.random().toString(36).substr(2, 8);
+            
+            userId = `user_${deviceFingerprint}_${timestamp}_${randomPart}`;
+            localStorage.setItem('easyfarms_user_id', userId);
+            
+            console.log('Generated new user ID:', userId);
+        } else {
+            console.log('Using existing user ID:', userId);
+        }
+        
+        currentUserId = userId;
+        
+        // Also save to Firebase if available
+        if (useFirebase && firebaseDB) {
+            try {
+                await saveUserData(userId, {
+                    created_at: new Date().toISOString(),
+                    device_fingerprint: generateUserFingerprint(),
+                    last_seen: new Date().toISOString()
+                });
+            } catch (error) {
+                console.error('Failed to save user data to Firebase:', error);
+            }
+        }
+        
+        return userId;
+    };
+
+    const saveUserData = async (userId, userData) => {
+        try {
+            if (useFirebase && firebaseDB) {
+                const { setDoc, doc } = await import('https://www.gstatic.com/firebasejs/9.0.0/firebase-firestore.js');
+                await setDoc(doc(firebaseDB, 'users', userId), {
+                    ...userData,
+                    updatedAt: new Date().toISOString()
+                });
+                console.log('User data saved to Firebase:', userId);
+            } else {
+                // Fallback to localStorage
+                const users = JSON.parse(localStorage.getItem('easyfarms_users') || '{}');
+                users[userId] = {
+                    ...userData,
+                    updatedAt: new Date().toISOString()
+                };
+                localStorage.setItem('easyfarms_users', JSON.stringify(users));
+                console.log('User data saved to localStorage:', userId);
+            }
+        } catch (error) {
+            console.error('Failed to save user data:', error);
+        }
+    };
+
     // --- Firebase Configuration ---
     const initializeFirebase = async () => {
         try {
-            // Check if Firebase is available and configured
             if (typeof firebase !== 'undefined' && window.firebaseConfig) {
                 const { initializeApp } = await import('https://www.gstatic.com/firebasejs/9.0.0/firebase-app.js');
                 const { getFirestore, collection, doc, setDoc, getDoc, getDocs, deleteDoc } = await import('https://www.gstatic.com/firebasejs/9.0.0/firebase-firestore.js');
@@ -53,32 +150,35 @@ document.addEventListener('DOMContentLoaded', () => {
         }
     };
 
-    // --- Storage Management ---
+    // --- Storage Management with User Isolation ---
     const saveSessionData = async (chatId, data) => {
         try {
             if (useFirebase && firebaseDB) {
                 const { setDoc, doc } = await import('https://www.gstatic.com/firebasejs/9.0.0/firebase-firestore.js');
-                await setDoc(doc(firebaseDB, 'chat_sessions', chatId), {
+                await setDoc(doc(firebaseDB, 'user_sessions', `${currentUserId}_${chatId}`), {
                     ...data,
+                    userId: currentUserId,
+                    chatId: chatId,
                     updatedAt: new Date().toISOString()
                 });
-                console.log('Session saved to Firebase:', chatId);
+                console.log('Session saved to Firebase:', chatId, 'for user:', currentUserId);
             } else {
-                // Fallback to localStorage
-                const sessions = JSON.parse(localStorage.getItem('easyfarms_sessions') || '{}');
-                sessions[chatId] = {
+                // Fallback to localStorage with user isolation
+                const userSessions = JSON.parse(localStorage.getItem(`easyfarms_sessions_${currentUserId}`) || '{}');
+                userSessions[chatId] = {
                     ...data,
+                    userId: currentUserId,
                     updatedAt: new Date().toISOString()
                 };
-                localStorage.setItem('easyfarms_sessions', JSON.stringify(sessions));
-                console.log('Session saved to localStorage:', chatId);
+                localStorage.setItem(`easyfarms_sessions_${currentUserId}`, JSON.stringify(userSessions));
+                console.log('Session saved to localStorage:', chatId, 'for user:', currentUserId);
             }
         } catch (error) {
             console.error('Failed to save session data:', error);
             // Fallback to localStorage
-            const sessions = JSON.parse(localStorage.getItem('easyfarms_sessions') || '{}');
-            sessions[chatId] = data;
-            localStorage.setItem('easyfarms_sessions', JSON.stringify(sessions));
+            const userSessions = JSON.parse(localStorage.getItem(`easyfarms_sessions_${currentUserId}`) || '{}');
+            userSessions[chatId] = data;
+            localStorage.setItem(`easyfarms_sessions_${currentUserId}`, JSON.stringify(userSessions));
         }
     };
 
@@ -86,7 +186,7 @@ document.addEventListener('DOMContentLoaded', () => {
         try {
             if (useFirebase && firebaseDB) {
                 const { getDoc, doc } = await import('https://www.gstatic.com/firebasejs/9.0.0/firebase-firestore.js');
-                const docRef = doc(firebaseDB, 'chat_sessions', chatId);
+                const docRef = doc(firebaseDB, 'user_sessions', `${currentUserId}_${chatId}`);
                 const docSnap = await getDoc(docRef);
                 if (docSnap.exists()) {
                     console.log('Session loaded from Firebase:', chatId);
@@ -94,42 +194,44 @@ document.addEventListener('DOMContentLoaded', () => {
                 }
             } else {
                 // Fallback to localStorage
-                const sessions = JSON.parse(localStorage.getItem('easyfarms_sessions') || '{}');
-                if (sessions[chatId]) {
+                const userSessions = JSON.parse(localStorage.getItem(`easyfarms_sessions_${currentUserId}`) || '{}');
+                if (userSessions[chatId]) {
                     console.log('Session loaded from localStorage:', chatId);
-                    return sessions[chatId];
+                    return userSessions[chatId];
                 }
             }
             return null;
         } catch (error) {
             console.error('Failed to load session data:', error);
             // Fallback to localStorage
-            const sessions = JSON.parse(localStorage.getItem('easyfarms_sessions') || '{}');
-            return sessions[chatId] || null;
+            const userSessions = JSON.parse(localStorage.getItem(`easyfarms_sessions_${currentUserId}`) || '{}');
+            return userSessions[chatId] || null;
         }
     };
 
     const loadAllSessions = async () => {
         try {
             if (useFirebase && firebaseDB) {
-                const { getDocs, collection } = await import('https://www.gstatic.com/firebasejs/9.0.0/firebase-firestore.js');
-                const querySnapshot = await getDocs(collection(firebaseDB, 'chat_sessions'));
+                const { getDocs, collection, query, where } = await import('https://www.gstatic.com/firebasejs/9.0.0/firebase-firestore.js');
+                const q = query(collection(firebaseDB, 'user_sessions'), where('userId', '==', currentUserId));
+                const querySnapshot = await getDocs(q);
                 const sessions = {};
                 querySnapshot.forEach((doc) => {
-                    sessions[doc.id] = doc.data();
+                    const data = doc.data();
+                    sessions[data.chatId] = data;
                 });
-                console.log('All sessions loaded from Firebase');
+                console.log('All sessions loaded from Firebase for user:', currentUserId);
                 return sessions;
             } else {
                 // Fallback to localStorage
-                const sessions = JSON.parse(localStorage.getItem('easyfarms_sessions') || '{}');
-                console.log('All sessions loaded from localStorage');
+                const sessions = JSON.parse(localStorage.getItem(`easyfarms_sessions_${currentUserId}`) || '{}');
+                console.log('All sessions loaded from localStorage for user:', currentUserId);
                 return sessions;
             }
         } catch (error) {
             console.error('Failed to load all sessions:', error);
             // Fallback to localStorage
-            return JSON.parse(localStorage.getItem('easyfarms_sessions') || '{}');
+            return JSON.parse(localStorage.getItem(`easyfarms_sessions_${currentUserId}`) || '{}');
         }
     };
 
@@ -137,13 +239,13 @@ document.addEventListener('DOMContentLoaded', () => {
         try {
             if (useFirebase && firebaseDB) {
                 const { deleteDoc, doc } = await import('https://www.gstatic.com/firebasejs/9.0.0/firebase-firestore.js');
-                await deleteDoc(doc(firebaseDB, 'chat_sessions', chatId));
+                await deleteDoc(doc(firebaseDB, 'user_sessions', `${currentUserId}_${chatId}`));
                 console.log('Session deleted from Firebase:', chatId);
             } else {
                 // Fallback to localStorage
-                const sessions = JSON.parse(localStorage.getItem('easyfarms_sessions') || '{}');
-                delete sessions[chatId];
-                localStorage.setItem('easyfarms_sessions', JSON.stringify(sessions));
+                const userSessions = JSON.parse(localStorage.getItem(`easyfarms_sessions_${currentUserId}`) || '{}');
+                delete userSessions[chatId];
+                localStorage.setItem(`easyfarms_sessions_${currentUserId}`, JSON.stringify(userSessions));
                 console.log('Session deleted from localStorage:', chatId);
             }
         } catch (error) {
@@ -160,11 +262,15 @@ document.addEventListener('DOMContentLoaded', () => {
 
         // Initialize Firebase
         firebase = await initializeFirebase();
+        
+        // Initialize user
+        await initializeUser();
 
         await fetchConfig();
         await loadCachedSessions();
         await renderChatHistoryFromAPI();
         setupSearchAndFilter();
+        setupUserStats();
 
         // Event Listeners
         sendBtn.addEventListener('click', sendMessage);
@@ -195,7 +301,10 @@ document.addEventListener('DOMContentLoaded', () => {
             }
         });
 
-        console.log('EasyFarms Chat initialized with new backend integration');
+        // Update last seen periodically
+        setInterval(updateLastSeen, 30000); // Every 30 seconds
+
+        console.log('EasyFarms Chat initialized with user authentication for user:', currentUserId);
     };
 
     const fetchConfig = async () => {
@@ -211,6 +320,63 @@ document.addEventListener('DOMContentLoaded', () => {
         }
     };
 
+    const setupUserStats = () => {
+        // Create user stats div
+        if (!userStatsDiv) {
+            userStatsDiv = document.createElement('div');
+            userStatsDiv.className = 'user-stats';
+            userStatsDiv.style.cssText = `
+                padding: 10px;
+                margin: 10px 0;
+                background: #f8f9fa;
+                border-radius: 4px;
+                font-size: 12px;
+                color: #666;
+                border-top: 1px solid #eee;
+            `;
+            
+            // Insert at the bottom of sidebar
+            chatHistoryList.parentNode.appendChild(userStatsDiv);
+        }
+        
+        updateUserStats();
+    };
+
+    const updateUserStats = async () => {
+        try {
+            const response = await fetch(`/users/${currentUserId}/stats`);
+            if (response.ok) {
+                const stats = await response.json();
+                userStatsDiv.innerHTML = `
+                    <div><strong>Your Stats</strong></div>
+                    <div>Chats: ${stats.total_sessions || 0}</div>
+                    <div>Messages: ${stats.total_messages || 0}</div>
+                    <div>Recent: ${stats.recent_sessions_24h || 0}</div>
+                `;
+            }
+        } catch (error) {
+            console.error('Failed to update user stats:', error);
+            // Show basic local stats
+            const sessions = await loadAllSessions();
+            const sessionCount = Object.keys(sessions).length;
+            userStatsDiv.innerHTML = `
+                <div><strong>Local Stats</strong></div>
+                <div>Chats: ${sessionCount}</div>
+                <div>User: ${currentUserId.substr(0, 12)}...</div>
+            `;
+        }
+    };
+
+    const updateLastSeen = async () => {
+        try {
+            await saveUserData(currentUserId, {
+                last_seen: new Date().toISOString()
+            });
+        } catch (error) {
+            console.error('Failed to update last seen:', error);
+        }
+    };
+
     const setupSearchAndFilter = () => {
         // Create search input if it doesn't exist
         if (!searchInput) {
@@ -227,14 +393,13 @@ document.addEventListener('DOMContentLoaded', () => {
                 font-size: 14px;
             `;
             
-            // Insert before chat history list
             chatHistoryList.parentNode.insertBefore(searchInput, chatHistoryList);
         }
 
         // Create clear all button if it doesn't exist
         if (!clearAllBtn) {
             clearAllBtn = document.createElement('button');
-            clearAllBtn.textContent = 'Clear All Chats';
+            clearAllBtn.textContent = 'Clear All My Chats';
             clearAllBtn.className = 'clear-all-btn';
             clearAllBtn.style.cssText = `
                 width: 100%;
@@ -248,7 +413,6 @@ document.addEventListener('DOMContentLoaded', () => {
                 font-size: 12px;
             `;
             
-            // Insert after search input
             searchInput.parentNode.insertBefore(clearAllBtn, chatHistoryList);
         }
 
@@ -271,32 +435,29 @@ document.addEventListener('DOMContentLoaded', () => {
     };
 
     const confirmClearAllChats = () => {
-        if (confirm('Are you sure you want to delete all chats? This action cannot be undone.')) {
+        if (confirm('Are you sure you want to delete all your chats? This action cannot be undone.')) {
             clearAllChats();
         }
     };
 
     const clearAllChats = async () => {
         try {
-            // Get all chat IDs
+            // Delete all user chats from backend
+            const response = await fetch(`/users/${currentUserId}/chats`, {
+                method: 'DELETE'
+            });
+            
+            if (response.ok) {
+                console.log('All chats deleted from backend for user:', currentUserId);
+            } else {
+                console.error('Failed to delete all chats from backend');
+            }
+            
+            // Clear local storage
             const allSessions = await loadAllSessions();
             const chatIds = Object.keys(allSessions);
             
-            // Delete each chat from backend
             for (const chatId of chatIds) {
-                try {
-                    const response = await fetch(`/chat/${chatId}`, {
-                        method: 'DELETE'
-                    });
-                    
-                    if (response.ok) {
-                        console.log('Deleted chat from backend:', chatId);
-                    }
-                } catch (error) {
-                    console.error('Failed to delete chat from backend:', chatId, error);
-                }
-                
-                // Delete from local storage
                 await deleteSessionData(chatId);
             }
             
@@ -304,21 +465,14 @@ document.addEventListener('DOMContentLoaded', () => {
             conversationsCache = {};
             chatHistoryList.innerHTML = '';
             startNewChat();
+            updateUserStats();
             
-            console.log('All chats cleared successfully');
+            console.log('All chats cleared successfully for user:', currentUserId);
         } catch (error) {
             console.error('Failed to clear all chats:', error);
         }
     };
-    // Generate or get user ID (could be device-based or login-based)
-    const getUserId = () => {
-        let userId = localStorage.getItem('easyfarms_user_id');
-        if (!userId) {
-            userId = 'user_' + Date.now() + '_' + Math.random().toString(36).substr(2, 9);
-            localStorage.setItem('easyfarms_user_id', userId);
-        }
-        return userId;
-    };
+
     // --- Core Chat Functions ---
     const sendMessage = async () => {
         const messageText = messageInput.value.trim();
@@ -343,18 +497,18 @@ document.addEventListener('DOMContentLoaded', () => {
                 permanentImageUrl = await uploadImageToImgBB(selectedImageFile); 
             }
 
-            // Prepare request data
+            // Prepare request data with user ID
             const requestData = {
                 query: messageText,
                 session_id: currentChatId,
-                user_id: getUserId() // Add this
+                user_id: currentUserId
             };
 
             if (permanentImageUrl) {
                 requestData.image_url = permanentImageUrl;
             }
 
-            // Send to new API endpoint
+            // Send to API endpoint
             const response = await fetch('/chat', {
                 method: 'POST',
                 headers: {
@@ -395,6 +549,9 @@ document.addEventListener('DOMContentLoaded', () => {
                 data.is_new_chat
             );
 
+            // Update stats
+            updateUserStats();
+
         } catch (error) {
             console.error('Error sending message:', error);
             
@@ -437,14 +594,14 @@ document.addEventListener('DOMContentLoaded', () => {
     // --- Chat Management ---
     const startNewChat = () => {
         currentChatId = null;
-        chatMessages.innerHTML = `<div class="welcome-message"><h1>EasyFarms Assistant</h1></div>`;
+        chatMessages.innerHTML = `<div class="welcome-message"><h1>EasyFarms Assistant</h1><p>User: ${currentUserId.substr(0, 16)}...</p></div>`;
         chatTitle.textContent = "New Chat";
         updateActiveChatItem();
-        console.log('Started new chat');
+        console.log('Started new chat for user:', currentUserId);
     };
 
     const switchChat = async (chatId) => {
-        console.log('Switching to chat:', chatId);
+        console.log('Switching to chat:', chatId, 'for user:', currentUserId);
         currentChatId = chatId;
         chatMessages.innerHTML = '';
 
@@ -461,7 +618,7 @@ document.addEventListener('DOMContentLoaded', () => {
             });
             
             try {
-                const response = await fetch(`/chat/${chatId}/messages`);
+                const response = await fetch(`/chat/${chatId}/messages?user_id=${currentUserId}`);
                 
                 if (!response.ok) {
                     throw new Error(`HTTP error! status: ${response.status}`);
@@ -505,7 +662,7 @@ document.addEventListener('DOMContentLoaded', () => {
         
         try {
             // Delete from backend
-            const response = await fetch(`/chat/${chatId}`, {
+            const response = await fetch(`/chat/${chatId}?user_id=${currentUserId}`, {
                 method: 'DELETE'
             });
             
@@ -530,6 +687,7 @@ document.addEventListener('DOMContentLoaded', () => {
             }
             
             console.log('Chat deleted successfully:', chatId);
+            updateUserStats();
             
         } catch (error) {
             console.error('Error deleting chat:', error);
@@ -546,14 +704,15 @@ document.addEventListener('DOMContentLoaded', () => {
             conversationsCache[chatId] = { 
                 title, 
                 messages: [],
-                created_at: new Date().toISOString()
+                created_at: new Date().toISOString(),
+                user_id: currentUserId
             };
             
             // Add to UI
             const item = createChatHistoryItem(chatId, title);
             chatHistoryList.prepend(item);
             
-            console.log('Created new chat:', chatId);
+            console.log('Created new chat:', chatId, 'for user:', currentUserId);
         }
         
         // Create message objects with IDs
@@ -575,6 +734,7 @@ document.addEventListener('DOMContentLoaded', () => {
         // Add to cache
         conversationsCache[chatId].messages.push(userMessage, assistantMessage);
         conversationsCache[chatId].updated_at = new Date().toISOString();
+        conversationsCache[chatId].user_id = currentUserId;
         
         // Save to storage
         await saveSessionData(chatId, conversationsCache[chatId]);
@@ -582,7 +742,7 @@ document.addEventListener('DOMContentLoaded', () => {
         updateActiveChatItem();
         chatTitle.textContent = conversationsCache[chatId].title;
         
-        console.log('Updated cache for chat:', chatId);
+        console.log('Updated cache for chat:', chatId, 'user:', currentUserId);
     };
 
     const createChatHistoryItem = (chatId, title) => {
@@ -629,7 +789,7 @@ document.addEventListener('DOMContentLoaded', () => {
     const loadCachedSessions = async () => {
         try {
             conversationsCache = await loadAllSessions();
-            console.log('Loaded cached sessions:', Object.keys(conversationsCache).length);
+            console.log('Loaded cached sessions for user:', currentUserId, 'Count:', Object.keys(conversationsCache).length);
         } catch (error) {
             console.error('Failed to load cached sessions:', error);
             conversationsCache = {};
@@ -692,14 +852,14 @@ document.addEventListener('DOMContentLoaded', () => {
     // --- API Integration ---
     const renderChatHistoryFromAPI = async () => {
         try {
-            const response = await fetch('/chats');
+            const response = await fetch(`/chats?user_id=${currentUserId}`);
             
             if (!response.ok) {
                 throw new Error(`HTTP error! status: ${response.status}`);
             }
             
             const sessions = await response.json();
-            console.log('Loaded chat sessions from API:', sessions.length);
+            console.log('Loaded chat sessions from API for user:', currentUserId, sessions.length);
             
             chatHistoryList.innerHTML = '';
             
@@ -715,6 +875,7 @@ document.addEventListener('DOMContentLoaded', () => {
                 conversationsCache[session.session_id].message_count = session.message_count;
                 conversationsCache[session.session_id].created_at = session.created_at;
                 conversationsCache[session.session_id].updated_at = session.updated_at;
+                conversationsCache[session.session_id].user_id = currentUserId;
                 
                 // Create UI item
                 const item = createChatHistoryItem(session.session_id, session.title);
@@ -723,18 +884,21 @@ document.addEventListener('DOMContentLoaded', () => {
             
             // Save updated cache
             for (const [chatId, data] of Object.entries(conversationsCache)) {
-                await saveSessionData(chatId, data);
+                if (data.user_id === currentUserId) {
+                    await saveSessionData(chatId, data);
+                }
             }
             
             updateActiveChatItem();
+            updateUserStats();
             
         } catch (error) { 
             console.error("Failed to render chat history from API:", error); 
             
             // Fallback to local cache
-            console.log('Falling back to local cache...');
+            console.log('Falling back to local cache for user:', currentUserId);
             for (const [chatId, data] of Object.entries(conversationsCache)) {
-                if (data.title) {
+                if (data.title && data.user_id === currentUserId) {
                     const item = createChatHistoryItem(chatId, data.title);
                     chatHistoryList.appendChild(item);
                 }
@@ -750,6 +914,241 @@ document.addEventListener('DOMContentLoaded', () => {
 
     const closeSidebar = () => appContainer.classList.remove('sidebar-visible');
 
-    // --- Start the Application ---
-    init();
+    // --- User Account Management ---
+    const exportUserData = async () => {
+        try {
+            const response = await fetch(`/users/${currentUserId}/export`);
+            if (response.ok) {
+                const userData = await response.json();
+                const dataStr = JSON.stringify(userData, null, 2);
+                const dataUri = 'data:application/json;charset=utf-8,'+ encodeURIComponent(dataStr);
+                
+                const exportFileDefaultName = `easyfarms_data_${currentUserId.substr(0, 8)}_${new Date().toISOString().split('T')[0]}.json`;
+                
+                const linkElement = document.createElement('a');
+                linkElement.setAttribute('href', dataUri);
+                linkElement.setAttribute('download', exportFileDefaultName);
+                linkElement.click();
+                
+                console.log('User data exported successfully');
+            } else {
+                throw new Error('Export failed');
+            }
+        } catch (error) {
+            console.error('Failed to export user data:', error);
+            alert('Failed to export data. Please try again.');
+        }
+    };
+
+    const deleteUserAccount = async () => {
+        if (!confirm('Are you sure you want to delete your account? This will permanently delete all your chats and data.')) {
+            return;
+        }
+        
+        if (!confirm('This action cannot be undone. Type "DELETE" to confirm:') || 
+            prompt('Type "DELETE" to confirm account deletion:') !== 'DELETE') {
+            return;
+        }
+        
+        try {
+            const response = await fetch(`/users/${currentUserId}`, {
+                method: 'DELETE'
+            });
+            
+            if (response.ok) {
+                // Clear all local data
+                localStorage.removeItem('easyfarms_user_id');
+                localStorage.removeItem(`easyfarms_sessions_${currentUserId}`);
+                localStorage.removeItem(`easyfarms_users`);
+                
+                // Clear Firebase data if applicable
+                if (useFirebase && firebaseDB) {
+                    try {
+                        const { deleteDoc, doc, collection, getDocs, query, where } = await import('https://www.gstatic.com/firebasejs/9.0.0/firebase-firestore.js');
+                        
+                        // Delete user document
+                        await deleteDoc(doc(firebaseDB, 'users', currentUserId));
+                        
+                        // Delete user sessions
+                        const q = query(collection(firebaseDB, 'user_sessions'), where('userId', '==', currentUserId));
+                        const querySnapshot = await getDocs(q);
+                        querySnapshot.forEach(async (doc) => {
+                            await deleteDoc(doc.ref);
+                        });
+                        
+                    } catch (firebaseError) {
+                        console.error('Failed to clean Firebase data:', firebaseError);
+                    }
+                }
+                
+                alert('Account deleted successfully. Refreshing page...');
+                location.reload();
+                
+            } else {
+                throw new Error('Account deletion failed');
+            }
+            
+        } catch (error) {
+            console.error('Failed to delete account:', error);
+            alert('Failed to delete account. Please try again.');
+        }
+    };
+
+    const createUserMenu = () => {
+        // Create user menu button
+        const userMenuBtn = document.createElement('button');
+        userMenuBtn.innerHTML = '⚙️';
+        userMenuBtn.title = 'User Settings';
+        userMenuBtn.style.cssText = `
+            position: fixed;
+            top: 10px;
+            right: 10px;
+            background: #007bff;
+            color: white;
+            border: none;
+            border-radius: 50%;
+            width: 40px;
+            height: 40px;
+            cursor: pointer;
+            font-size: 16px;
+            z-index: 1000;
+        `;
+        
+        // Create user menu dropdown
+        const userMenu = document.createElement('div');
+        userMenu.style.cssText = `
+            position: fixed;
+            top: 55px;
+            right: 10px;
+            background: white;
+            border: 1px solid #ddd;
+            border-radius: 4px;
+            padding: 10px;
+            box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+            display: none;
+            z-index: 1001;
+            min-width: 200px;
+        `;
+        
+        userMenu.innerHTML = `
+            <div style="margin-bottom: 10px; font-weight: bold;">User Settings</div>
+            <div style="margin-bottom: 5px; font-size: 12px; color: #666;">ID: ${currentUserId.substr(0, 16)}...</div>
+            <button id="export-data-btn" style="width: 100%; margin: 5px 0; padding: 5px; background: #28a745; color: white; border: none; border-radius: 3px; cursor: pointer;">Export My Data</button>
+            <button id="delete-account-btn" style="width: 100%; margin: 5px 0; padding: 5px; background: #dc3545; color: white; border: none; border-radius: 3px; cursor: pointer;">Delete Account</button>
+        `;
+        
+        document.body.appendChild(userMenuBtn);
+        document.body.appendChild(userMenu);
+        
+        // Toggle menu
+        userMenuBtn.addEventListener('click', (e) => {
+            e.stopPropagation();
+            userMenu.style.display = userMenu.style.display === 'none' ? 'block' : 'none';
+        });
+        
+        // Close menu when clicking outside
+        document.addEventListener('click', () => {
+            userMenu.style.display = 'none';
+        });
+        
+        // Menu actions
+        document.getElementById('export-data-btn').addEventListener('click', exportUserData);
+        document.getElementById('delete-account-btn').addEventListener('click', deleteUserAccount);
+    };
+
+    // --- Enhanced Error Handling ---
+    window.addEventListener('unhandledrejection', (event) => {
+        console.error('Unhandled promise rejection:', event.reason);
+        // Don't show error to user for now, just log it
+    });
+
+    window.addEventListener('error', (event) => {
+        console.error('Global error:', event.error);
+        // Don't show error to user for now, just log it
+    });
+
+    // --- Session Recovery ---
+    const attemptSessionRecovery = async () => {
+        try {
+            // Try to recover any unsaved messages
+            const unsavedMessages = localStorage.getItem(`unsaved_messages_${currentUserId}`);
+            if (unsavedMessages) {
+                const messages = JSON.parse(unsavedMessages);
+                console.log('Found unsaved messages, attempting recovery:', messages);
+                
+                // Display recovered messages
+                messages.forEach(msg => {
+                    displayMessage({
+                        role: msg.role,
+                        content: `[RECOVERED] ${msg.content}`,
+                        isError: true
+                    });
+                });
+                
+                // Clear unsaved messages
+                localStorage.removeItem(`unsaved_messages_${currentUserId}`);
+            }
+        } catch (error) {
+            console.error('Session recovery failed:', error);
+        }
+    };
+
+    const saveUnsavedMessage = (role, content) => {
+        try {
+            const unsavedMessages = JSON.parse(localStorage.getItem(`unsaved_messages_${currentUserId}`) || '[]');
+            unsavedMessages.push({
+                role,
+                content,
+                timestamp: new Date().toISOString()
+            });
+            
+            // Keep only last 5 unsaved messages
+            if (unsavedMessages.length > 5) {
+                unsavedMessages.splice(0, unsavedMessages.length - 5);
+            }
+            
+            localStorage.setItem(`unsaved_messages_${currentUserId}`, JSON.stringify(unsavedMessages));
+        } catch (error) {
+            console.error('Failed to save unsaved message:', error);
+        }
+    };
+
+    // --- Performance Monitoring ---
+    const logPerformance = () => {
+        if (performance && performance.timing) {
+            const perfData = performance.timing;
+            const loadTime = perfData.loadEventEnd - perfData.navigationStart;
+            console.log(`Page load time: ${loadTime}ms`);
+            
+            // Log to backend if needed
+            // fetch('/analytics/performance', {
+            //     method: 'POST',
+            //     body: JSON.stringify({ user_id: currentUserId, load_time: loadTime })
+            // });
+        }
+    };
+
+    // --- Initialize Everything ---
+    // Add user menu after initialization
+    const setupUserInterface = () => {
+        createUserMenu();
+        attemptSessionRecovery();
+        logPerformance();
+    };
+
+    // Start the Application
+    init().then(() => {
+        setupUserInterface();
+        console.log('EasyFarms Chat fully initialized for user:', currentUserId);
+    }).catch((error) => {
+        console.error('Failed to initialize application:', error);
+        document.body.innerHTML = `
+            <div style="padding: 20px; text-align: center;">
+                <h2>Failed to Load EasyFarms Chat</h2>
+                <p>Please refresh the page and try again.</p>
+                <p style="color: #666; font-size: 12px;">Error: ${error.message}</p>
+            </div>
+        `;
+    });
+
 });