Update app.py

app.py

@@ -32,6 +32,108 @@ logger = logging.getLogger(__name__)
 # Add parent directory to path
 sys.path.insert(0, str(Path(__file__).parent))
 
+# ===========================================
+# ARF INSTALLATION CHECK SYSTEM - NEW
+# ===========================================
+def check_arf_installation():
+    """
+    Check if real ARF packages are installed
+    Returns detailed installation status
+    """
+    results = {
+        "oss_installed": False,
+        "enterprise_installed": False,
+        "oss_version": None,
+        "enterprise_version": None,
+        "oss_edition": "unknown",
+        "oss_license": "unknown",
+        "execution_allowed": False,
+        "recommendations": [],
+        "badges": {
+            "oss": {"text": "⚠️ Mock ARF", "color": "#f59e0b", "icon": "⚠️"},
+            "enterprise": {"text": "🔒 Enterprise Required", "color": "#64748b", "icon": "🔒"}
+        },
+        "timestamp": datetime.datetime.now().isoformat()
+    }
+    
+    # Check OSS package
+    try:
+        import agentic_reliability_framework as arf_oss
+        results["oss_installed"] = True
+        results["oss_version"] = getattr(arf_oss, '__version__', '3.3.7')
+        
+        # Try to get more info
+        try:
+            results["oss_edition"] = arf_oss.OSS_EDITION
+            results["oss_license"] = arf_oss.OSS_LICENSE
+            results["execution_allowed"] = arf_oss.EXECUTION_ALLOWED
+        except Exception as e:
+            logger.debug(f"Could not get OSS details: {e}")
+            
+        results["badges"]["oss"] = {
+            "text": f"✅ ARF OSS v{results['oss_version']}", 
+            "color": "#10b981", 
+            "icon": "✅"
+        }
+        
+        logger.info(f"✅ ARF OSS v{results['oss_version']} detected")
+        
+    except ImportError as e:
+        results["recommendations"].append(
+            "Install real ARF OSS: `pip install agentic-reliability-framework==3.3.7`"
+        )
+        logger.info("⚠️ ARF OSS not installed - using mock mode")
+    
+    # Check Enterprise package
+    try:
+        import arf_enterprise
+        results["enterprise_installed"] = True
+        results["enterprise_version"] = getattr(arf_enterprise, '__version__', '1.0.2')
+        results["badges"]["enterprise"] = {
+            "text": f"🚀 Enterprise v{results['enterprise_version']}", 
+            "color": "#8b5cf6", 
+            "icon": "🚀"
+        }
+        logger.info(f"✅ ARF Enterprise v{results['enterprise_version']} detected")
+    except ImportError as e:
+        results["recommendations"].append(
+            "Install ARF Enterprise: `pip install agentic-reliability-enterprise` (requires license)"
+        )
+        logger.info("⚠️ ARF Enterprise not installed - using simulation")
+    
+    return results
+
+# Global installation status cache
+_installation_status = None
+
+def get_installation_status():
+    """Get cached installation status"""
+    global _installation_status
+    if _installation_status is None:
+        _installation_status = check_arf_installation()
+    return _installation_status
+
+def get_installation_badges():
+    """Get formatted badge HTML for UI"""
+    installation = get_installation_status()
+    oss_badge = installation["badges"]["oss"]
+    enterprise_badge = installation["badges"]["enterprise"]
+    
+    return f"""
+    <div style="display: flex; justify-content: center; gap: 10px; margin-top: 10px; flex-wrap: wrap;">
+        <span style="padding: 4px 12px; background: {oss_badge['color']}; 
+                    color: white; border-radius: 20px; font-size: 12px; font-weight: bold; 
+                    display: flex; align-items: center; gap: 6px;">
+            {oss_badge['icon']} {oss_badge['text']}
+        </span>
+        <span style="padding: 4px 12px; background: {enterprise_badge['color']}; 
+                    color: white; border-radius: 20px; font-size: 12px; font-weight: bold;
+                    display: flex; align-items: center; gap: 6px;">
+            {enterprise_badge['icon']} {enterprise_badge['text']}
+        </span>
+    </div>
+    """
+
 # ===========================================
 # ASYNC UTILITIES - ENHANCED VERSION
 # ===========================================
