Spaces:

AumCoreAI
/

AumCore-AI

Sleeping

App Files Files Community

AumCoreAI commited on Dec 28, 2025

Commit

82e3567

verified ·

1 Parent(s): 3d0d89c

Update app.py

Browse files

Files changed (1) hide show

app.py +229 -5

app.py CHANGED Viewed

@@ -1,8 +1,9 @@
 # app.py - FULL VERBOSE 450+ LINES VERSION
 import os
 import uvicorn
 from fastapi import FastAPI, Form
-from fastapi.responses import HTMLResponse
 from groq import Groq
 # --- CORE INITIALIZATION ---
@@ -12,10 +13,11 @@ client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 # Configuration
 USERNAME = "AumCore AI"
-# Import System Orchestrator
 try:
-    from system_orchestrator import AumCoreMaster
     SYSTEM_ORCHESTRATOR_ENABLED = True
     orchestrator = None
     # Initialize orchestrator in background
@@ -23,18 +25,29 @@ try:
         global orchestrator
         orchestrator = AumCoreMaster()
         print("🚀 System Orchestrator initialized")
     @app.on_event("startup")
     async def startup_event():
-        import asyncio
         if SYSTEM_ORCHESTRATOR_ENABLED:
             task = asyncio.create_task(init_orchestrator())
-            # Don't wait for completion to avoid blocking startup
             print("⏳ Orchestrator initialization started...")
 except ImportError as e:
     print(f"⚠️ System Orchestrator not available: {e}")
     SYSTEM_ORCHESTRATOR_ENABLED = False
     orchestrator = None
 # HTML UI - VERBOSE, 450+ lines
