local-deep-research 0.4.4__py3-none-any.whl → 0.5.0__py3-none-any.whl

local_deep_research/web/static/js/components/detail.js

@@ -1,348 +1,714 @@
 /**
- * Detail Component
- * Manages the display of detailed information for a specific research topic
+ * Research Details Component
+ * Manages the display of research details and token metrics for a specific research
  */
 (function() {
     // DOM Elements
-    let detailContainer = null;
-    let sourcesList = null;
-    let tabsContainer = null;
-    
+    let queryElement = null;
+    let statusElement = null;
+    let modeElement = null;
+    let progressBar = null;
+    let progressPercentage = null;
+    let pollInterval = null;
+
+    // Token metrics elements
+    let timelineChart = null;
+
     // Component state
     let currentResearchId = null;
-    let currentTopicId = null;
-    let detailData = null;
-    
+    let isCompleted = false;
+
     /**
-     * Initialize the detail component
+     * Initialize the research details component
      */
-    function initializeDetail() {
-        // Get IDs from URL
-        const ids = getIdsFromUrl();
-        currentResearchId = ids.researchId;
-        currentTopicId = ids.topicId;
-        
-        if (!currentResearchId || !currentTopicId) {
-            console.error('No research or topic ID found');
+    async function initializeResearchDetails() {
+        console.log('Initializing research details component...');
+
+        // Get research ID from URL
+        currentResearchId = getResearchIdFromUrl();
+
+        if (!currentResearchId) {
+            console.error('No research ID found in URL');
             if (window.ui && window.ui.showError) {
-                window.ui.showError('Invalid research or topic. Please return to the results page.');
+                window.ui.showError('No research ID found. Please return to the history page.');
             }
             return;
         }
-        
+
+        console.log('Research ID:', currentResearchId);
+
         // Get DOM elements
-        detailContainer = document.getElementById('research-log');
-        
-        if (!detailContainer) {
-            console.error('Required DOM elements not found for detail component');
-            return;
-        }
-        
+        queryElement = document.getElementById('detail-query');
+        statusElement = document.getElementById('detail-status');
+        modeElement = document.getElementById('detail-mode');
+        progressBar = document.getElementById('detail-progress-fill');
+        progressPercentage = document.getElementById('detail-progress-percentage');
+
+        console.log('DOM elements found, setting up event listeners...');
+
         // Set up event listeners
         setupEventListeners();
-        
-        // Load topic detail
-        loadTopicDetail();
-        
-        console.log('Detail component initialized for research:', currentResearchId, 'topic:', currentTopicId);
+
+        // Load initial research details
+        await loadResearchDetails();
+
+        // Start polling for updates (only if needed)
+        startPolling();
+
+        console.log('Research details component initialized');
     }
-    
+
     /**
-     * Extract research and topic IDs from URL
-     * @returns {Object} Object with researchId and topicId
+     * Extract research ID from URL
+     * @returns {string|null} Research ID or null if not found
      */
-    function getIdsFromUrl() {
+    function getResearchIdFromUrl() {
         const pathParts = window.location.pathname.split('/');
-        const detailIndex = pathParts.indexOf('detail');
-        
-        if (detailIndex > 0 && detailIndex + 2 < pathParts.length) {
-            return {
-                researchId: pathParts[detailIndex + 1],
-                topicId: pathParts[detailIndex + 2]
-            };
+        const detailsIndex = pathParts.indexOf('details');
+
+        if (detailsIndex > 0 && detailsIndex + 1 < pathParts.length) {
+            return pathParts[detailsIndex + 1];
         }
-        
-        return { researchId: null, topicId: null };
+
+        return null;
     }
-    
+
     /**
      * Set up event listeners
      */
     function setupEventListeners() {
-        // Back button
+        const viewResultsButton = document.getElementById('view-results-btn');
+        if (viewResultsButton) {
+            viewResultsButton.addEventListener('click', function() {
+                window.location.href = `/research/results/${currentResearchId}`;
+            });
+        }
+
         const backButton = document.getElementById('back-to-history-from-details');
         if (backButton) {
             backButton.addEventListener('click', function() {
                 window.location.href = '/research/history';
             });
         }
-        
-        // Progress elements
-        const progressBar = document.getElementById('detail-progress-fill');
-        const progressPercentage = document.getElementById('detail-progress-percentage');
-        
-        // Tab click events
-        if (tabsContainer) {
-            tabsContainer.addEventListener('click', function(e) {
-                if (e.target && e.target.matches('.tab-item')) {
-                    const tabId = e.target.dataset.tab;
-                    switchTab(tabId);
-                }
-            });
-        }
-        
-        // Source highlight and citation copy
-        if (sourcesList) {
-            sourcesList.addEventListener('click', function(e) {
-                // Handle citation copy
-                if (e.target && e.target.matches('.copy-citation-btn')) {
-                    const sourceId = e.target.closest('.source-item').dataset.id;
-                    handleCopyCitation(sourceId);
-                }
-                
-                // Handle source highlighting
-                if (e.target && e.target.closest('.source-item')) {
-                    const sourceItem = e.target.closest('.source-item');
-                    if (!e.target.matches('.copy-citation-btn')) {
-                        toggleSourceHighlight(sourceItem);
-                    }
+    }
+
+    /**
+     * Load basic research details (for polling)
+     */
+    async function loadBasicResearchDetails() {
+        console.log('Loading basic research details...');
+
+        try {
+            const response = await fetch(`/research/api/research/${currentResearchId}`, {
+                headers: {
+                    'Accept': 'application/json'
                 }
             });
+
+            if (!response.ok) {
+                throw new Error(`Failed to load research details: ${response.statusText}`);
+            }
+
+            const data = await response.json();
+
+            // Update UI with research details
+            updateResearchInfo(data);
+
+            // Check if research is completed
+            isCompleted = data.status === 'Completed' || data.status === 'Failed';
+
+        } catch (error) {
+            console.error('Error loading basic research details:', error);
         }
     }
-    
+
     /**
-     * Load topic detail from API
+     * Load research details from API (full load including token metrics)
      */
-    async function loadTopicDetail() {
-        // Show loading state
-        window.ui.showSpinner(detailContainer, 'Loading topic details...');
-        
+    async function loadResearchDetails() {
+        console.log('Loading research details...');
+
         try {
-            // Get topic detail
-            detailData = await window.api.getTopicDetail(currentResearchId, currentTopicId);
-            
-            if (!detailData) {
-                throw new Error('No topic details found');
+            // Fetch research details
+            const response = await fetch(`/research/api/research/${currentResearchId}`, {
+                headers: {
+                    'Accept': 'application/json'
+                }
+            });
+
+            if (!response.ok) {
+                throw new Error(`Failed to load research details: ${response.statusText}`);
+            }
+
+            const data = await response.json();
+            console.log('Research details loaded:', data);
+
+            // Update UI with research details
+            updateResearchInfo(data);
+
+            // Check if research is completed
+            isCompleted = data.status === 'Completed' || data.status === 'Failed';
+            if (isCompleted && pollInterval) {
+                clearInterval(pollInterval);
+                pollInterval = null;
             }
-            
-            // Render detail
-            renderDetail(detailData);
+
+            // Load token metrics
+            await loadTokenMetrics();
+
+            // Load search metrics
+            await loadSearchMetrics();
+
         } catch (error) {
-            console.error('Error loading topic detail:', error);
-            window.ui.hideSpinner(detailContainer);
-            window.ui.showError('Error loading topic detail: ' + error.message);
+            console.error('Error loading research details:', error);
+            if (window.ui && window.ui.showError) {
+                window.ui.showError('Error loading research details: ' + error.message);
+            }
         }
     }
-    
+
     /**
-     * Render topic detail
-     * @param {Object} data - The topic detail data
+     * Update research info display
+     * @param {Object} data - Research data
      */
-    function renderDetail(data) {
-        // Hide spinner
-        window.ui.hideSpinner(detailContainer);
-        
-        // Set page title
-        document.title = `${data.title || 'Topic Detail'} - Local Deep Research`;
-        
-        // Update page header
-        const pageTitle = document.querySelector('h1.page-title');
-        if (pageTitle && data.title) {
-            pageTitle.textContent = data.title;
+    function updateResearchInfo(data) {
+        if (queryElement) {
+            queryElement.textContent = data.query || 'N/A';
         }
-        
-        // Render content
-        if (data.content) {
-            // Render markdown
-            const renderedHtml = window.ui.renderMarkdown(data.content);
-            detailContainer.innerHTML = renderedHtml;
-            
-            // Add syntax highlighting
-            highlightCodeBlocks();
-        } else {
-            detailContainer.innerHTML = '<p class="text-error">No content available for this topic.</p>';
+
+        if (statusElement) {
+            statusElement.textContent = data.status || 'N/A';
+            statusElement.className = 'metadata-value status-' + (data.status || 'unknown').toLowerCase();
         }
-        
-        // Render sources
-        if (sourcesList && data.sources && data.sources.length > 0) {
-            renderSources(data.sources);
+
+        if (modeElement) {
+            modeElement.textContent = data.mode || 'standard';
         }
-        
-        // Set sources count
-        const sourcesCountEl = document.getElementById('sources-count');
-        if (sourcesCountEl && data.sources) {
-            sourcesCountEl.textContent = `${data.sources.length} source${data.sources.length !== 1 ? 's' : ''}`;
+
+        // Update progress
+        const progress = data.progress_percentage || 0;
+        if (progressBar) {
+            progressBar.style.width = progress + '%';
         }
-        
-        // Show first tab
-        if (tabsContainer) {
-            const firstTab = tabsContainer.querySelector('.tab-item');
-            if (firstTab) {
-                switchTab(firstTab.dataset.tab);
+        if (progressPercentage) {
+            progressPercentage.textContent = progress + '%';
+        }
+    }
+
+    /**
+     * Load search metrics for this research
+     */
+    async function loadSearchMetrics() {
+        console.log('Loading search metrics...');
+
+        try {
+            const response = await fetch(`/metrics/api/metrics/research/${currentResearchId}/search`, {
+                headers: {
+                    'Accept': 'application/json'
+                }
+            });
+
+            if (!response.ok) {
+                console.log('No search metrics available for this research');
+                return;
+            }
+
+            const data = await response.json();
+
+            if (data.status === 'success' && data.metrics) {
+                console.log('Search metrics loaded:', data.metrics);
+                displaySearchMetrics(data.metrics);
+                displaySearchEnginePerformance(data.metrics.engine_stats || []);
+                displaySearchTimeline(data.metrics.search_calls || []);
             }
+
+        } catch (error) {
+            console.error('Error loading search metrics:', error);
         }
     }
-    
+
     /**
-     * Render sources list
-     * @param {Array} sources - Array of source objects
+     * Display search metrics summary
+     * @param {Object} searchData - Search metrics data
      */
-    function renderSources(sources) {
-        sourcesList.innerHTML = '';
-        
-        sources.forEach((source, index) => {
-            const sourceEl = document.createElement('div');
-            sourceEl.className = 'source-item';
-            sourceEl.dataset.id = index;
-            
-            const title = source.title || source.url || `Source ${index + 1}`;
-            const url = source.url || '';
-            const author = source.author || '';
-            const date = source.date || '';
-            
-            sourceEl.innerHTML = `
-                <div class="source-header">
-                    <h4 class="source-title">${title}</h4>
-                    <button class="copy-citation-btn" title="Copy citation">
-                        <i class="fas fa-clipboard"></i>
-                    </button>
-                </div>
-                ${url ? `<div class="source-url"><a href="${url}" target="_blank">${url}</a></div>` : ''}
-                <div class="source-metadata">
-                    ${author ? `<span><i class="fas fa-user"></i> ${author}</span>` : ''}
-                    ${date ? `<span><i class="far fa-calendar"></i> ${date}</span>` : ''}
+    function displaySearchMetrics(searchData) {
+        const searchSummary = document.getElementById('search-summary');
+        if (!searchSummary || !searchData) return;
+
+        // Show the search analytics section
+        const searchSection = document.getElementById('search-metrics-section');
+        if (searchSection) {
+            searchSection.style.display = 'block';
+        }
+
+        const totalSearches = searchData.total_searches || 0;
+        const successRate = searchData.success_rate || 0;
+        const avgResponseTime = (searchData.avg_response_time || 0) / 1000; // Convert ms to seconds
+        const totalResults = searchData.total_results || 0;
+
+        // Update individual metric elements
+        const totalSearchesEl = document.getElementById('total-searches');
+        const totalResultsEl = document.getElementById('total-search-results');
+        const avgResponseTimeEl = document.getElementById('avg-search-response-time');
+        const successRateEl = document.getElementById('search-success-rate');
+
+        if (totalSearchesEl) totalSearchesEl.textContent = totalSearches;
+        if (totalResultsEl) totalResultsEl.textContent = formatNumber(totalResults);
+        if (avgResponseTimeEl) avgResponseTimeEl.textContent = avgResponseTime.toFixed(2) + 's';
+        if (successRateEl) successRateEl.textContent = successRate.toFixed(1) + '%';
+    }
+
+    /**
+     * Display search engine performance breakdown
+     * @param {Array} enginePerformance - Search engine performance data
+     */
+    function displaySearchEnginePerformance(enginePerformance) {
+        const engineContainer = document.getElementById('search-engine-performance');
+        if (!engineContainer) return;
+
+        if (!enginePerformance || enginePerformance.length === 0) {
+            engineContainer.innerHTML = '<p style="text-align: center; color: var(--text-secondary);">No search engine data available</p>';
+            return;
+        }
+
+        const engineHtml = enginePerformance.map(engine => `
+            <div class="search-engine-item">
+                <div class="engine-name">${escapeHtml(engine.engine)}</div>
+                <div class="engine-stats">
+                    <span class="stat">${engine.call_count} searches</span>
+                    <span class="stat">${engine.success_rate.toFixed(1)}% success</span>
+                    <span class="stat">${(engine.avg_response_time / 1000).toFixed(2)}s avg</span>
+                    <span class="stat">${formatNumber(engine.total_results)} results</span>
                 </div>
-                ${source.relevance ? `<div class="source-relevance">Relevance: ${source.relevance}</div>` : ''}
-            `;
-            
-            sourcesList.appendChild(sourceEl);
-        });
+            </div>
+        `).join('');
+
+        engineContainer.innerHTML = engineHtml;
     }
-    
+
     /**
-     * Switch between tabs
-     * @param {string} tabId - The tab ID to switch to
+     * Display search timeline
+     * @param {Array} searchTimeline - Search timeline data
      */
-    function switchTab(tabId) {
-        // Update active tab
-        const tabs = document.querySelectorAll('.tab-item');
-        tabs.forEach(tab => {
-            tab.classList.toggle('active', tab.dataset.tab === tabId);
-        });
-        
-        // Update visible content
-        const tabContents = document.querySelectorAll('.tab-content');
-        tabContents.forEach(content => {
-            content.style.display = content.dataset.tab === tabId ? 'block' : 'none';
-        });
+    function displaySearchTimeline(searchTimeline) {
+        const timelineContainer = document.getElementById('search-timeline');
+        if (!timelineContainer) return;
+
+        if (!searchTimeline || searchTimeline.length === 0) {
+            timelineContainer.innerHTML = '<p style="text-align: center; color: var(--text-secondary);">No search timeline data available</p>';
+            return;
+        }
+
+        const timelineHtml = searchTimeline.map(search => {
+            const timestamp = new Date(search.timestamp).toLocaleTimeString();
+            const statusClass = search.success_status === 'success' ? 'success' : 'error';
+            const statusText = search.success_status === 'success' ? 'Success' : 'Failed';
+
+            return `
+                <div class="timeline-item">
+                    <div class="timeline-time">${timestamp}</div>
+                    <div class="timeline-engine">${escapeHtml(search.engine)}</div>
+                    <div class="timeline-query">${escapeHtml(search.query)}</div>
+                    <div class="timeline-results">${formatNumber(search.results_count)} results</div>
+                    <div class="timeline-response">${(search.response_time_ms / 1000).toFixed(2)}s</div>
+                    <div class="timeline-status ${statusClass}">${statusText}</div>
+                </div>
+            `;
+        }).join('');
+
+        timelineContainer.innerHTML = timelineHtml;
     }
-    
+
+
     /**
-     * Toggle source highlight
-     * @param {HTMLElement} sourceEl - The source element to highlight
+     * Start polling for updates (only if research is not completed)
      */
-    function toggleSourceHighlight(sourceEl) {
-        // Remove highlight from all sources
-        const allSources = document.querySelectorAll('.source-item');
-        allSources.forEach(s => s.classList.remove('highlighted'));
-        
-        // Add highlight to clicked source
-        sourceEl.classList.add('highlighted');
-        
-        // Get source index
-        const sourceIndex = parseInt(sourceEl.dataset.id);
-        
-        // Highlight content with this source
-        highlightContentFromSource(sourceIndex);
+    function startPolling() {
+        // Only poll if research is not completed
+        if (!isCompleted) {
+            console.log('Starting polling for updates...');
+
+            // Poll every 10 seconds (much less frequent)
+            pollInterval = setInterval(async () => {
+                if (!isCompleted) {
+                    // Only reload basic research info, not token metrics
+                    await loadBasicResearchDetails();
+                } else {
+                    // Stop polling once completed
+                    if (pollInterval) {
+                        clearInterval(pollInterval);
+                        pollInterval = null;
+                        console.log('Research completed, stopped polling');
+                    }
+                }
+            }, 10000);
+        } else {
+            console.log('Research already completed, no polling needed');
+        }
     }
-    
+
     /**
-     * Highlight content sections from a specific source
-     * @param {number} sourceIndex - The source index to highlight
+     * Load token metrics for this research
      */
-    function highlightContentFromSource(sourceIndex) {
-        // Remove all existing highlights
-        const allCitations = document.querySelectorAll('.citation-highlight');
-        allCitations.forEach(el => {
-            el.classList.remove('citation-highlight');
-        });
-        
-        // Add highlight to citations from this source
-        const citations = document.querySelectorAll(`.citation[data-source="${sourceIndex}"]`);
-        citations.forEach(citation => {
-            citation.classList.add('citation-highlight');
-            
-            // Scroll to first citation
-            if (citations.length > 0) {
-                citations[0].scrollIntoView({ behavior: 'smooth', block: 'center' });
+    async function loadTokenMetrics() {
+        console.log('Loading token metrics...');
+
+        try {
+            const response = await fetch(`/metrics/api/metrics/research/${currentResearchId}/timeline`, {
+                headers: {
+                    'Accept': 'application/json'
+                }
+            });
+
+            if (!response.ok) {
+                console.log('No token metrics available for this research');
+                return;
             }
-        });
+
+            const data = await response.json();
+
+            if (data.status === 'success') {
+                console.log('Token metrics loaded:', data.metrics);
+                displayTokenMetrics(data.metrics);
+            }
+
+        } catch (error) {
+            console.error('Error loading token metrics:', error);
+        }
     }
-    
+
     /**
-     * Handle copying citation
-     * @param {number} sourceIndex - The source index to copy
+     * Display token metrics on the page
+     * @param {Object} metrics - Token metrics data
      */
-    async function handleCopyCitation(sourceIndex) {
-        if (!detailData || !detailData.sources || !detailData.sources[sourceIndex]) {
-            return;
+    function displayTokenMetrics(metrics) {
+        console.log('Displaying token metrics with data:', metrics);
+
+        // Show the metrics section
+        const metricsSection = document.getElementById('token-metrics-section');
+        if (metricsSection) {
+            metricsSection.style.display = 'block';
         }
-        
-        const source = detailData.sources[sourceIndex];
-        
-        // Format citation
-        let citation = '';
-        
-        if (source.author) {
-            citation += source.author;
-            if (source.date) {
-                citation += ` (${source.date})`;
+
+        // Update summary cards with safe data access
+        const summary = metrics.summary || {};
+        const totalTokensEl = document.getElementById('total-tokens');
+        const totalPromptTokensEl = document.getElementById('total-prompt-tokens');
+        const totalCompletionTokensEl = document.getElementById('total-completion-tokens');
+        const totalCallsEl = document.getElementById('total-calls');
+        const avgResponseTimeEl = document.getElementById('avg-response-time');
+        const successRateEl = document.getElementById('success-rate');
+
+        if (totalTokensEl) totalTokensEl.textContent = formatNumber(summary.total_tokens || 0);
+        if (totalPromptTokensEl) totalPromptTokensEl.textContent = formatNumber(summary.total_prompt_tokens || 0);
+        if (totalCompletionTokensEl) totalCompletionTokensEl.textContent = formatNumber(summary.total_completion_tokens || 0);
+        if (totalCallsEl) totalCallsEl.textContent = formatNumber(summary.total_calls || 0);
+        if (avgResponseTimeEl) avgResponseTimeEl.textContent = (summary.avg_response_time || 0) + 'ms';
+        if (successRateEl) successRateEl.textContent = (summary.success_rate || 0) + '%';
+
+        // Create timeline chart
+        createTimelineChart(metrics.timeline);
+
+        // Display phase breakdown
+        displayPhaseBreakdown(metrics.phase_stats);
+
+        // Display call stack traces
+        displayCallStackTraces(metrics.timeline);
+    }
+
+    /**
+     * Create timeline chart
+     * @param {Array} timeline - Timeline data
+     */
+    function createTimelineChart(timeline) {
+        const ctx = document.getElementById('timeline-chart');
+        if (!ctx) return;
+
+        // Handle empty timeline
+        if (!timeline || timeline.length === 0) {
+            console.log('No timeline data available');
+            ctx.getContext('2d').clearRect(0, 0, ctx.width, ctx.height);
+
+            // Show a message in the chart area
+            const chartContainer = ctx.parentElement;
+            if (chartContainer) {
+                chartContainer.innerHTML = '<div style="text-align: center; padding: 2rem; color: var(--text-secondary);">No token usage timeline available for this research.<br>This research may have used search-only mode or meta search engine.</div>';
             }
-            citation += '. ';
-        }
-        
-        if (source.title) {
-            citation += `"${source.title}". `;
+            return;
         }
-        
-        if (source.url) {
-            citation += `Retrieved from ${source.url}`;
+
+        // Prepare data
+        const labels = timeline.map(item => {
+            const date = new Date(item.timestamp);
+            return date.toLocaleTimeString('en-US', {
+                hour: '2-digit',
+                minute: '2-digit',
+                second: '2-digit'
+            });
+        });
+
+        const cumulativeTokens = timeline.map(item => item.cumulative_tokens);
+        const cumulativePromptTokens = timeline.map(item => item.cumulative_prompt_tokens);
+        const cumulativeCompletionTokens = timeline.map(item => item.cumulative_completion_tokens);
+        const promptTokens = timeline.map(item => item.prompt_tokens);
+        const completionTokens = timeline.map(item => item.completion_tokens);
+
+        // Debug logging
+        console.log('Timeline data sample:', timeline.slice(0, 2));
+        console.log('Prompt tokens sample:', promptTokens.slice(0, 5));
+        console.log('Completion tokens sample:', completionTokens.slice(0, 5));
+
+        // Destroy existing chart
+        if (timelineChart) {
+            timelineChart.destroy();
         }
-        
-        // Copy to clipboard
-        try {
-            await navigator.clipboard.writeText(citation);
-            
-            // Show tooltip
-            const btn = document.querySelector(`.source-item[data-id="${sourceIndex}"] .copy-citation-btn`);
-            if (btn) {
-                const originalHTML = btn.innerHTML;
-                btn.innerHTML = '<i class="fas fa-check"></i>';
-                
-                setTimeout(() => {
-                    btn.innerHTML = originalHTML;
-                }, 2000);
+
+        timelineChart = new Chart(ctx, {
+            type: 'line',
+            data: {
+                labels: labels,
+                datasets: [{
+                    label: 'Cumulative Total Tokens',
+                    data: cumulativeTokens,
+                    borderColor: 'rgba(75, 192, 192, 1)',
+                    backgroundColor: 'rgba(75, 192, 192, 0.1)',
+                    borderWidth: 3,
+                    fill: true,
+                    tension: 0.1,
+                    pointRadius: 4,
+                    pointHoverRadius: 6,
+                }, {
+                    label: 'Cumulative Input Tokens',
+                    data: cumulativePromptTokens,
+                    borderColor: 'rgba(54, 162, 235, 1)',
+                    backgroundColor: 'rgba(54, 162, 235, 0.05)',
+                    borderWidth: 2,
+                    fill: true,
+                    tension: 0.1,
+                    pointRadius: 2,
+                    pointHoverRadius: 4,
+                }, {
+                    label: 'Cumulative Output Tokens',
+                    data: cumulativeCompletionTokens,
+                    borderColor: 'rgba(255, 99, 132, 1)',
+                    backgroundColor: 'rgba(255, 99, 132, 0.05)',
+                    borderWidth: 2,
+                    fill: true,
+                    tension: 0.1,
+                    pointRadius: 2,
+                    pointHoverRadius: 4,
+                }, {
+                    label: 'Input Tokens per Call',
+                    data: promptTokens,
+                    borderColor: 'rgba(54, 162, 235, 0.8)',
+                    backgroundColor: 'rgba(54, 162, 235, 0.1)',
+                    borderWidth: 1,
+                    fill: false,
+                    tension: 0.1,
+                    pointRadius: 2,
+                    pointHoverRadius: 4,
+                    yAxisID: 'y1',
+                    borderDash: [5, 5],
+                }, {
+                    label: 'Output Tokens per Call',
+                    data: completionTokens,
+                    borderColor: 'rgba(255, 99, 132, 0.8)',
+                    backgroundColor: 'rgba(255, 99, 132, 0.1)',
+                    borderWidth: 1,
+                    fill: false,
+                    tension: 0.1,
+                    pointRadius: 2,
+                    pointHoverRadius: 4,
+                    yAxisID: 'y1',
+                    borderDash: [5, 5],
+                }]
+            },
+            options: {
+                responsive: true,
+                maintainAspectRatio: false,
+                interaction: {
+                    intersect: false,
+                    mode: 'index'
+                },
+                scales: {
+                    x: {
+                        display: true,
+                        title: {
+                            display: true,
+                            text: 'Time'
+                        }
+                    },
+                    y: {
+                        type: 'linear',
+                        display: true,
+                        position: 'left',
+                        title: {
+                            display: true,
+                            text: 'Cumulative Tokens'
+                        },
+                        beginAtZero: true
+                    },
+                    y1: {
+                        type: 'linear',
+                        display: true,
+                        position: 'right',
+                        title: {
+                            display: true,
+                            text: 'Tokens per Call'
+                        },
+                        beginAtZero: true,
+                        grid: {
+                            drawOnChartArea: false,
+                        },
+                    }
+                },
+                plugins: {
+                    legend: {
+                        position: 'top'
+                    },
+                    tooltip: {
+                        callbacks: {
+                            title: function(context) {
+                                const index = context[0].dataIndex;
+                                const item = timeline[index];
+                                return `${item.research_phase || 'Unknown Phase'} - ${item.model_name || 'Unknown Model'}`;
+                            },
+                            label: function(context) {
+                                let label = context.dataset.label || '';
+                                if (label) {
+                                    label += ': ';
+                                }
+                                label += formatNumber(context.parsed.y);
+                                return label;
+                            },
+                            afterBody: function(context) {
+                                const index = context[0].dataIndex;
+                                const item = timeline[index];
+                                return [
+                                    `Response Time: ${item.response_time_ms || 0}ms`,
+                                    `Status: ${item.success_status || 'unknown'}`,
+                                    `Engine: ${item.search_engine_selected || 'N/A'}`
+                                ];
+                            }
+                        }
+                    }
+                }
             }
-        } catch (error) {
-            console.error('Failed to copy citation:', error);
-        }
+        });
     }
-    
+
     /**
-     * Apply syntax highlighting to code blocks
+     * Display phase breakdown
+     * @param {Object} phaseStats - Phase statistics
      */
-    function highlightCodeBlocks() {
-        // Check if Prism is available
-        if (typeof Prism !== 'undefined') {
-            Prism.highlightAllUnder(detailContainer);
+    function displayPhaseBreakdown(phaseStats) {
+        const container = document.getElementById('phase-breakdown');
+        if (!container) return;
+
+        container.innerHTML = '';
+
+        Object.keys(phaseStats).forEach(phase => {
+            const stats = phaseStats[phase];
+            const item = document.createElement('div');
+            item.className = 'phase-stat-item';
+            item.innerHTML = `
+                <div class="phase-name">${phase}</div>
+                <div class="phase-tokens">${formatNumber(stats.tokens)} tokens</div>
+                <div class="phase-calls">${stats.count} calls | ${stats.avg_response_time}ms avg</div>
+            `;
+            container.appendChild(item);
+        });
+    }
+
+    /**
+     * Display call stack traces for LLM calls
+     * @param {Array} timeline - Timeline data with call stack info
+     */
+    function displayCallStackTraces(timeline) {
+        const container = document.getElementById('call-stack-traces');
+        if (!container) return;
+
+        container.innerHTML = '';
+
+        // Filter timeline items that have call stack information
+        const itemsWithCallStack = timeline.filter(item => item.call_stack || item.calling_function);
+
+        if (itemsWithCallStack.length === 0) {
+            container.innerHTML = '<p style="text-align: center; color: var(--text-secondary);">No call stack traces available for this research</p>';
+            return;
         }
+
+        itemsWithCallStack.forEach((item, index) => {
+            const trace = document.createElement('div');
+            trace.style.marginBottom = '1rem';
+            trace.style.padding = '0.75rem';
+            trace.style.backgroundColor = 'var(--card-bg)';
+            trace.style.borderRadius = '6px';
+            trace.style.border = '1px solid var(--border-color)';
+
+            // Format timestamp
+            const timestamp = item.timestamp ? new Date(item.timestamp).toLocaleTimeString() : 'Unknown time';
+
+            trace.innerHTML = `
+                <div style="display: flex; justify-content: space-between; align-items: center; margin-bottom: 0.5rem;">
+                    <div style="font-weight: 500; color: var(--text-primary);">
+                        ${escapeHtml(item.calling_function || 'Unknown Function')}
+                    </div>
+                    <div style="font-size: 0.875rem; color: var(--text-secondary);">
+                        ${timestamp} - ${item.prompt_tokens || 0} in + ${item.completion_tokens || 0} out = ${item.tokens || 0} tokens, ${item.response_time_ms || 0}ms
+                    </div>
+                </div>
+                <div style="font-family: 'Courier New', monospace; font-size: 0.75rem; background: var(--bg-color); padding: 0.5rem; border-radius: 4px; color: var(--text-secondary); overflow-x: auto; margin-bottom: 0.5rem;">
+                    ${escapeHtml(item.call_stack || 'No stack trace available')}
+                </div>
+                <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(120px, 1fr)); gap: 0.5rem; font-size: 0.875rem;">
+                    <div><strong>File:</strong> ${escapeHtml((item.calling_file || 'Unknown').split('/').pop())}</div>
+                    <div><strong>Phase:</strong> ${escapeHtml(item.research_phase || 'N/A')}</div>
+                    <div><strong>Model:</strong> ${escapeHtml(item.model_name || 'N/A')}</div>
+                    <div><strong>Status:</strong>
+                        <span style="color: ${item.success_status === 'success' ? 'green' : 'red'}">
+                            ${escapeHtml(item.success_status || 'Unknown')}
+                        </span>
+                    </div>
+                </div>
+            `;
+            container.appendChild(trace);
+        });
+    }
+
+    /**
+     * Format large numbers with commas
+     * @param {number} num - Number to format
+     * @returns {string} Formatted number
+     */
+    function formatNumber(num) {
+        return num.toString().replace(/\B(?=(\d{3})+(?!\d))/g, ",");
     }
-    
+
+    /**
+     * Escape HTML to prevent XSS
+     * @param {string} text - Text to escape
+     * @returns {string} Escaped text
+     */
+    function escapeHtml(text) {
+        const div = document.createElement('div');
+        div.textContent = text;
+        return div.innerHTML;
+    }
+
+    // Clean up on page unload
+    window.addEventListener('beforeunload', function() {
+        if (pollInterval) {
+            clearInterval(pollInterval);
+        }
+    });
+
     // Initialize on DOM content loaded
     if (document.readyState === 'loading') {
-        document.addEventListener('DOMContentLoaded', initializeDetail);
+        document.addEventListener('DOMContentLoaded', () => {
+            initializeResearchDetails().catch(console.error);
+        });
     } else {
-        initializeDetail();
+        initializeResearchDetails().catch(console.error);
     }
-})(); 
+})();