@@ -511,7 +613,7 @@ def update_scenario_display(scenario_name: str) -> tuple:
     )
 
 # ===========================================
-# TRUE ARF ANALYSIS HANDLER - UPDATED VERSION
+# TRUE ARF ANALYSIS HANDLER - UPDATED WITH REAL ARF INDICATORS
 # ===========================================
 @AsyncRunner.async_to_sync
 async def run_true_arf_analysis(scenario_name: str):
@@ -524,7 +626,11 @@ async def run_true_arf_analysis(scenario_name: str):
         if not scenario:
             raise ValueError(f"Scenario '{scenario_name}' not found")
         
-        # Use TrueARF337Orchestrator
+        # Check installation status
+        installation = get_installation_status()
+        real_arf_available = installation["oss_installed"]
+        
+        # Use TrueARF337Orchestrator if available
         orchestrator = get_components()["DemoOrchestrator"]()
         analysis = await orchestrator.analyze_incident(scenario_name, scenario)
         
@@ -611,7 +717,7 @@ async def run_true_arf_analysis(scenario_name: str):
             success_rate = healing_intent.get("success_rate", 0.87)
             
             oss_results = {
-                "status": "✅ OSS Analysis Complete (Enhanced Mock)",
+                "status": "⚠️ Enhanced Mock Analysis",
                 "arf_version": "mock",
                 "scenario": scenario_name,
                 "confidence": decision_confidence,
@@ -633,51 +739,36 @@ async def run_true_arf_analysis(scenario_name: str):
                 "install_command": "pip install agentic-reliability-framework==3.3.7"
             }
         
-        # Update agent status HTML
-        detection_html = f"""
-        <div style="border: 2px solid #3b82f6; border-radius: 14px; padding: 18px; background: #eff6ff; text-align: center; min-height: 180px; display: flex; flex-direction: column; align-items: center; justify-content: center;">
-            <div style="font-size: 32px; margin-bottom: 10px;">🕵️‍♂️</div>
-            <div style="width: 100%;">
-                <h4 style="margin: 0 0 8px 0; font-size: 16px; color: #1e293b;">Detection Agent</h4>
-                <p style="font-size: 13px; color: #475569; margin-bottom: 12px; line-height: 1.4;">Anomaly detected: <strong>{detection_confidence:.1%} confidence</strong></p>
-                <div style="display: flex; justify-content: space-around; margin-bottom: 12px;">
-                    <span style="font-size: 11px; padding: 3px 8px; background: rgba(255, 255, 255, 0.8); border-radius: 6px; color: #475569; font-weight: 500;">Response: {detection_time_seconds:.1f}s</span>
-                    <span style="font-size: 11px; padding: 3px 8px; background: rgba(255, 255, 255, 0.8); border-radius: 6px; color: #475569; font-weight: 500;">True ARF: {'✅' if true_oss_used else '⚠️'}</span>
-                </div>
-                <div style="display: inline-block; padding: 5px 14px; background: linear-gradient(135deg, #10b981 0%, #059669 100%); border-radius: 20px; font-size: 12px; font-weight: bold; color: white; text-transform: uppercase; letter-spacing: 0.5px;">ACTIVE</div>
-            </div>
-        </div>
-        """
+        # Create agent HTML with real ARF indicators
+        detection_html = create_agent_html(
+            agent_name="Detection",
+            status=f"Anomaly detected: <strong>{detection_confidence:.1%} confidence</strong>",
+            metrics=f"""
+            <span style="font-size: 11px; padding: 3px 8px; background: rgba(255, 255, 255, 0.8); border-radius: 6px; color: #475569; font-weight: 500;">Response: {detection_time_seconds:.1f}s</span>
+            <span style="font-size: 11px; padding: 3px 8px; background: rgba(255, 255, 255, 0.8); border-radius: 6px; color: #475569; font-weight: 500;">True ARF: {'✅' if true_oss_used else '⚠️'}</span>
+            """,
+            is_real_arf=true_oss_used
+        )
         
