Update hf_demo.py

hf_demo.py

@@ -17,6 +17,7 @@ from enum import Enum
 from typing import Dict, List, Optional, Any, Tuple
 
 import yaml
+import numpy as np
 from fastapi import FastAPI, HTTPException, Depends, status
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
@@ -260,7 +261,7 @@ class BayesianRiskEngine:
         except sqlite3.Error as e:
             logger.error(f"Failed to record outcome: {e}")
 
-    # ---------- NEW: Enhanced risk using HMC coefficients ----------
+    # ---------- ENHANCED RISK USING HMC COEFFICIENTS ----------
     def enhanced_risk(self, action_text: str, context: Dict, hmc_coeffs: Optional[Dict] = None) -> float:
         """
         Compute a risk score using HMC coefficients if available.
@@ -280,7 +281,10 @@ class BayesianRiskEngine:
         env_prod = 1 if context.get('environment') == 'production' else 0
         role_junior = 1 if context.get('user_role') == 'junior' else 0
         hour = datetime.now().hour
-        # Use the simple posterior risk as a feature (normalized)
+        hour_sin = np.sin(2 * np.pi * hour / 24)
+        hour_cos = np.cos(2 * np.pi * hour / 24)
+
+        # Use the simple posterior risk as a feature (centered)
         simple_risk = self.calculate_posterior(action_text, context)["score"]
         confidence = context.get('confidence', 0.85)
 
@@ -290,7 +294,8 @@ class BayesianRiskEngine:
             hmc_coeffs.get('β_env', {}).get('mean', 0) * env_prod +
             hmc_coeffs.get('β_role', {}).get('mean', 0) * role_junior +
             hmc_coeffs.get('β_risk', {}).get('mean', 0) * (simple_risk - 0.5) +
-            hmc_coeffs.get('β_hour', {}).get('mean', 0) * ((hour - 12) / 12) +
+            hmc_coeffs.get('β_hour_sin', {}).get('mean', 0) * hour_sin +
+            hmc_coeffs.get('β_hour_cos', {}).get('mean', 0) * hour_cos +
             hmc_coeffs.get('β_conf', {}).get('mean', 0) * (confidence - 0.5)
         )
         # Convert to probability