Update chatbot/main.py

chatbot/main.py

@@ -1,47 +1,31 @@
-from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
-from chatbot.routes.chat_router import router as chat_router
-from chatbot.routes.daily_plan_route import router as daily_plan_router
-from chatbot.routes.food_replace_route import router as food_replace_router
-from chatbot.routes.meal_replace_route import router as meal_replace_router
-from chatbot.models.embeddings import load_model, embeddings_model
-from contextlib import asynccontextmanager
-
-@asynccontextmanager
-async def lifespan(app: FastAPI):
-    # Load model đồng bộ khi startup (HF Spaces yêu cầu)
-    print("🚀 Starting model loading...")
-    load_model()
-    print("✅ Model ready!")
-    yield
-    print("🔻 Shutdown complete")
-
-app = FastAPI(
-    title="AI Meal Chatbot API",
-    lifespan=lifespan
-)
-
-# CORS
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# Routes
-app.include_router(chat_router)
-app.include_router(daily_plan_router)
-app.include_router(food_replace_router)
-app.include_router(meal_replace_router)
-
-@app.get("/")
-def root():
-    return {"message": "AI Meal Chatbot API is running 🚀"}
-
-@app.get("/model_status")
-def model_status():
-    if embeddings_model is None:
-        return {"status": "loading"}
-    return {"status": "ready"}
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from chatbot.routes.chat_router import router as chat_router
+from chatbot.routes.daily_plan_route import router as daily_plan_router
+from chatbot.routes.food_replace_route import router as food_replace_router
+from chatbot.routes.meal_replace_route import router as meal_replace_router
+
+app = FastAPI(
+    title="AI Meal Chatbot API",
+    description="API gợi ý món ăn, thực đơn, và tư vấn dinh dưỡng bằng AI",
+    version="1.0.0",
+)
+
+# Cho phép CORS để kết nối frontend
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # ⚠️ đổi sau nếu cần bảo mật
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Đăng ký route
+app.include_router(chat_router)
+app.include_router(daily_plan_router)
+app.include_router(food_replace_router)
+app.include_router(meal_replace_router)
+
+@app.get("/")
+def root():
+    return {"message": "AI Meal Chatbot API is running 🚀"}