Update main.py

main.py

@@ -404,8 +404,14 @@ async def unified_scan_endpoint(payload: UnifiedRequest):
     except Exception as e:
         v2_msg = f"V2 Vision Unavailable"
 
-    # 4. Calculate the Master Risk Score (Weighted Average)
-    master_score = (v1_score * 0.4) + (v4_score * 0.4) + (v2_score * 0.2)
+    # 4. Calculate the Master Risk Score (Dynamic Weights)
+    if "Skipped" in v2_msg or "Unavailable" in v2_msg:
+        # If CNN times out, re-balance weights so a dummy 0.0 doesn't dilute the risk
+        master_score = (v1_score * 0.5) + (v4_score * 0.5)
+    else:
+        # Standard 40/40/20 split
+        master_score = (v1_score * 0.4) + (v4_score * 0.4) + (v2_score * 0.2)
+        
     latency = (time.perf_counter() - start_time) * 1000
 
     # 5. Return safely serialized JSON to the frontend