-        recall_html = f"""
-        <div style="border: 2px solid #8b5cf6; border-radius: 14px; padding: 18px; background: #f5f3ff; text-align: center; min-height: 180px; display: flex; flex-direction: column; align-items: center; justify-content: center;">
-            <div style="font-size: 32px; margin-bottom: 10px;">🧠</div>
-            <div style="width: 100%;">
-                <h4 style="margin: 0 0 8px 0; font-size: 16px; color: #1e293b;">Recall Agent</h4>
-                <p style="font-size: 13px; color: #475569; margin-bottom: 12px; line-height: 1.4;"><strong>{similar_count} similar incidents</strong> found in RAG memory</p>
-                <div style="display: flex; justify-content: space-around; margin-bottom: 12px;">
-                    <span style="font-size: 11px; padding: 3px 8px; background: rgba(255, 255, 255, 0.8); border-radius: 6px; color: #475569; font-weight: 500;">Recall: 92%</span>
-                    <span style="font-size: 11px; padding: 3px 8px; background: rgba(255, 255, 255, 0.8); border-radius: 6px; color: #475569; font-weight: 500;">Patterns: {similar_count}</span>
-                </div>
-                <div style="display: inline-block; padding: 5px 14px; background: linear-gradient(135deg, #10b981 0%, #059669 100%); border-radius: 20px; font-size: 12px; font-weight: bold; color: white; text-transform: uppercase; letter-spacing: 0.5px;">ACTIVE</div>
-            </div>
-        </div>
-        """
+        recall_html = create_agent_html(
+            agent_name="Recall",
+            status=f"<strong>{similar_count} similar incidents</strong> found in RAG memory",
+            metrics=f"""
+            <span style="font-size: 11px; padding: 3px 8px; background: rgba(255, 255, 255, 0.8); border-radius: 6px; color: #475569; font-weight: 500;">Recall: 92%</span>
+            <span style="font-size: 11px; padding: 3px 8px; background: rgba(255, 255, 255, 0.8); border-radius: 6px; color: #475569; font-weight: 500;">Patterns: {similar_count}</span>
+            """,
+            is_real_arf=true_oss_used
+        )
         
-        decision_html = f"""
-        <div style="border: 2px solid #10b981; border-radius: 14px; padding: 18px; background: #f0fdf4; text-align: center; min-height: 180px; display: flex; flex-direction: column; align-items: center; justify-content: center;">
-            <div style="font-size: 32px; margin-bottom: 10px;">🎯</div>
-            <div style="width: 100%;">
-                <h4 style="margin: 0 0 8px 0; font-size: 16px; color: #1e293b;">Decision Agent</h4>
-                <p style="font-size: 13px; color: #475569; margin-bottom: 12px; line-height: 1.4;">Generating healing intent with <strong>{decision_confidence:.1%} confidence</strong></p>
-                <div style="display: flex; justify-content: space-around; margin-bottom: 12px;">
-                    <span style="font-size: 11px; padding: 3px 8px; background: rgba(255, 255, 255, 0.8); border-radius: 6px; color: #475569; font-weight: 500;">Success Rate: {success_rate:.1%}</span>
-                    <span style="font-size: 11px; padding: 3px 8px; background: rgba(255, 255, 255, 0.8); border-radius: 6px; color: #475569; font-weight: 500;">Enterprise: {'✅' if enterprise_simulated else '⚠️'}</span>
-                </div>
-                <div style="display: inline-block; padding: 5px 14px; background: linear-gradient(135deg, #10b981 0%, #059669 100%); border-radius: 20px; font-size: 12px; font-weight: bold; color: white; text-transform: uppercase; letter-spacing: 0.5px;">ACTIVE</div>
-            </div>
-        </div>
-        """
+        decision_html = create_agent_html(
+            agent_name="Decision",
+            status=f"Generating healing intent with <strong>{decision_confidence:.1%} confidence</strong>",
+            metrics=f"""
+            <span style="font-size: 11px; padding: 3px 8px; background: rgba(255, 255, 255, 0.8); border-radius: 6px; color: #475569; font-weight: 500;">Success Rate: {success_rate:.1%}</span>
+            <span style="font-size: 11px; padding: 3px 8px; background: rgba(255, 255, 255, 0.8); border-radius: 6px; color: #475569; font-weight: 500;">Enterprise: {'✅' if enterprise_simulated else '⚠️'}</span>
+            """,
+            is_real_arf=true_oss_used
+        )
         
         logger.info(f"Analysis completed successfully for {scenario_name} (True ARF: {real_arf_version})")
         return (
@@ -712,6 +803,43 @@ async def run_true_arf_analysis(scenario_name: str):
             error_results, []
         )
 
