Update app.py

app.py

@@ -9,11 +9,23 @@ import torch
 from concurrent.futures import ThreadPoolExecutor
 from datetime import datetime
 from transformers import AutoModelForCausalLM, AutoTokenizer
+
+# Add parent directory to path for local execution
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
-from src.components.ai_core import AICore
-from src.aegis_integration import AegisBridge
-from src.aegis_integration.config import AEGIS_CONFIG
-from src.components.search_engine import SearchEngine
+# Add src directory to path for container execution
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../..'))
+
+try:
+    from components.ai_core import AICore
+    from components.aegis_integration import AegisBridge
+    from components.aegis_integration.config import AEGIS_CONFIG
+    from components.search_engine import SearchEngine
+except ImportError:
+    # Fallback for container environment
+    from src.components.ai_core import AICore
+    from src.components.aegis_integration import AegisBridge
+    from src.components.aegis_integration.config import AEGIS_CONFIG
+    from src.components.search_engine import SearchEngine
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -70,10 +82,7 @@ try:
         # Initialize cognitive processor with default modes
         from cognitive_processor import CognitiveProcessor
         cognitive_modes = ["scientific", "creative", "quantum", "philosophical"]
-        ai_core.cognitive_processor = CognitiveProcessor(
-            modes=cognitive_modes,
-            quantum_state={"coherence": 0.5}
-        )
+        ai_core.cognitive_processor = CognitiveProcessor()
         logger.info(
             f"AI Core initialized successfully with modes: {cognitive_modes}"
         )