Update modules/diagnostics.py

modules/diagnostics.py

@@ -6,7 +6,7 @@ from datetime import datetime
 
 def register_module(app, client, username):
     """Register diagnostics module with FastAPI app"""
-    router = APIRouter()
+    router = APIRouter(prefix="/system")
     
     @router.get("/diagnostics/health")
     async def diagnostics_health():
@@ -22,7 +22,7 @@ def register_module(app, client, username):
                 "system": {
                     "cpu_usage": cpu,
                     "memory_used": memory.percent,
-                    "memory_available": round(memory.available / (1024**3), 2),  # GB
+                    "memory_available": round(memory.available / (1024**3), 2),
                     "timestamp": datetime.now().isoformat()
                 }
             }
@@ -37,12 +37,9 @@ def register_module(app, client, username):
     async def full_diagnostics():
         """Complete system diagnostics"""
         try:
-            # Direct implementation without circular import
             cpu = psutil.cpu_percent()
             memory = psutil.virtual_memory()
             disk = psutil.disk_usage('/')
-            
-            # Check if app.py is running
             app_running = any('app.py' in p.info().get('cmdline', []) for p in psutil.process_iter(['cmdline']))
             
             return {
@@ -83,8 +80,8 @@ def register_module(app, client, username):
     
     app.include_router(router)
     print("✅ Diagnostics module registered with FastAPI")
+    return {"status": "registered"}
 
-# Export function for direct imports
 async def get_basic_metrics():
     """Get basic system metrics"""
     try: