Update app.py

app.py

@@ -7,8 +7,9 @@ import asyncio
 import importlib.util
 import json
 from pathlib import Path
-from fastapi import FastAPI, Form, APIRouter
-from fastapi.responses import HTMLResponse, JSONResponse
+from fastapi import FastAPI, Form
+from fastapi.responses import HTMLResponse
+from contextlib import asynccontextmanager
 from groq import Groq
 
 # ============================================
@@ -132,29 +133,70 @@ class ModuleManager:
         }
 
 # ============================================
-# 3. CORE FASTAPI APPLICATION
+# 3. LIFESPAN MANAGEMENT (MODERN APPROACH)
+# ============================================
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """Modern lifespan handler for startup/shutdown events"""
+    # Startup code
+    print("=" * 60)
+    print("🚀 AUMCORE AI - ULTIMATE FINAL VERSION")
+    print("=" * 60)
+    print(f"📁 Version: {AumCoreConfig.VERSION}")
+    print(f"👤 Username: {AumCoreConfig.USERNAME}")
+    print(f"🌐 Server: http://{AumCoreConfig.HOST}:{AumCoreConfig.PORT}")
+    print(f"🤖 AI Model: llama-3.3-70b-versatile")
+    print(f"💾 Database: TiDB Cloud")
+    print(f"🎨 UI Features: Code formatting + Copy button")
+    
+    # Load all modules
+    if hasattr(app.state, 'module_manager'):
+        app.state.module_manager.load_all_modules()
+    
+    # Initial health check
+    print("\n🔍 Initial System Check:")
+    print(f"   Groq API: {'✅ Available' if hasattr(app.state, 'groq_available') and app.state.groq_available else '❌ Not Available'}")
+    print(f"   Modules: {len(app.state.module_manager.loaded_modules) if hasattr(app.state, 'module_manager') else 0} loaded")
+    print(f"   Directories: All created")
+    print("=" * 60)
+    print("✅ System ready! Waiting for requests...")
+    print("=" * 60)
+    
+    yield  # Application runs here
+    
+    # Shutdown code
+    print("\n🛑 System shutting down...")
+    print("✅ Cleanup completed")
+
+# ============================================
+# 4. CORE FASTAPI APPLICATION
 # ============================================
 
 app = FastAPI(
     title="AumCore AI",
     description="Advanced Modular AI Assistant System",
-    version=AumCoreConfig.VERSION
+    version=AumCoreConfig.VERSION,
+    lifespan=lifespan
 )
 
 # Initialize Groq client
 try:
     client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
     GROQ_AVAILABLE = True
+    app.state.groq_available = True
 except Exception as e:
     print(f"⚠️ Groq client initialization failed: {e}")
     client = None
     GROQ_AVAILABLE = False
+    app.state.groq_available = False
 
 # Initialize Module Manager
 module_manager = ModuleManager(app, client)
+app.state.module_manager = module_manager
 
 # ============================================
-# 4. CORE UI (NEVER CHANGES)
+# 5. CORE UI (NEVER CHANGES)
 # ============================================
 
 HTML_UI = '''
@@ -621,7 +663,7 @@ document.addEventListener('DOMContentLoaded',()=>{const input=document.getElemen
 '''
 
 # ============================================
-# 5. CORE ENDPOINTS (NEVER CHANGES)
+# 6. CORE ENDPOINTS (NEVER CHANGES)
 # ============================================
 
 @app.get("/", response_class=HTMLResponse)
@@ -710,7 +752,7 @@ async def chat(message: str = Form(...)):
         return {"response": error_msg}
 
 # ============================================
-# 6. SYSTEM MANAGEMENT ENDPOINTS
+# 7. SYSTEM MANAGEMENT ENDPOINTS
 # ============================================
 
 @app.get("/system/health")
@@ -721,13 +763,13 @@ async def system_health():
         "timestamp": asyncio.get_event_loop().time(),
         "version": AumCoreConfig.VERSION,
         "status": "OPERATIONAL",
-        "modules_loaded": len(module_manager.loaded_modules),
-        "groq_available": GROQ_AVAILABLE,
+        "modules_loaded": len(app.state.module_manager.loaded_modules),
+        "groq_available": app.state.groq_available,
         "health_score": 95  # Default high score
     }
     
     # Add module-specific health if available
-    diagnostics_module = module_manager.get_module("sys_diagnostics")
+    diagnostics_module = app.state.module_manager.get_module("sys_diagnostics")
     if diagnostics_module and hasattr(diagnostics_module, 'get_health'):
         try:
             module_health = await diagnostics_module.get_health()
@@ -742,14 +784,14 @@ async def modules_status():
     """Get status of all loaded modules"""
     return {
         "success": True,
-        "total": len(module_manager.loaded_modules),
+        "total": len(app.state.module_manager.loaded_modules),
         "modules": [
             {
                 "name": name,
                 "status": info["status"],
                 "active": True
             }
-            for name, info in module_manager.loaded_modules.items()
+            for name, info in app.state.module_manager.loaded_modules.items()
         ]
     }
 
@@ -769,9 +811,9 @@ async def system_info():
             "code_generation": True,
             "hindi_english": True,
             "memory_storage": True,
-            "system_monitoring": "diagnostics" in module_manager.loaded_modules,
-            "automated_testing": "testing" in module_manager.loaded_modules,
-            "task_orchestration": "orchestrator" in module_manager.loaded_modules
+            "system_monitoring": "diagnostics" in app.state.module_manager.loaded_modules,
+            "automated_testing": "testing" in app.state.module_manager.loaded_modules,
+            "task_orchestration": "orchestrator" in app.state.module_manager.loaded_modules
         },
         "endpoints": [
             "/", "/chat", "/reset",
@@ -779,41 +821,6 @@ async def system_info():
         ]
     }
 
-# ============================================
-# 7. STARTUP AND SHUTDOWN EVENTS
-# ============================================
-
-@app.on_event("startup")
-async def startup_event():
-    """Initialize system on startup"""
-    print("=" * 60)
-    print("🚀 AUMCORE AI - ULTIMATE FINAL VERSION")
-    print("=" * 60)
-    print(f"📁 Version: {AumCoreConfig.VERSION}")
-    print(f"👤 Username: {AumCoreConfig.USERNAME}")
-    print(f"🌐 Server: http://{AumCoreConfig.HOST}:{AumCoreConfig.PORT}")
-    print(f"🤖 AI Model: llama-3.3-70b-versatile")
-    print(f"💾 Database: TiDB Cloud")
-    print(f"🎨 UI Features: Code formatting + Copy button")
-    
-    # Load all modules
-    module_manager.load_all_modules()
-    
-    # Initial health check
-    print("\n🔍 Initial System Check:")
-    print(f"   Groq API: {'✅ Available' if GROQ_AVAILABLE else '❌ Not Available'}")
-    print(f"   Modules: {len(module_manager.loaded_modules)} loaded")
-    print(f"   Directories: All created")
-    print("=" * 60)
-    print("✅ System ready! Waiting for requests...")
-    print("=" * 60)
-
-@app.on_event("shutdown")
-async def shutdown_event():
-    """Cleanup on shutdown"""
-    print("\n🛑 System shutting down...")
-    print("✅ Cleanup completed")
-
 # ============================================
 # 8. MAIN EXECUTION
 # ============================================