Update components/CodriaoCore.py

components/CodriaoCore.py

@@ -1,10 +1,9 @@
-
 import aiohttp
 import json
 import logging
 import torch
 import faiss
-import numpy 
+import numpy as np
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from typing import List, Dict, Any
 from cryptography.fernet import Fernet
@@ -25,13 +24,15 @@ from components.multi_model_analyzer import MultiAgentSystem
 # Codriao's enhanced modules
 from codriao_tb_module import CodriaoHealthModule
 from secure_memory_loader import load_secure_memory_module
+secure_memory = load_secure_memory_module()
+SecureMemorySession = secure_memory.SecureMemorySession
+
 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 secure_memory_module 
 
 class CodriaoCore:
     def __init__(self, config_path: str = "config.json"):
@@ -45,10 +46,6 @@ class CodriaoCore:
         self.http_session = aiohttp.ClientSession()
         self.database = Database()
 
-        # 🧠 Replace static SecureMemory with dynamic, temp version
-        secure_memory_module = load_secure_memory_module()
-        SecureMemorySession = secure_memory_module.SecureMemorySession
-
         # Cognitive & ethical subsystems
         self.sentiment_analyzer = EnhancedSentimentAnalyzer()
         self.self_improving_ai = SelfImprovingAI()
@@ -75,14 +72,12 @@ class CodriaoCore:
 
     async def generate_response(self, query: str, user_id: int) -> Dict[str, Any]:
         try:
-            # Ethical Safety
             check = self.ethical_filter.analyze_query(query)
             if check["status"] == "blocked":
                 return {"error": check["reason"]}
             if check["status"] == "flagged":
                 logger.warning(check["warning"])
 
-            # Optional: Trigger TB diagnostics by user request
             if any(trigger in query.lower() for trigger in ["tb check", "run tb diagnostics", "tb test"]):
                 result = await self.run_tb_diagnostics("tb_image.jpg", "tb_cough.wav", user_id)
                 return result
@@ -117,13 +112,11 @@ class CodriaoCore:
         result_filename = save_result(result)
         result["shareable_link"] = f"https://huggingface.co/spaces/Raiff1982/codriao/blob/main/results/{result_filename}"
 
-        # 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."
+            result["next_steps"] = "🔍 Consider additional testing for confirmation."
 
-        # Multi-language support
         if language != "en":
             try:
                 translated_result = GoogleTranslator(source="auto", target=language).translate(json.dumps(result))
@@ -147,7 +140,7 @@ class CodriaoCore:
                 self.speech_engine.say(response)
                 self.speech_engine.runAndWait()
             except:
-                pass  # Ignore if running in a non-audio environment
+                pass
 
     def generate_jwt(self, user_id: int):
         payload = {