+def create_agent_html(agent_name: str, status: str, metrics: str, is_real_arf: bool = True):
+    """Create agent HTML with real ARF indicators"""
+    icons = {
+        "Detection": "🕵️‍♂️",
+        "Recall": "🧠",
+        "Decision": "🎯"
+    }
+    
+    real_arf_badge = """
+    <span style="position: absolute; top: 8px; right: 8px; padding: 2px 6px; background: #10b981; 
+                color: white; border-radius: 4px; font-size: 10px; font-weight: bold;">
+        ✅ REAL ARF
+    </span>
+    """ if is_real_arf else """
+    <span style="position: absolute; top: 8px; right: 8px; padding: 2px 6px; background: #f59e0b; 
+                color: white; border-radius: 4px; font-size: 10px; font-weight: bold;">
+        ⚠️ MOCK
+    </span>
+    """
+    
+    return f"""
+    <div style="border: 2px solid {'#10b981' if is_real_arf else '#f59e0b'}; border-radius: 14px; padding: 18px; background: {'#f0fdf4' if is_real_arf else '#fef3c7'}; text-align: center; min-height: 180px; display: flex; flex-direction: column; align-items: center; justify-content: center; position: relative;">
+        {real_arf_badge}
+        <div style="font-size: 32px; margin-bottom: 10px;">{icons.get(agent_name, '🤖')}</div>
+        <div style="width: 100%;">
+            <h4 style="margin: 0 0 8px 0; font-size: 16px; color: #1e293b;">{agent_name} Agent</h4>
+            <p style="font-size: 13px; color: #475569; margin-bottom: 12px; line-height: 1.4;">{status}</p>
+            <div style="display: flex; justify-content: space-around; margin-bottom: 12px;">
+                {metrics}
+            </div>
+            <div style="display: inline-block; padding: 5px 14px; background: linear-gradient(135deg, {'#10b981' if is_real_arf else '#f59e0b'} 0%, {'#059669' if is_real_arf else '#d97706'} 100%); border-radius: 20px; font-size: 12px; font-weight: bold; color: white; text-transform: uppercase; letter-spacing: 0.5px;">
+                {'ACTIVE (REAL)' if is_real_arf else 'MOCK'}
+            </div>
+        </div>
+    </div>
+    """
+
 # ===========================================
 # ENTERPRISE EXECUTION HANDLER - UPDATED FOR TRUE ARF
 # ===========================================
@@ -1184,15 +1312,16 @@ def create_demo_interface():
         # ============ TAB 3 HANDLERS ============
         
         def validate_license():
