Update hf_demo.py

hf_demo.py

@@ -18,6 +18,8 @@ import logging
 import asyncio
 import sqlite3
 import requests
+import fcntl  # <-- NEW: for file locking
+import sys
 from datetime import datetime, timedelta
 from typing import Dict, List, Optional, Any, Tuple
 from contextlib import contextmanager
@@ -31,44 +33,64 @@ gr.close_all()
 from fastapi import FastAPI, HTTPException, Depends, status
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
-from pydantic import BaseModel, Field, field_validator
-from pydantic_settings import BaseSettings  # <-- NEW: Pydantic settings
+from pydantic import BaseModel, Field, field_validator, ConfigDict  # <-- NEW: ConfigDict
+from pydantic_settings import BaseSettings, SettingsConfigDict  # <-- NEW: SettingsConfigDict
 from gradio import mount_gradio_app
 
-# ============== CONFIGURATION (Pydantic) ==============
+# ============== SINGLE INSTANCE LOCK ==============
+LOCK_FILE = '/tmp/arf_app.lock'
+try:
+    lock_fd = open(LOCK_FILE, 'w')
+    fcntl.flock(lock_fd, fcntl.LOCK_EX | fcntl.LOCK_NB)
+except (IOError, OSError):
+    print("Another instance is already running. Exiting.")
+    sys.exit(1)
+# ==================================================
+
+# ============== CONFIGURATION (Pydantic V2) ==============
 class Settings(BaseSettings):
-    """Centralized configuration using Pydantic Settings"""
+    """Centralized configuration using Pydantic Settings V2"""
     
-    # Hugging Face settings
-    HF_SPACE_ID: str = Field(default='local', env='SPACE_ID')
-    HF_TOKEN: str = Field(default='', env='HF_TOKEN')
+    # Hugging Face settings (aliased to match expected env vars)
+    hf_space_id: str = Field(default='local', alias='SPACE_ID')
+    hf_token: str = Field(default='', alias='HF_TOKEN')
     
     # Persistence - HF persistent storage
-    DATA_DIR: str = Field(default='/data' if os.path.exists('/data') else './data')
+    data_dir: str = Field(
+        default='/data' if os.path.exists('/data') else './data',
+        alias='DATA_DIR'
+    )
     
     # Lead generation
-    LEAD_EMAIL: str = "petter2025us@outlook.com"
-    CALENDLY_URL: str = "https://calendly.com/petter2025us/arf-demo"
+    lead_email: str = "petter2025us@outlook.com"
+    calendly_url: str = "https://calendly.com/petter2025us/arf-demo"
     
     # Webhook for lead alerts (set in HF secrets)
-    SLACK_WEBHOOK: str = Field(default='', env='SLACK_WEBHOOK')
-    SENDGRID_API_KEY: str = Field(default='', env='SENDGRID_API_KEY')
+    slack_webhook: str = Field(default='', alias='SLACK_WEBHOOK')
+    sendgrid_api_key: str = Field(default='', alias='SENDGRID_API_KEY')
     
     # Security
-    API_KEY: str = Field(default_factory=lambda: str(uuid.uuid4()), env='ARF_API_KEY')
+    api_key: str = Field(
+        default_factory=lambda: str(uuid.uuid4()),
+        alias='ARF_API_KEY'
+    )
     
     # ARF defaults
-    DEFAULT_CONFIDENCE_THRESHOLD: float = 0.9
-    DEFAULT_MAX_RISK: str = "MEDIUM"
+    default_confidence_threshold: float = 0.9
+    default_max_risk: str = "MEDIUM"
     
-    class Config:
-        env_file = '.env'  # optionally load from .env file
-        extra = 'ignore'   # ignore extra env vars
+    # Pydantic V2 configuration
+    model_config = SettingsConfigDict(
+        populate_by_name=True,   # allows use of field names or aliases
+        extra='ignore',           # ignore extra env vars
+        env_prefix='',            # no prefix
+        case_sensitive=False      # case-insensitive matching
+    )
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
         # Ensure data directory exists
-        os.makedirs(self.DATA_DIR, exist_ok=True)
+        os.makedirs(self.data_dir, exist_ok=True)
 
 settings = Settings()
 
@@ -77,7 +99,7 @@ logging.basicConfig(
     level=logging.INFO,
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
     handlers=[
-        logging.FileHandler(f'{settings.DATA_DIR}/arf.log'),
+        logging.FileHandler(f'{settings.data_dir}/arf.log'),
         logging.StreamHandler()
     ]
 )
