Update app.py

app.py

@@ -1,27 +1,40 @@
 """
-LibBee - KU Library AI Agent
-New Modular Architecture
+LibBee - KU Library AI Agent v2.0
+Khalifa University Library AI Assistant
 """
 from fastapi import FastAPI, HTTPException, Request
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import HTMLResponse
 from contextlib import asynccontextmanager
 import os
 import sys
+import json
 
 # Add src to path
 sys.path.append(os.path.dirname(os.path.abspath(__file__)))
 
-from src.api import agent
+from src.api import agent, search, admin
+from src.services.cache_service import CacheService
+from src.config import get_settings
+
+# Global cache instance (HF Space is single-instance, so in-memory is fine)
+cache = CacheService()
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     """Startup and shutdown events"""
     print("🚀 LibBee starting up...")
-    # Load your vector store here later
+    app.state.cache = cache
+    app.state.start_time = __import__('time').time()
+    
+    # Load staff directory into cache
+    from src.services.staff_service import build_staff_index
+    build_staff_index()
+    
     yield
+    
     print("🛑 Shutting down")
 
-# Create app
 app = FastAPI(
     title="LibBee - KU Library AI",
     description="Khalifa University Library AI Assistant",
@@ -29,13 +42,14 @@ app = FastAPI(
     lifespan=lifespan
 )
 
-# SECURITY: CORS - Replace with your actual GitHub Pages URL
+# Security: Restrict CORS to your GitHub Pages
 app.add_middleware(
     CORSMiddleware,
     allow_origins=[
-        "https://nikeshn.github.io",  # Your GitHub Pages (change if different)
+        "https://nikeshn.github.io",  # Your GitHub Pages
         "http://localhost:8080",
-        "http://localhost:3000"
+        "http://localhost:3000",
+        "http://127.0.0.1:5500"
     ],
     allow_credentials=True,
     allow_methods=["POST", "GET", "OPTIONS"],
@@ -45,21 +59,31 @@ app.add_middleware(
 
 # Include routers
 app.include_router(agent.router, prefix="/agent", tags=["Agent"])
-# We'll add search and admin later
+app.include_router(search.router, prefix="/search", tags=["Search"])
+app.include_router(admin.router, prefix="/admin", tags=["Admin"])
 
 @app.get("/")
 def health_check():
+    import time
     return {
         "status": "ok",
         "version": "2.0.0",
-        "message": "LibBee API is running"
+        "service": "LibBee KU Library AI",
+        "endpoints": ["/agent", "/search", "/admin", "/config"]
     }
 
 @app.get("/config")
 def public_config():
     """Public configuration for frontend"""
     return {
-        "welcome_message": "Hi! I'm LibBee, the Khalifa University Library AI Assistant.",
+        "welcome_message": "Hi! I'm LibBee, the Khalifa University Library AI Assistant. I can help you find articles and books, search databases, access full text, and answer library questions.",
         "default_model": "gpt",
-        "max_results": "5"
+        "max_results": "5",
+        "maintenance_mode": "false"
     }
+
+@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")}