Sync from GitHub: 1cb3988f8558cd7c7ec8e62c67906f3ffc8cd8e1

frontend/src/components/ResultCard.jsx

@@ -1,5 +1,5 @@
 import React, { useRef, useEffect, useState } from 'react';
-import { SlidersHorizontal } from 'lucide-react';
+import { SlidersHorizontal, Download, Eye } from 'lucide-react';
 
 const ResultCard = ({ result, imageData, processedImageData, onReprocess, isProcessing }) => {
   const canvasRef = useRef(null);
@@ -11,6 +11,7 @@ const ResultCard = ({ result, imageData, processedImageData, onReprocess, isProc
   const [adjustedDataUrl, setAdjustedDataUrl] = useState(null);
   const [previewDimensions, setPreviewDimensions] = useState({ width: 0, height: 0 });
   const [currentImageData, setCurrentImageData] = useState(processedImageData || imageData);
+  const [showEnhanced, setShowEnhanced] = useState(false);
 
   // Function to crop image regions
   const cropRegion = (img, coords, scaleX, scaleY) => {
@@ -253,6 +254,43 @@ const ResultCard = ({ result, imageData, processedImageData, onReprocess, isProc
               <span>100% (Best)</span>
             </div>
           </div>
+
+          {/* Enhanced Image Toggle & Download */}
+          {result.image_enhanced && result.enhanced_image && (
+            <div className="bg-green-50 rounded-lg p-4 shadow-sm border border-green-200">
+              <div className="flex items-center justify-between mb-2">
+                <div className="flex items-center gap-2">
+                  <svg className="w-5 h-5 text-green-600" fill="none" viewBox="0 0 24 24" stroke="currentColor">
+                    <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M9 12l2 2 4-4m6 2a9 9 0 11-18 0 9 9 0 0118 0z" />
+                  </svg>
+                  <span className="text-sm font-medium text-green-800">Image Enhanced (2x)</span>
+                </div>
+                <div className="flex gap-2">
+                  <button
+                    onClick={() => {
+                      const enhancedData = `data:image/png;base64,${result.enhanced_image}`;
+                      setCurrentImageData(showEnhanced ? imageData : enhancedData);
+                      setShowEnhanced(!showEnhanced);
+                    }}
+                    className="flex items-center gap-1 px-3 py-1 bg-green-600 hover:bg-green-700 text-white rounded text-xs font-medium transition-colors"
+                  >
+                    <Eye className="w-3 h-3" />
+                    {showEnhanced ? 'Show Original' : 'Show Enhanced'}
+                  </button>
+                  <a
+                    href={`data:image/png;base64,${result.enhanced_image}`}
+                    download={`enhanced_${result.filename || 'invoice'}.png`}
+                    className="flex items-center gap-1 px-3 py-1 bg-blue-600 hover:bg-blue-700 text-white rounded text-xs font-medium transition-colors"
+                  >
+                    <Download className="w-3 h-3" />
+                    Download
+                  </a>
+                </div>
+              </div>
+              <p className="text-xs text-green-700">Real-ESRGAN enhancement applied before processing</p>
+            </div>
+          )}
+          
           <div className="relative bg-gray-50 rounded-lg p-4 flex justify-center items-center">
             <canvas ref={canvasRef} className="max-w-full h-auto rounded shadow-md" />
             {isProcessing && (

inference.py

@@ -343,6 +343,19 @@ class InferenceProcessor:
         image, enhanced_image_path = InferenceProcessor.preprocess_image(image_path, enhance=enhance)
         timing_breakdown['image_preprocessing'] = round(time.time() - t1, 3)
         
+        # Save enhanced image info for response
+        image_was_enhanced = (enhanced_image_path != image_path)
+        enhanced_image_base64 = None
+        
+        if image_was_enhanced:
+            # Convert enhanced image to base64 for response
+            import base64
+            try:
+                with open(enhanced_image_path, 'rb') as f:
+                    enhanced_image_base64 = base64.b64encode(f.read()).decode('utf-8')
+            except:
+                pass
+        
         # Step 2: YOLO Detection (use enhanced image path for consistency)
         t2 = time.time()
         signature_info, stamp_info, signature_conf, stamp_conf = model_manager.detect_sign_stamp(enhanced_image_path)
@@ -399,7 +412,9 @@ class InferenceProcessor:
             "processing_time_sec": round(total_time, 2),
             "timing_breakdown": timing_breakdown,
             "cost_estimate_usd": round(cost_estimate, 6),
-            "warnings": warnings if warnings else None
+            "warnings": warnings if warnings else None,
+            "image_enhanced": image_was_enhanced,
+            "enhanced_image": enhanced_image_base64 if image_was_enhanced else None
         }
         
         return result