Update modules/CodriaoCore.py

modules/CodriaoCore.py

@@ -1,3 +1,4 @@
+
 import aiohttp
 import json
 import logging
@@ -18,18 +19,19 @@ from deep_translator import GoogleTranslator
 from components.adaptive_learning import AdaptiveLearningEnvironment
 from components.real_time_data import RealTimeDataIntegrator
 from components.sentiment_analysis import EnhancedSentimentAnalyzer
-from components.self_improving_ai import SelfImprovingAI
-from components.multi_agent import MultiAgentSystem
+from components.self_improving_ai import SelfImprovingAI 
+from components.multi_model_analyzer import MultiAgentSystem
 
 # Codriao's enhanced modules
 from codriao_tb_module import CodriaoHealthModule
-from secure_memory import SecureMemorySession
+from secure_memory_loader import load_secure_memory_module
 from ethical_filter import EthicalFilter
 from results_store import save_result
 
 # Utilities
 from utils.database import Database
 from utils.logger import logger
+from utils.secure_memory_loader import load_secure_memory_module 
 
 class CodriaoCore:
     def __init__(self, config_path: str = "config.json"):
@@ -43,6 +45,10 @@ class CodriaoCore:
         self.http_session = aiohttp.ClientSession()
         self.database = Database()
 
+        # 🧠 Replace static SecureMemory with dynamic, temp version
+        SecureMemorySession = load_secure_memory_module()
+        self.secure_memory = SecureMemorySession(self._encryption_key)
+
         # Cognitive & ethical subsystems
         self.sentiment_analyzer = EnhancedSentimentAnalyzer()
         self.self_improving_ai = SelfImprovingAI()
@@ -113,14 +119,17 @@ class CodriaoCore:
 
         # Auto-escalation for HIGH risk
         if result["tb_risk"] == "HIGH":
-            result["next_steps"] = "⚠️ Immediate follow-up required. Please visit a healthcare provider."
+            result["next_steps"] = "⚠️ Immediate follow-up required. Please visit a healthcare provider."
         elif result["tb_risk"] == "MEDIUM":
             result["next_steps"] = "🔍 Consider additional testing for confirmation."
 
         # Multi-language support
         if language != "en":
-            translated_result = GoogleTranslator(source="auto", target=language).translate(json.dumps(result))
-            return json.loads(translated_result)
+            try:
+                translated_result = GoogleTranslator(source="auto", target=language).translate(json.dumps(result))
+                return json.loads(translated_result)
+            except Exception as e:
+                result["translation_error"] = str(e)
 
         return result
 
@@ -152,3 +161,6 @@ class CodriaoCore:
             return decode(token, self.jwt_secret, algorithms=["HS256"])
         except ExpiredSignatureError:
             return None
+
+    async def shutdown(self):
+        await self.http_session.close()