Update main.py

main.py

@@ -1,99 +1,91 @@
-from fastapi import FastAPI, HTTPException, Depends, Request
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
-import decode_token
-import function.recommender
-import function.recommender.recommend
-from controller import CallbackController,ChatController,RedirectController,GroupOrderController
-from request import RequestChat
-from typing import Dict
-from fastapi.responses import RedirectResponse
-
-app = FastAPI(
-    title="ChatBot HMDrinks",
-    description="ChatBot HMDrinks for Customizing Chatbots by Nguyen Le Gia Han(21110432) & Nguyen Thuy Diem My(21110)",
-    version="1.0.0",
-    contact={
-        "name": "Nguyen Le Gia Han",
-        "url": "https://pychatbot1.streamlit.app",
-        "email": "21110432@student.hcmute.edu.vn",
-    },
-    license_info={
-        "name": "Apache 2.0",
-        "url": "https://www.apache.org/licenses/LICENSE-2.0.html",
-    },
-     swagger_ui_parameters={
-        "syntaxHighlight.theme": "monokai"
-    } 
-)
-
-
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["http://localhost:1010"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-class JWTBearer(HTTPBearer):
-    def __init__(self, auto_error: bool = True):
-        super(JWTBearer, self).__init__(auto_error=auto_error)
-
-    async def __call__(self, request: Request):
-        credentials: HTTPAuthorizationCredentials = await super(JWTBearer, self).__call__(request)
-        if credentials:
-            if credentials.scheme != "Bearer":
-                raise HTTPException(status_code=401, detail="Invalid authentication scheme.")
-            return credentials.credentials
-        else:
-            raise HTTPException(status_code=401, detail="Invalid authorization code.")
-
-jwt_bearer = JWTBearer()
-from fastapi.responses import JSONResponse
-
-@app.post("/api/v1/recommend", tags=["Recommend"])
-async def get_chat_details(request: RequestChat.RecommendRequest,token: str = Depends(jwt_bearer)):
-  try:
-    user_id_token = decode_token.JwtService.extract_user_id(token)
-    data =  await function.recommender.recommend.recommend(int(user_id_token),request.number)
-    return data 
-  except Exception as e:
-        print("Lỗi khi gọi recommend:", e)
-        return JSONResponse(content={"error": "Internal server error"}, status_code=500)
-
-
-@app.get("/grouporder/",tags=["GroupOrder"])
-async def group_order_redirect(request: Request):
-  try:
-    params: Dict[str, str] = dict(request.query_params)
-    code = params.get("code")
-    print(code)
-    if not code:
-        return RedirectResponse(url="myapp://open/group-order?status=-1")
-    
-    return RedirectResponse(url=f"myapp://open/group-order?status=1&code={code}")
-  except Exception as e:
-        print("Lỗi khi gọi group order:", e)
-        return JSONResponse(content={"error": "Internal server error"}, status_code=500)
-
-app.include_router(CallbackController.router, prefix="/api/v1/callback",tags=["CallBack"])
-app.include_router(ChatController.router, prefix="/api/v1/chat",tags=["ChatBot"])
-app.include_router(RedirectController.router, prefix="/api/v1/redirect",tags=["Redirect"])
-app.include_router(GroupOrderController.router, prefix="/api/v1/group-order",tags=["GroupOrder"])
-
-import nest_asyncio
-from uvicorn import run
-nest_asyncio.apply() 
-from pyngrok import ngrok
-NGROK_STATIC_DOMAIN = "rightly-poetic-amoeba.ngrok-free.app"
-NGROK_TOKEN = "2fEEmIGqKqV2EoE6n5lBRgObtov_4KmEfqwPJLdMi2G7GKnXU"
-
-# NGROK_STATIC_DOMAIN = "living-accurately-mackerel.ngrok-free.app"
-# NGROK_TOKEN = "2osXAUfmjMduYmyQUwjR2wSbUJ7_2nr9qR43nduNTWrwEYHmq"
-
-if __name__ == "__main__":
-    ngrok.set_auth_token(NGROK_TOKEN)
-    ngrok_tunnel = ngrok.connect(8085, domain=NGROK_STATIC_DOMAIN)
-    print('Public URL:', ngrok_tunnel.public_url)
+from fastapi import FastAPI, HTTPException, Depends, Request
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
+import decode_token
+import function.recommender
+import function.recommender.recommend
+from controller import CallbackController,ChatController,RedirectController,GroupOrderController
+from request import RequestChat
+from typing import Dict
+from fastapi.responses import RedirectResponse
+
+app = FastAPI(
+    title="ChatBot HMDrinks",
+    description="ChatBot HMDrinks for Customizing Chatbots by Nguyen Le Gia Han(21110432) & Nguyen Thuy Diem My(21110)",
+    version="1.0.0",
+    contact={
+        "name": "Nguyen Le Gia Han",
+        "url": "https://pychatbot1.streamlit.app",
+        "email": "21110432@student.hcmute.edu.vn",
+    },
+    license_info={
+        "name": "Apache 2.0",
+        "url": "https://www.apache.org/licenses/LICENSE-2.0.html",
+    },
+     swagger_ui_parameters={
+        "syntaxHighlight.theme": "monokai"
+    } 
+)
+
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["http://localhost:1010"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+class JWTBearer(HTTPBearer):
+    def __init__(self, auto_error: bool = True):
+        super(JWTBearer, self).__init__(auto_error=auto_error)
+
+    async def __call__(self, request: Request):
+        credentials: HTTPAuthorizationCredentials = await super(JWTBearer, self).__call__(request)
+        if credentials:
+            if credentials.scheme != "Bearer":
+                raise HTTPException(status_code=401, detail="Invalid authentication scheme.")
+            return credentials.credentials
+        else:
+            raise HTTPException(status_code=401, detail="Invalid authorization code.")
+
+jwt_bearer = JWTBearer()
+from fastapi.responses import JSONResponse
+
+@app.post("/api/v1/recommend", tags=["Recommend"])
+async def get_chat_details(request: RequestChat.RecommendRequest,token: str = Depends(jwt_bearer)):
+  try:
+    user_id_token = decode_token.JwtService.extract_user_id(token)
+    data =  await function.recommender.recommend.recommend(int(user_id_token),request.number)
+    return data 
+  except Exception as e:
+        print("Lỗi khi gọi recommend:", e)
+        return JSONResponse(content={"error": "Internal server error"}, status_code=500)
+
+
+@app.get("/grouporder/",tags=["GroupOrder"])
+async def group_order_redirect(request: Request):
+  try:
+    params: Dict[str, str] = dict(request.query_params)
+    code = params.get("code")
+    print(code)
+    if not code:
+        return RedirectResponse(url="myapp://open/group-order?status=-1")
+    
+    return RedirectResponse(url=f"myapp://open/group-order?status=1&code={code}")
+  except Exception as e:
+        print("Lỗi khi gọi group order:", e)
+        return JSONResponse(content={"error": "Internal server error"}, status_code=500)
+
+app.include_router(CallbackController.router, prefix="/api/v1/callback",tags=["CallBack"])
+app.include_router(ChatController.router, prefix="/api/v1/chat",tags=["ChatBot"])
+app.include_router(RedirectController.router, prefix="/api/v1/redirect",tags=["Redirect"])
+app.include_router(GroupOrderController.router, prefix="/api/v1/group-order",tags=["GroupOrder"])
+
+import nest_asyncio
+from uvicorn import run
+nest_asyncio.apply() 
+from pyngrok import ngrok
+
+if __name__ == "__main__":
     run(app, port=8085)