Update app.py

app.py

@@ -17,7 +17,7 @@ try:
     from 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
+    from components.search_engine import SearchEngine
 except ImportError:
     # Fallback for container environment
     from src.components.ai_core import AICore
@@ -94,7 +94,16 @@ try:
     
     # Initialize cocoon manager
     try:
-        from ..utils.cocoon_manager import CocoonManager
+        # Handle both direct execution and package import
+        try:
+            from ..utils.cocoon_manager import CocoonManager
+        except (ImportError, ValueError):
+            # Fallback for direct execution
+            import sys
+            import os
+            sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
+            from utils.cocoon_manager import CocoonManager
+        
         cocoon_manager = CocoonManager("./cocoons")
         cocoon_manager.load_cocoons()