@@ -249,14 +262,31 @@ body {
 .typing-dot:nth-child(2) { animation-delay: 0.2s; }
 .typing-dot:nth-child(3) { animation-delay: 0.4s; }
 @keyframes blink { 0%,80%,100%{opacity:0;} 40%{opacity:1;} }
 </style>
 </head>
 <body>
 <div class="sidebar">
 <button class="nav-item new-chat-btn" onclick="window.location.reload()"><i class="fas fa-plus"></i> New Chat</button>
 <div class="nav-item"><i class="fas fa-history"></i> History</div>
 <div class="mt-auto">
 <div class="nav-item reset-btn" onclick="confirmReset()"><i class="fas fa-trash-alt"></i> Reset Memory</div>
 <div class="nav-item"><i class="fas fa-cog"></i> Settings</div>
 </div>
 </div>
@@ -305,6 +335,81 @@ async function confirmReset(){
         }catch(e){alert("Reset failed: "+e.message);}
     }
 }
 // Send function
 async function send(){
     const input=document.getElementById('user-input');
@@ -427,6 +532,123 @@ async def process_system_task():
     except Exception as e:
         return {"error": f"Task processing failed: {str(e)}"}
 if __name__ == "__main__":
     print("="*60)
     print("🚀 AUMCORE AI - FINAL BUILD STARTING")
@@ -437,5 +659,7 @@ if __name__ == "__main__":
     print(f"💾 Database: TiDB Cloud")
     print(f"🎨 UI Features: Code formatting + Copy button")
     print(f"🔄 System Orchestrator: {'ENABLED' if SYSTEM_ORCHESTRATOR_ENABLED else 'DISABLED'}")
     print("="*60)
     uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info")

 # app.py - FULL VERBOSE 450+ LINES VERSION
 import os
 import uvicorn
+import asyncio
 from fastapi import FastAPI, Form
+from fastapi.responses import HTMLResponse, JSONResponse
 from groq import Groq
 # --- CORE INITIALIZATION ---
 # Configuration
 USERNAME = "AumCore AI"
+# Import System Orchestrator with Diagnostics
 try:
+    from system_orchestrator import AumCoreMaster, get_system_diagnostics
     SYSTEM_ORCHESTRATOR_ENABLED = True
+    ORCHESTRATOR_WITH_DIAGNOSTICS = True
     orchestrator = None
     # Initialize orchestrator in background
         global orchestrator
         orchestrator = AumCoreMaster()
         print("🚀 System Orchestrator initialized")
+        # Run initial diagnostics
+        try:
+            from system_orchestrator import get_system_diagnostics
+            diag_result = await get_system_diagnostics()
+            if diag_result.get("success"):
+                health = diag_result["diagnostics"].get("health_score", 0)
+                print(f"📊 Initial Diagnostics: Health Score {health}/100")
+            else:
+                print(f"⚠️ Diagnostics failed: {diag_result.get('error')}")
+        except Exception as e:
+            print(f"⚠️ Initial diagnostics error: {e}")
     @app.on_event("startup")
     async def startup_event():
         if SYSTEM_ORCHESTRATOR_ENABLED:
             task = asyncio.create_task(init_orchestrator())
             print("⏳ Orchestrator initialization started...")
 except ImportError as e:
     print(f"⚠️ System Orchestrator not available: {e}")
     SYSTEM_ORCHESTRATOR_ENABLED = False
+    ORCHESTRATOR_WITH_DIAGNOSTICS = False
     orchestrator = None
 # HTML UI - VERBOSE, 450+ lines
 .typing-dot:nth-child(2) { animation-delay: 0.2s; }
 .typing-dot:nth-child(3) { animation-delay: 0.4s; }
 @keyframes blink { 0%,80%,100%{opacity:0;} 40%{opacity:1;} }
+/* Diagnostics Health Indicator */
+.health-indicator {
+    display: inline-flex;
+    align-items: center;
+    gap: 8px;
+    padding: 6px 12px;
+    border-radius: 20px;
+    font-size: 14px;
+    font-weight: 600;
+}
+.health-green { background: #238636; color: white; }
+.health-yellow { background: #d29922; color: black; }
+.health-red { background: #da3633; color: white; }
+.health-value { font-family: 'Fira Code', monospace; }
 </style>
 </head>
 <body>
 <div class="sidebar">
 <button class="nav-item new-chat-btn" onclick="window.location.reload()"><i class="fas fa-plus"></i> New Chat</button>
+<div class="nav-item" onclick="checkSystemHealth()"><i class="fas fa-heartbeat"></i> System Health</div>
 <div class="nav-item"><i class="fas fa-history"></i> History</div>
 <div class="mt-auto">
 <div class="nav-item reset-btn" onclick="confirmReset()"><i class="fas fa-trash-alt"></i> Reset Memory</div>
+<div class="nav-item" onclick="runDiagnostics()"><i class="fas fa-stethoscope"></i> Run Diagnostics</div>
 <div class="nav-item"><i class="fas fa-cog"></i> Settings</div>
 </div>
 </div>
         }catch(e){alert("Reset failed: "+e.message);}
     }
 }
+// System Health Check
+async function checkSystemHealth(){
+    try{
+        const res=await fetch('/system/diagnostics/summary');
+        const data=await res.json();
+        if(data.success){
+            const health=data.health_score;
+            let healthClass='health-red';
+            if(health>=80) healthClass='health-green';
+            else if(health>=50) healthClass='health-yellow';
+            alert(`System Health: ${health}/100\\nStatus: ${data.status}\\nMemory: ${data.memory_used}%\\nCPU: ${data.cpu_used}%`);
+        }else{
+            alert('Health check failed: '+data.error);
+        }
+    }catch(e){
+        alert('Health check error: '+e.message);
+    }
+}
+// Run Full Diagnostics
+async function runDiagnostics(){
+    const log=document.getElementById('chat-log');
+    const typingId='diagnostics-'+Date.now();
+    log.innerHTML+=`<div class="message-wrapper" id="${typingId}"><div class="typing-indicator"><div class="typing-dot"></div><div class="typing-dot"></div><div class="typing-dot"></div> Running System Diagnostics...</div></div>`;
+    log.scrollTop=log.scrollHeight;
+    try{
+        const res=await fetch('/system/diagnostics/full');
+        const data=await res.json();
+        const typingElem=document.getElementById(typingId);
+        if(typingElem) typingElem.remove();
+        if(data.success){
+            const report=data.diagnostics;
+            const health=report.health_score;
+            let healthClass='health-red';
+            if(health>=80) healthClass='health-green';
+            else if(health>=50) healthClass='health-yellow';
+            let html=`<div class="message-wrapper">
+                <div class="bubble ai-text">
+                    <h3>📊 System Diagnostics Report</h3>
+                    <div class="health-indicator ${healthClass}">
+                        <i class="fas fa-heartbeat"></i>
+                        <span class="health-value">Health: ${health}/100</span>
+                        <span>(${report.status})</span>
+                    </div>
+                    <br>
+                    <strong>System Resources:</strong><br>
+                    • CPU: ${report.sections?.system_resources?.cpu?.usage_percent || 'N/A'}%<br>
+                    • Memory: ${report.sections?.system_resources?.memory?.used_percent || 'N/A'}%<br>
+                    • Disk: ${report.sections?.system_resources?.disk?.used_percent || 'N/A'}%<br>
+                    <br>
+                    <strong>Services:</strong><br>
+                    • Groq API: ${report.sections?.external_services?.groq_api?.status || 'N/A'}<br>
+                    • TiDB: ${report.sections?.external_services?.tidb_database?.status || 'N/A'}<br>
+                    <br>
+                    <strong>Logs:</strong><br>
+                    • Recent Errors: ${report.sections?.logs_analysis?.error_count || 0}<br>
+                    • Log Size: ${report.sections?.logs_analysis?.log_file_size_mb || 0} MB<br>
+                    <br>
+                    <small>Report ID: ${report.system_id}</small>
+                </div>
+            </div>`;
+            log.innerHTML+=html;
+        }else{
+            log.innerHTML+=`<div class="message-wrapper"><div class="error-message"><i class="fas fa-exclamation-circle"></i> Diagnostics failed: ${data.error}</div></div>`;
+        }
+    }catch(e){
+        const typingElem=document.getElementById(typingId);
+        if(typingElem) typingElem.remove();
+        log.innerHTML+=`<div class="message-wrapper"><div class="error-message"><i class="fas fa-exclamation-circle"></i> Diagnostics error: ${e.message}</div></div>`;
+    }
+    log.scrollTop=log.scrollHeight;
+}
 // Send function
 async function send(){
     const input=document.getElementById('user-input');
     except Exception as e:
         return {"error": f"Task processing failed: {str(e)}"}
+# --- LEVEL 2: SYSTEM DIAGNOSTICS ENDPOINTS ---
+@app.get("/system/diagnostics/summary")
+async def diagnostics_summary():
+    """Get system health summary"""
+    if not SYSTEM_ORCHESTRATOR_ENABLED or not ORCHESTRATOR_WITH_DIAGNOSTICS:
+        return JSONResponse(
+            status_code=503,
+            content={
+                "success": False,
+                "error": "Diagnostics not available",
+                "message": "System orchestrator with diagnostics not enabled"
+            }
+        )
+    try:
+        from system_orchestrator import get_system_diagnostics
+        result = await get_system_diagnostics()
+        if result.get("success"):
+            diag = result["diagnostics"]
+            return {
+                "success": True,
+                "health_score": diag.get("health_score", 0),
+                "status": diag.get("status", "UNKNOWN"),
+                "timestamp": diag.get("timestamp"),
+                "cpu_used": diag.get("sections", {}).get("system_resources", {}).get("cpu", {}).get("usage_percent", 0),
+                "memory_used": diag.get("sections", {}).get("system_resources", {}).get("memory", {}).get("used_percent", 0),
+                "disk_used": diag.get("sections", {}).get("system_resources", {}).get("disk", {}).get("used_percent", 0),
+                "groq_status": diag.get("sections", {}).get("external_services", {}).get("groq_api", {}).get("status", "UNKNOWN"),
+                "error_count": diag.get("sections", {}).get("logs_analysis", {}).get("error_count", 0),
+                "system_id": diag.get("system_id", "N/A")
+            }
+        else:
+            return JSONResponse(
+                status_code=500,
+                content={
+                    "success": False,
+                    "error": result.get("error", "Unknown diagnostics error"),
+                    "timestamp": result.get("timestamp")
+                }
+            )
+    except Exception as e:
+        return JSONResponse(
+            status_code=500,
+            content={
+                "success": False,
+                "error": f"Diagnostics failed: {str(e)}",
+                "timestamp": datetime.now().isoformat()
+            }
+        )
+@app.get("/system/diagnostics/full")
+async def diagnostics_full():
+    """Get complete system diagnostics report"""
+    if not SYSTEM_ORCHESTRATOR_ENABLED or not ORCHESTRATOR_WITH_DIAGNOSTICS:
+        return JSONResponse(
+            status_code=503,
+            content={
+                "success": False,
+                "error": "Diagnostics not available",
+                "message": "System orchestrator with diagnostics not enabled"
+            }
+        )
+    try:
+        from system_orchestrator import get_system_diagnostics
+        result = await get_system_diagnostics()
+        return result
+    except Exception as e:
+        return JSONResponse(
+            status_code=500,
+            content={
+                "success": False,
+                "error": f"Diagnostics failed: {str(e)}",
+                "timestamp": datetime.now().isoformat()
+            }
+        )
+@app.get("/system/diagnostics/history")
+async def diagnostics_history():
+    """Get diagnostics history (last 10 reports)"""
+    if not SYSTEM_ORCHESTRATOR_ENABLED or not ORCHESTRATOR_WITH_DIAGNOSTICS:
+        return JSONResponse(
+            status_code=503,
+            content={
+                "success": False,
+                "error": "Diagnostics not available"
+            }
+        )
+    try:
+        import json
+        if os.path.exists("data/system_state.json"):
+            with open("data/system_state.json", "r") as f:
+                data = json.load(f)
+                history = data.get("diagnostics_history", [])
+                return {
+                    "success": True,
+                    "count": len(history),
+                    "history": history,
+                    "latest": history[-1] if history else None
+                }
+        return {
+            "success": True,
+            "count": 0,
+            "history": [],
+            "message": "No diagnostics history found"
+        }
+    except Exception as e:
+        return JSONResponse(
+            status_code=500,
+            content={
+                "success": False,
+                "error": f"Failed to get history: {str(e)}"
+            }
+        )
 if __name__ == "__main__":
     print("="*60)
     print("🚀 AUMCORE AI - FINAL BUILD STARTING")
     print(f"💾 Database: TiDB Cloud")
     print(f"🎨 UI Features: Code formatting + Copy button")
     print(f"🔄 System Orchestrator: {'ENABLED' if SYSTEM_ORCHESTRATOR_ENABLED else 'DISABLED'}")
+    print(f"📊 Diagnostics Engine: {'ENABLED' if ORCHESTRATOR_WITH_DIAGNOSTICS else 'DISABLED'}")
+    print(f"🔍 New Endpoints: /system/diagnostics/summary, /system/diagnostics/full, /system/diagnostics/history")
     print("="*60)
     uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info")