Update static/verify.js

static/verify.js

@@ -6,23 +6,26 @@ document.addEventListener('DOMContentLoaded', () => {
     const progressPercent = document.getElementById('progressPercent');
     const loaderText = document.getElementById('loaderText');
     const resultsWrapper = document.getElementById('resultsWrapper');
-    const resultsArea = document.getElementById('results'); // The JSON Report Textarea
-    const correctedSrtOutput = document.getElementById('correctedSrtOutput'); // The Corrections Textarea
+    const resultsArea = document.getElementById('results'); 
+    const correctedSrtOutput = document.getElementById('correctedSrtOutput');
+    const ocrErrorOutput = document.getElementById('ocrErrorOutput');
     const errorMessage = document.getElementById('errorMessage');
 
-    // Stats Elements
+    // Stats
     const statChunks = document.getElementById('statChunks');
     const statCorrections = document.getElementById('statCorrections');
     const statErrors = document.getElementById('statErrors');
-    const statOcrErrors = document.getElementById('statOcrErrors'); // NEW
-
-    // Output Areas
-    const ocrErrorOutput = document.getElementById('ocrErrorOutput');
+    const statOcrErrors = document.getElementById('statOcrErrors');
 
-    // --- File Input Listeners (For UI visual feedback) ---
+    // File Inputs
     const pdfUpload = document.getElementById('pdfUpload');
     const srtUpload = document.getElementById('srtUpload');
 
+    // Global Store for Original SRT Content
+    let originalSrtContent = "";
+
+    // --- Event Listeners ---
+
     if (pdfUpload) {
         pdfUpload.addEventListener('change', function() {
             if(this.files[0]) document.getElementById('pdfFileName').textContent = this.files[0].name;
@@ -31,28 +34,31 @@ document.addEventListener('DOMContentLoaded', () => {
 
     if (srtUpload) {
         srtUpload.addEventListener('change', function() {
-            if(this.files[0]) document.getElementById('srtFileName').textContent = this.files[0].name;
+            if(this.files[0]) {
+                document.getElementById('srtFileName').textContent = this.files[0].name;
+                
+                // Read content immediately for later patching
+                const reader = new FileReader();
+                reader.onload = (e) => { originalSrtContent = e.target.result; };
+                reader.readAsText(this.files[0]);
+            }
         });
     }
 
-    // --- Main Verify Button Click ---
     if (verifyBtn) {
         verifyBtn.addEventListener('click', async () => {
-            // 1. Gather Data
             const apiKeys = document.getElementById('apiKeyInput').value.trim();
             const pdfFile = pdfUpload ? pdfUpload.files[0] : null;
             const srtFile = srtUpload ? srtUpload.files[0] : null;
             const pagesPerRequest = document.getElementById('pagesPerRequestInput').value;
             const modelName = document.getElementById('modelSelect').value;
 
-            // 2. Validation
             if (errorMessage) errorMessage.classList.add('hidden');
-            
-            if (!apiKeys) return showError("Please enter at least one Gemini API Key.");
-            if (!pdfFile) return showError("Please upload a PDF file.");
-            if (!srtFile) return showError("Please upload an SRT file.");
+            if (!apiKeys) return showError("Please enter API Keys.");
+            if (!pdfFile) return showError("Please upload a PDF.");
+            if (!srtFile) return showError("Please upload an SRT.");
 
-            // 3. UI State: Loading
+            // Reset UI
             verifyBtn.disabled = true;
             verifyBtn.classList.add('opacity-50', 'cursor-not-allowed');
             if (loader) loader.classList.remove('hidden');
@@ -65,7 +71,7 @@ document.addEventListener('DOMContentLoaded', () => {
             formData.append('pages_per_request', pagesPerRequest);
             formData.append('model_name', modelName);
 
-            // 4. Fake Progress Animation
+            // Fake Progress
             let progress = 0;
             const interval = setInterval(() => {
                 if (progress < 90) {
@@ -73,34 +79,22 @@ document.addEventListener('DOMContentLoaded', () => {
                     if (progress > 90) progress = 90;
                     if (progressBar) progressBar.style.width = `${progress}%`;
                     if (progressPercent) progressPercent.textContent = `${Math.round(progress)}%`;
-                    
-                    if (loaderText) {
-                        if (progress > 20 && progress < 50) loaderText.textContent = "Splitting PDF into batches...";
-                        if (progress > 50 && progress < 80) loaderText.textContent = "Processing with Gemini (Parallel)...";
-                        if (progress > 80) loaderText.textContent = "Aggregating results...";
-                    }
                 }
             }, 800);
 
             try {
-                // 5. Send Request
                 const response = await fetch('/verify_batch', {
                     method: 'POST',
                     body: formData
                 });
 
                 clearInterval(interval);
-                
                 const data = await response.json();
 
-                if (!response.ok) {
-                    throw new Error(data.detail || data.error || "Server Error");
-                }
+                if (!response.ok) throw new Error(data.detail || data.error || "Server Error");
 
-                // 6. Success State
                 if (progressBar) progressBar.style.width = '100%';
                 if (progressPercent) progressPercent.textContent = '100%';
-                if (loaderText) loaderText.textContent = "Complete!";
                 
                 setTimeout(() => {
                     if (loader) loader.classList.add('hidden');
@@ -126,72 +120,128 @@ document.addEventListener('DOMContentLoaded', () => {
             errorMessage.textContent = `Error: ${msg}`;
             errorMessage.classList.remove('hidden');
         } else {
-            alert(`Error: ${msg}`);
+            alert(msg);
         }
     }
 
-function displayResults(data) {
+    function displayResults(data) {
         if (resultsWrapper) resultsWrapper.classList.remove('hidden');
         
-        // 1. Basic Stats
+        // 1. Stats
         if (statChunks) statChunks.textContent = data.total_chunks || 0;
         if (statErrors) statErrors.textContent = data.system_errors ? data.system_errors.length : 0;
 
         const corrections = data.corrections || [];
-        
-        // 2. Separate Corrections from Error Reports
-        let validCorrectionsCount = 0;
-        let ocrAlertsCount = 0;
-        let ocrReportText = "";
-        
-        // Loop to process the prompt output
+        let validCorrections = [];
+        let ocrReports = "";
+
+        // 2. Filter Data
         corrections.forEach((item, index) => {
-            // Count actual corrections (where correctedSrt exists)
+            // Only count if it has a correctedSrt field
             if (item.correctedSrt) {
-                validCorrectionsCount++;
+                validCorrections.push(item);
             }
-
-            // Check for significant error reports
-            // We filter out "No significant errors found." or empty strings
-            if (item.errorReport && 
-                !item.errorReport.includes("No significant errors") && 
-                !item.errorReport.includes("No errors")) {
-                
-                ocrAlertsCount++;
-                ocrReportText += `[Batch ${index + 1}]\n${item.errorReport}\n\n----------------\n\n`;
+            // Collect significant error reports
+            if (item.errorReport && !item.errorReport.includes("No significant errors")) {
+                ocrReports += `[Report ${index+1}]\n${item.errorReport}\n\n`;
             }
         });
 
-        // 3. Update UI
-        if (statCorrections) statCorrections.textContent = validCorrectionsCount;
-        if (statOcrErrors) statOcrErrors.textContent = ocrAlertsCount;
+        if (statCorrections) statCorrections.textContent = validCorrections.length;
+        if (statOcrErrors) statOcrErrors.textContent = ocrReports ? "!" : "0";
 
-        if (ocrErrorOutput) {
-            ocrErrorOutput.value = ocrReportText || "No significant OCR errors reported.";
-        }
+        // 3. Fill Textareas
+        if (resultsArea) resultsArea.value = JSON.stringify(data, null, 2);
+        if (ocrErrorOutput) ocrErrorOutput.value = ocrReports || "No significant OCR errors reported.";
+        if (correctedSrtOutput) correctedSrtOutput.value = JSON.stringify(validCorrections, null, 2);
+    }
 
-        if (correctedSrtOutput) {
-            correctedSrtOutput.value = JSON.stringify(corrections, null, 2);
-        }
+    // --- ROBUST SRT DOWNLOAD LOGIC ---
+    
+    const downloadSrtBtn = document.getElementById('downloadSrtBtn');
+    if (downloadSrtBtn) {
+        downloadSrtBtn.addEventListener('click', () => {
+            if (!originalSrtContent) {
+                alert("Original SRT content is missing. Please re-upload the SRT file to enable patching.");
+                return;
+            }
 
-        if (resultsArea) {
-            resultsArea.value = JSON.stringify(data, null, 2);
-        }
+            // Parse results from the text area
+            let corrections = [];
+            try {
+                // We use the filtered list we displayed in correctedSrtOutput
+                corrections = JSON.parse(correctedSrtOutput.value);
+            } catch (e) {
+                alert("No valid corrections data found to apply.");
+                return;
+            }
+
+            if (!corrections || corrections.length === 0) {
+                alert("There are no corrections to apply.");
+                return;
+            }
+
+            try {
+                const patchedSrt = patchSrtFile(originalSrtContent, corrections);
+                downloadFile(patchedSrt, 'corrected_subtitles.srt');
+            } catch (e) {
+                alert("Error applying patches: " + e.message);
+            }
+        });
     }
-    
-    // Download Button Handler
-    const downloadBtn = document.getElementById('downloadJsonBtn');
-    if (downloadBtn) {
-        downloadBtn.addEventListener('click', () => {
-            const text = resultsArea ? resultsArea.value : "{}";
-            const blob = new Blob([text], { type: 'application/json' });
-            const url = URL.createObjectURL(blob);
-            const a = document.createElement('a');
-            a.href = url;
-            a.download = 'verification_report.json';
-            document.body.appendChild(a);
-            a.click();
-            document.body.removeChild(a);
+
+    /**
+     * Replaces blocks in the original SRT with the corrected blocks.
+     */
+    function patchSrtFile(originalText, corrections) {
+        // 1. Split original SRT into blocks (Double newline separator)
+        // Normalize line endings to \n first
+        const normalizedText = originalText.replace(/\r\n/g, '\n').replace(/\r/g, '\n');
+        let blocks = normalizedText.split(/\n\n+/);
+        
+        // 2. Create a Map for faster updates
+        // We map "ID" -> Index in the blocks array
+        let blockMap = new Map();
+        
+        blocks.forEach((block, index) => {
+            const lines = block.trim().split('\n');
+            if (lines.length > 0) {
+                const id = lines[0].trim();
+                // Store the index so we can update the array directly
+                blockMap.set(id, index); 
+            }
+        });
+
+        // 3. Apply Corrections
+        let appliedCount = 0;
+        corrections.forEach(fix => {
+            // Extract ID from the CORRECTION block (it's usually the first line)
+            // Example fix.correctedSrt: "10\n00:00:10 --> ... \nHello"
+            if (!fix.correctedSrt) return;
+
+            const fixLines = fix.correctedSrt.trim().split('\n');
+            const fixId = fixLines[0].trim();
+
+            if (blockMap.has(fixId)) {
+                const originalIndex = blockMap.get(fixId);
+                // REPLACE the entire original block with the new corrected block
+                blocks[originalIndex] = fix.correctedSrt.trim();
+                appliedCount++;
+            }
         });
+
+        console.log(`Applied ${appliedCount} corrections.`);
+        return blocks.join('\n\n');
+    }
+
+    function downloadFile(content, filename) {
+        const blob = new Blob([content], { type: 'text/plain' });
+        const url = URL.createObjectURL(blob);
+        const a = document.createElement('a');
+        a.href = url;
+        a.download = filename;
+        document.body.appendChild(a);
+        a.click();
+        document.body.removeChild(a);
     }
 });