Update app.py

app.py

@@ -1,68 +1,99 @@
 """
 LibBee - Main Application Entry Point
 """
-import sys
-import os
-
-# Add current directory to path for src imports
-sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
-
+import logging
+from contextlib import asynccontextmanager
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 
-# Import your routers
-from src.api import agent
-from src.api import search
-from src.api import admin
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# ── Singletons (imported by routers via `from app import ...`) ──────────────
+from src.services.rag_service import RAGService
+from src.services.cache_service import CacheService
+
+rag_service = RAGService()
+cache_service = CacheService()
+
+# ── Lifespan ────────────────────────────────────────────────────────────────
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # STARTUP
+    from src.config import get_settings
+    from src.services.staff_service import build_staff_index
+
+    settings = get_settings()
+
+    # 1. Build in-memory staff role index (instant)
+    build_staff_index()
+    logger.info("✅ Staff index built")
+
+    # 2. Vectorize knowledge base (calls OpenAI embeddings API ~10-15s)
+    if settings.openai_api_key:
+        await rag_service.initialize(openai_api_key=settings.openai_api_key)
+    else:
+        logger.warning("⚠️  OPENAI_API_KEY not set — RAG disabled")
 
-# Create the FastAPI app instance
+    logger.info("✅ LibBee startup complete")
+    yield
+    # SHUTDOWN — nothing needed for in-memory FAISS
+    logger.info("LibBee shutting down")
+
+
+# ── App ──────────────────────────────────────────────────────────────────────
 app = FastAPI(
     title="LibBee - KU Library AI",
     description="Khalifa University Library AI Assistant",
-    version="2.0.0"
+    version="2.0.0",
+    lifespan=lifespan,
 )
 
-# CORS Configuration
+# ── CORS ─────────────────────────────────────────────────────────────────────
 app.add_middleware(
     CORSMiddleware,
     allow_origins=[
-        "https://nikeshn.github.io",  # Your GitHub Pages
+        "https://nikeshn.github.io",
         "http://localhost:8080",
         "http://localhost:3000",
-        "http://127.0.0.1:5500"
+        "http://127.0.0.1:5500",
     ],
     allow_credentials=True,
-    allow_methods=["POST", "GET", "OPTIONS"],
+    allow_methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"],
     allow_headers=["*"],
     max_age=86400,
 )
 
-# Include routers
+# ── Routers ───────────────────────────────────────────────────────────────────
+from src.api import agent, search, admin  # noqa: E402 — after app created
+
 app.include_router(agent.router, prefix="/agent", tags=["Agent"])
 app.include_router(search.router, prefix="/search", tags=["Search"])
 app.include_router(admin.router, prefix="/admin", tags=["Admin"])
 
+
+# ── Health & Config endpoints ─────────────────────────────────────────────────
 @app.get("/")
 def health_check():
     return {
         "status": "ok",
         "version": "2.0.0",
         "service": "LibBee KU Library AI",
-        "endpoints": ["/agent", "/search", "/admin"]
+        "rag_ready": rag_service.is_ready(),
+        "endpoints": ["/agent", "/search", "/admin"],
     }
 
+
 @app.get("/config")
 def public_config():
     return {
         "welcome_message": "Hi! I'm LibBee, the Khalifa University Library AI Assistant.",
-        "default_model": "gpt",
-        "max_results": "5"
+        "max_results": 5,
     }
 
+
 @app.get("/year")
 def get_year():
     from datetime import datetime
     now = datetime.utcnow()
     return {"year": now.year, "month": now.month, "date": now.strftime("%Y-%m-%d")}
-
-print("✅ LibBee API loaded successfully")