@@ -123,7 +145,7 @@ class BayesianRiskEngine:
             'default': {'alpha': 2.0, 'beta': 5.0}
         }
         
-        self.evidence_db = f"{settings.DATA_DIR}/evidence.db"
+        self.evidence_db = f"{settings.data_dir}/evidence.db"
         self._init_db()
     
     def _init_db(self):
@@ -190,41 +212,31 @@ class BayesianRiskEngine:
                 return (row[0] or 0, row[1] or 0) if row else (0, 0)
         except sqlite3.Error as e:
             logger.error(f"Failed to retrieve evidence: {e}")
-            return (0, 0)  # fallback to no evidence
+            return (0, 0)
     
     def calculate_posterior(self, 
                            action_text: str,
                            context: Dict[str, Any]) -> Dict[str, Any]:
-        # ... (same as before, no changes needed) ...
-        # 1. Classify action for appropriate prior
         action_type = self.classify_action(action_text)
         alpha0, beta0 = self.get_prior(action_type)
         
-        # 2. Get historical evidence
         action_hash = hashlib.sha256(action_text.encode()).hexdigest()
         successes, trials = self.get_evidence(action_hash)
         
-        # 3. Update prior with evidence → posterior
         alpha_n = alpha0 + successes
         beta_n = beta0 + (trials - successes)
         
-        # 4. Posterior mean (expected risk)
         posterior_mean = alpha_n / (alpha_n + beta_n)
-        
-        # 5. Incorporate context as likelihood adjustment
         context_multiplier = self._context_likelihood(context)
         
-        # 6. Final risk score (posterior predictive)
         risk_score = posterior_mean * context_multiplier
         risk_score = min(0.99, max(0.01, risk_score))
         
-        # 7. 95% credible interval (approximation)
         variance = (alpha_n * beta_n) / ((alpha_n + beta_n)**2 * (alpha_n + beta_n + 1))
         std_dev = variance ** 0.5
         ci_lower = max(0.01, posterior_mean - 1.96 * std_dev)
         ci_upper = min(0.99, posterior_mean + 1.96 * std_dev)
         
-        # 8. Risk level
         if risk_score > 0.8:
             risk_level = RiskLevel.CRITICAL
         elif risk_score > 0.6:
@@ -289,11 +301,10 @@ class BayesianRiskEngine:
 
 # ============== POLICY ENGINE ==============
 class PolicyEngine:
-    # ... (unchanged) ...
     def __init__(self):
         self.config = {
-            "confidence_threshold": settings.DEFAULT_CONFIDENCE_THRESHOLD,
-            "max_autonomous_risk": settings.DEFAULT_MAX_RISK,
+            "confidence_threshold": settings.default_confidence_threshold,
+            "max_autonomous_risk": settings.default_max_risk,
             "risk_thresholds": {
                 RiskLevel.LOW: 0.7,
                 RiskLevel.MEDIUM: 0.5,
@@ -318,7 +329,6 @@ class PolicyEngine:
                 action: str,
                 risk: Dict[str, Any],
                 confidence: float) -> Dict[str, Any]:
-        # ... unchanged ...
         gates = []
         
         # Gate 1: Confidence threshold
@@ -387,7 +397,6 @@ class PolicyEngine:
             "type": "license"
         })
         
-        # Overall decision
         all_passed = all(g["passed"] for g in gates)
         
         if not all_passed:
@@ -416,9 +425,8 @@ class PolicyEngine:
 
 # ============== RAG MEMORY WITH PERSISTENCE ==============
 class RAGMemory:
-    # ... (unchanged except error handling) ...
     def __init__(self):
-        self.db_path = f"{settings.DATA_DIR}/memory.db"
+        self.db_path = f"{settings.data_dir}/memory.db"
         self._init_db()
         self.embedding_cache = {}
     
@@ -472,7 +480,6 @@ class RAGMemory:
                 conn.close()
     
     def _simple_embedding(self, text: str) -> List[float]:
-        # ... unchanged ...
         if text in self.embedding_cache:
             return self.embedding_cache[text]
         
@@ -592,19 +599,18 @@ class RAGMemory:
         return signal
     
     def _notify_sales_team(self, signal: Dict):
-        if settings.SLACK_WEBHOOK:
+        if settings.slack_webhook:
             try:
