plo

main.py

@@ -37,9 +37,11 @@ logger = logging.getLogger(__name__)
 try:
     from app.ai.routes.chat import router as ai_chat_router
     AI_AVAILABLE = True
-except ImportError:
-    logger.warning("⚠️ AI chat router not available - chat endpoint will be disabled")
+    logger.info("✅ AI chat router imported successfully")
+except ImportError as e:
+    logger.warning(f"⚠️ AI chat router not available: {e}")
     AI_AVAILABLE = False
+    ai_chat_router = None
 
 # ---------- Existing imports ----------
 try:
@@ -132,13 +134,14 @@ cors_origins = [
     "http://127.0.0.1:5000",
     "http://127.0.0.1:8080",
     "http://127.0.0.1:56205",
+    "https://destinyebuka-aida.hf.space",  # Add for HF Spaces deployment
 ]
 
 app.add_middleware(
     CORSMiddleware,
     allow_origins=cors_origins,
     allow_credentials=True,
-    allow_methods=["*"],
+    allow_methods=["GET", "POST", "PUT", "DELETE", "PATCH", "OPTIONS"],
     allow_headers=["*"],
     expose_headers=["*"],
     max_age=600,
@@ -194,11 +197,11 @@ async def general_exception_handler(request: Request, exc: Exception):
 app.include_router(auth.router, prefix="/api/auth", tags=["Authentication"])
 
 # Aida AI Chat routes (NEW)
-if AI_AVAILABLE:
-    app.include_router(ai_chat_router, prefix="/api/ai", tags=["Aida AI Chat"])
-    logger.info("✅ Aida AI chat router loaded at /api/ai")
+if AI_AVAILABLE and ai_chat_router:
+    app.include_router(ai_chat_router, prefix="/ai", tags=["Aida AI Chat"])
+    logger.info("✅ Aida AI chat router loaded at /ai")
 else:
-    logger.warning("⚠️ Aida AI chat router NOT loaded")
+    logger.warning("⚠️ Aida AI chat router NOT loaded - endpoint will not be available")
 
 # ====================================================================
 # Health & Status Endpoints
@@ -211,6 +214,7 @@ async def health_check():
         "service": "Lojiz Platform + Aida AI",
         "version": "1.0.0",
         "environment": environment,
+        "ai_available": AI_AVAILABLE,
     }
 
 @app.get("/", tags=["Root"])
@@ -221,6 +225,10 @@ async def root():
         "docs": "/docs",
         "health": "/health",
         "environment": environment,
+        "endpoints": {
+            "auth": "/api/auth",
+            "ai_chat": "/ai/ask" if AI_AVAILABLE else "NOT AVAILABLE",
+        }
     }
 
 @app.options("/{full_path:path}", include_in_schema=False)