+            installation = get_installation_status()
             return {
-                "status": "✅ Valid",
+                "status": "✅ OSS Installed" if installation["oss_installed"] else "⚠️ OSS Not Installed",
                 "tier": "OSS (Apache 2.0)",
-                "expires": "Never (OSS)",
-                "message": "OSS license validated successfully",
-                "arf_version": "3.3.7",
-                "novel_execution": "Enterprise Only",
-                "rollback_guarantees": "Enterprise Only",
-                "upgrade_cta": "Contact sales@arf.dev for Enterprise trial"
+                "oss_version": installation["oss_version"] or "Not installed",
+                "enterprise_installed": installation["enterprise_installed"],
+                "enterprise_version": installation["enterprise_version"] or "Not installed",
+                "execution_allowed": installation["execution_allowed"],
+                "recommendations": installation["recommendations"],
+                "badges": installation["badges"]
             }
         
         def start_trial():
@@ -1223,6 +1352,7 @@ def create_demo_interface():
         upgrade_btn.click(fn=upgrade_license, outputs=[license_display])
         
         def update_mcp_mode(mode):
+            installation = get_installation_status()
             mode_info = {
                 "advisory": {
                     "current_mode": "advisory",
@@ -1230,7 +1360,8 @@ def create_demo_interface():
                     "features": ["Incident analysis", "RAG similarity", "HealingIntent creation"],
                     "arf_version": "3.3.7 OSS",
                     "package": "agentic-reliability-framework==3.3.7",
-                    "license": "Apache 2.0"
+                    "license": "Apache 2.0",
+                    "installed": installation["oss_installed"]
                 },
                 "approval": {
                     "current_mode": "approval",
@@ -1238,7 +1369,8 @@ def create_demo_interface():
                     "features": ["All OSS features", "Approval workflows", "Audit trail", "Compliance", "Enhanced healing policies"],
                     "arf_version": "3.3.7 Enterprise",
                     "package": "arf_enterprise",
-                    "license": "Commercial"
+                    "license": "Commercial",
+                    "installed": installation["enterprise_installed"]
                 },
                 "autonomous": {
                     "current_mode": "autonomous",
@@ -1246,7 +1378,8 @@ def create_demo_interface():
                     "features": ["All approval features", "Auto-execution", "Predictive healing", "ML optimization", "Novel execution protocols"],
                     "arf_version": "3.3.7 Enterprise+",
                     "package": "arf_enterprise[plus]",
-                    "license": "Commercial Plus"
+                    "license": "Commercial Plus",
+                    "installed": installation["enterprise_installed"]
                 }
             }
             return mode_info.get(mode, mode_info["advisory"])
@@ -1341,14 +1474,24 @@ def main():
     """Main entry point - Hugging Face Spaces compatible"""
     print("🚀 Starting ARF Ultimate Investor Demo v3.8.0 with TRUE ARF v3.3.7...")
     print("=" * 70)
+    
+    # Check installation status first
+    installation = get_installation_status()
+    print(f"📦 Package Status:")
+    print(f"  • ARF OSS: {'✅ v' + installation['oss_version'] if installation['oss_installed'] else '⚠️ Not installed'}")
+    print(f"  • Enterprise: {'✅ v' + installation['enterprise_version'] if installation['enterprise_installed'] else '⚠️ Not installed'}")
+    print(f"  • Execution Allowed: {'✅ Yes' if installation['execution_allowed'] else '❌ No (OSS only)'}")
+    
     print(f"📊 Mode: {settings.arf_mode.upper()}")
     print(f"🤖 Using TRUE ARF: {settings.use_true_arf}")
     print(f"🎯 Default Scenario: {settings.default_scenario}")
     print(f"🏢 ARF Version: 3.3.7 with True OSS + Enterprise Simulation")
     print("=" * 70)
-    print("📦 Packages:")
-    print("  • OSS: agentic-reliability-framework==3.3.7")
-    print("  • Enterprise: arf_enterprise (simulated)")
+    
+    if installation["recommendations"]:
+        print("💡 Recommendations:")
+        for rec in installation["recommendations"]:
+            print(f"  • {rec}")
     print("=" * 70)
     
     import gradio as gr