Update app.py

app.py

@@ -18,7 +18,10 @@ import numpy as np
 from flask import Flask, Response, jsonify, render_template_string, request
 from PIL import Image
 
-from wall_pipeline import WallPipeline, _GPU
+from wall_pipeline import WallPipeline, _CUPY, _TORCH_CUDA, _CV_CUDA
+
+# Unified "any GPU active" flag for the UI badge
+_GPU = _CUPY or _TORCH_CUDA or _CV_CUDA
 
 app = Flask(__name__)
 
@@ -256,6 +259,11 @@ def result():
         "rooms"      : rooms,
         "calibration": sess.get("calibration", {}),
         "gpu"        : _GPU,
+        "gpu_detail" : {
+            "cupy"        : _CUPY,
+            "torch_cuda"  : _TORCH_CUDA,
+            "opencv_cuda" : _CV_CUDA,
+        },
     })
 
 
@@ -804,8 +812,18 @@ async function loadResult(){
 
   // GPU badge
   const badge = document.getElementById('gpu-badge');
-  badge.textContent = d.gpu ? 'GPU ⚡' : 'CPU';
-  badge.className   = 'gpu-badge'+(d.gpu?' on':'');
+  const gd = d.gpu_detail || {};
+  const layers = [];
+  if (gd.cupy)         layers.push('CuPy');
+  if (gd.torch_cuda)   layers.push('Torch');
+  if (gd.opencv_cuda)  layers.push('cv2');
+  if (layers.length) {
+    badge.textContent = '⚡ GPU: ' + layers.join('+');
+    badge.className   = 'gpu-badge on';
+  } else {
+    badge.textContent = 'CPU only';
+    badge.className   = 'gpu-badge';
+  }
 
   showMainImg('data:image/jpeg;base64,'+d.composite);
   if(d.wall_mask) document.getElementById('walls-img').src='data:image/jpeg;base64,'+d.wall_mask;
@@ -1089,7 +1107,7 @@ async function exportJSON(){
 if __name__ == "__main__":
     print("=" * 60)
     print(" Blueprint Room Extractor")
-    print(f" GPU acceleration: {'ON (CuPy)' if _GPU else 'OFF (CPU fallback)'}")
+    print(f" GPU: CuPy={_CUPY}  PyTorch-CUDA={_TORCH_CUDA}  OpenCV-CUDA={_CV_CUDA}")
     print(" Open: http://localhost:7860")
     print("=" * 60)
     app.run(host="0.0.0.0", port=7860, debug=False, threaded=True)