Create modules/diagnostics.py

modules/diagnostics.py

@@ -0,0 +1,56 @@
+# modules/diagnostics.py - System Diagnostics Module
+from fastapi import APIRouter
+import psutil
+from datetime import datetime
+
+def register_module(app, client, username):
+    """Register diagnostics module with FastAPI app"""
+    router = APIRouter()
+    
+    @router.get("/diagnostics/health")
+    async def diagnostics_health():
+        """Basic system health check"""
+        cpu = psutil.cpu_percent()
+        memory = psutil.virtual_memory()
+        
+        return {
+            "module": "diagnostics",
+            "status": "active",
+            "system": {
+                "cpu_usage": cpu,
+                "memory_used": memory.percent,
+                "memory_available": memory.available / (1024**3),  # GB
+                "timestamp": datetime.now().isoformat()
+            }
+        }
+    
+    @router.get("/diagnostics/full")
+    async def full_diagnostics():
+        """Complete system diagnostics"""
+        try:
+            from modules.orchestrator import get_system_diagnostics
+            return await get_system_diagnostics()
+        except ImportError:
+            return {
+                "module": "diagnostics",
+                "status": "fallback",
+                "message": "Using basic diagnostics",
+                "basic_metrics": {
+                    "cpu": psutil.cpu_percent(),
+                    "memory": psutil.virtual_memory().percent,
+                    "disk": psutil.disk_usage('/').percent
+                }
+            }
+    
+    app.include_router(router)
+    print("✅ Diagnostics module registered with FastAPI")
+
+# Export function for direct imports
+async def get_basic_metrics():
+    """Get basic system metrics"""
+    return {
+        "cpu": psutil.cpu_percent(),
+        "memory": psutil.virtual_memory().percent,
+        "disk": psutil.disk_usage('/').percent,
+        "timestamp": datetime.now().isoformat()
+    }