-                requests.post(settings.SLACK_WEBHOOK, json={
+                requests.post(settings.slack_webhook, json={
                     "text": f"🚨 *Enterprise Lead Signal*\n"
                            f"Type: {signal['signal_type']}\n"
                            f"Action: {signal['action']}\n"
                            f"Risk Score: {signal['risk_score']:.2f}\n"
                            f"Time: {signal['timestamp']}\n"
-                           f"Contact: {settings.LEAD_EMAIL}"
+                           f"Contact: {settings.lead_email}"
                 }, timeout=5)
             except requests.RequestException as e:
                 logger.error(f"Slack notification failed: {e}")
-        # Email via SendGrid (if configured) could be added similarly
     
     def get_uncontacted_signals(self) -> List[Dict]:
         try:
@@ -641,8 +647,7 @@ class RAGMemory:
 security = HTTPBearer()
 
 async def verify_api_key(credentials: HTTPAuthorizationCredentials = Depends(security)):
-    """Verify API key for protected endpoints"""
-    if credentials.credentials != settings.API_KEY:
+    if credentials.credentials != settings.api_key:
         raise HTTPException(
             status_code=status.HTTP_403_FORBIDDEN,
             detail="Invalid API key"
@@ -704,7 +709,7 @@ app = FastAPI(
     description="Real ARF OSS components for enterprise lead generation",
     contact={
         "name": "ARF Sales",
-        "email": settings.LEAD_EMAIL,
+        "email": settings.lead_email,
     }
 )
 
@@ -760,7 +765,6 @@ async def evaluate_action(request: ActionRequest):
     Real ARF OSS evaluation pipeline (protected)
     """
     try:
-        # Build context
         context = {
             "environment": "production",
             "user_role": request.user_role,
@@ -768,23 +772,19 @@ async def evaluate_action(request: ActionRequest):
             "requires_human": request.requiresHuman
         }
         
-        # 1. Bayesian risk assessment
         risk = risk_engine.calculate_posterior(
             action_text=request.proposedAction,
             context=context
         )
         
-        # 2. Policy evaluation
         policy = policy_engine.evaluate(
             action=request.proposedAction,
             risk=risk,
             confidence=request.confidenceScore
         )
         
-        # 3. RAG memory recall
         similar = memory.find_similar(request.proposedAction, limit=3)
         
-        # 4. Track enterprise signals
         if not policy["allowed"] and risk["score"] > 0.7:
             memory.track_enterprise_signal(
                 signal_type=LeadSignal.HIGH_RISK_BLOCKED,
@@ -805,7 +805,6 @@ async def evaluate_action(request: ActionRequest):
                 metadata={"similar_count": len(similar)}
             )
         
-        # 5. Store in memory
         memory.store_incident(
             action=request.proposedAction,
             risk_score=risk["score"],
@@ -815,7 +814,6 @@ async def evaluate_action(request: ActionRequest):
             gates=policy["gates"]
         )
         
-        # 6. Format gates for response
         gates = []
         for g in policy["gates"]:
             gates.append(GateResult(
@@ -828,7 +826,6 @@ async def evaluate_action(request: ActionRequest):
                 metadata=g.get("metadata")
             ))
         
-        # 7. Build execution ladder
         execution_ladder = {
             "levels": [
                 {"name": "AUTONOMOUS_LOW", "required": gates[0].passed and gates[1].passed},
@@ -888,31 +885,24 @@ async def get_enterprise_signals(contacted: bool = False):
 
 @app.post("/api/v1/enterprise/signals/{signal_id}/contact", dependencies=[Depends(verify_api_key)])
 async def mark_signal_contacted(signal_id: str):
-    """Mark a lead signal as contacted (protected)"""
     memory.mark_contacted(signal_id)
     return {"status": "success", "message": "Signal marked as contacted"}
 
 @app.get("/api/v1/memory/similar", dependencies=[Depends(verify_api_key)])
 async def get_similar_actions(action: str, limit: int = 5):
-    """Find similar historical actions (protected)"""
     similar = memory.find_similar(action, limit=limit)
     return {"similar": similar, "count": len(similar)}
 
 @app.post("/api/v1/feedback", dependencies=[Depends(verify_api_key)])
 async def record_outcome(action: str, success: bool):
-    """
-    Record actual outcome for Bayesian updating (protected)
-    """
     risk_engine.record_outcome(action, success)
     return {"status": "success", "message": "Outcome recorded"}
 
 # ============== GRADIO LEAD GENERATION UI ==============
 def create_lead_gen_ui():
     """Professional lead generation interface (no auth needed for UI)"""
-    # ... (unchanged) ...
     with gr.Blocks(title="ARF OSS - Enterprise Reliability Intelligence") as ui:
         
-        # Header
         gr.HTML(f"""
         <div style="padding: 2rem; border-radius: 1rem; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); color: white; text-align: center;">
             <h1 style="font-size: 3em; margin-bottom: 0.5rem;">🤖 ARF OSS v3.3.9</h1>
@@ -926,7 +916,6 @@ def create_lead_gen_ui():
         </div>
         """)
         
-        # Value Proposition
         with gr.Row():
             with gr.Column():
                 gr.HTML("""
@@ -939,7 +928,6 @@ def create_lead_gen_ui():
                 </div>
                 """)
         
-        # Features Grid
         with gr.Row():
             with gr.Column():
                 gr.HTML("""
@@ -972,7 +960,6 @@ def create_lead_gen_ui():
                 </div>
                 """)
         
-        # Live Demo Stats
         demo_stats = gr.JSON(
             label="📊 Live Demo Statistics",
             value={
@@ -983,7 +970,6 @@ def create_lead_gen_ui():
             }
         )
         
-        # CTA Section
         gr.HTML(f"""
         <div style="margin: 3rem 0; padding: 3rem; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
                     border-radius: 1rem; text-align: center; color: white;">
@@ -993,11 +979,11 @@ def create_lead_gen_ui():
             </p>
             
             <div style="display: flex; gap: 1rem; justify-content: center; flex-wrap: wrap;">
-                <a href="mailto:{settings.LEAD_EMAIL}?subject=ARF%20Enterprise%20Demo%20Request&body=I%20saw%20the%20real%20ARF%20OSS%20demo%20and%20would%20like%20to%20discuss%20Enterprise%20capabilities." 
+                <a href="mailto:{settings.lead_email}?subject=ARF%20Enterprise%20Demo%20Request&body=I%20saw%20the%20real%20ARF%20OSS%20demo%20and%20would%20like%20to%20discuss%20Enterprise%20capabilities." 
                    style="background: white; color: #667eea; padding: 1rem 2rem; border-radius: 2rem; font-weight: bold; text-decoration: none; display: inline-block; margin: 0.5rem;">
-                    📧 {settings.LEAD_EMAIL}
+                    📧 {settings.lead_email}
                 </a>
-                <a href="{settings.CALENDLY_URL}" target="_blank" 
+                <a href="{settings.calendly_url}" target="_blank" 
                    style="background: #FFD700; color: #333; padding: 1rem 2rem; border-radius: 2rem; font-weight: bold; text-decoration: none; display: inline-block; margin: 0.5rem;">
                     📅 Schedule Technical Demo
                 </a>
@@ -1010,11 +996,10 @@ def create_lead_gen_ui():
         </div>
         """)
         
-        # Footer
         gr.HTML(f"""
         <div style="text-align: center; padding: 2rem; color: #666; border-top: 1px solid #eee;">
             <p>
-                📧 <a href="mailto:{settings.LEAD_EMAIL}" style="color: #667eea;">{settings.LEAD_EMAIL}</a> • 
+                📧 <a href="mailto:{settings.lead_email}" style="color: #667eea;">{settings.lead_email}</a> • 
                 🐙 <a href="https://github.com/petterjuan/agentic-reliability-framework" style="color: #667eea;">GitHub</a>
             </p>
             <p style="font-size: 0.9rem;">
@@ -1046,9 +1031,9 @@ if __name__ == "__main__":
     
     logger.info("="*60)
     logger.info("🚀 ARF OSS v3.3.9 Starting")
-    logger.info(f"📊 Data directory: {settings.DATA_DIR}")
-    logger.info(f"📧 Lead email: {settings.LEAD_EMAIL}")
-    logger.info(f"🔑 API Key: {settings.API_KEY[:8]}... (set in HF secrets)")
+    logger.info(f"📊 Data directory: {settings.data_dir}")
+    logger.info(f"📧 Lead email: {settings.lead_email}")
+    logger.info(f"🔑 API Key: {settings.api_key[:8]}... (set in HF secrets)")
     logger.info(f"🌐 Serving at: http://0.0.0.0:{port}")
     logger.info("="*60)