refactor code

app/ai/service.py

@@ -0,0 +1,102 @@
+from loguru import logger
+from ...core.conf import settings
+
+
+class ElectraModel:
+    def __init__(self, model_dir: str = settings.NER_MODEL_DIR):
+        self.model_dir = model_dir
+        self.model = None
+        self.tokenizer = None
+        self.pipeline = None
+        self.load_model()
+
+    def load_model(self):
+        from transformers import AutoTokenizer
+        from optimum.onnxruntime import ORTModelForTokenClassification
+        from optimum.pipelines import pipeline
+
+        self.tokenizer = AutoTokenizer.from_pretrained(
+            self.model_dir, truncation=settings.TRUNCATE, max_length=settings.MAX_LENGTH
+        )
+        self.model = ORTModelForTokenClassification.from_pretrained(self.model_dir)
+        self.pipeline = pipeline(
+            task=settings.TASK_NAME,
+            model=self.model,
+            tokenizer=self.tokenizer,
+            device=settings.DEVICE,
+        )
+        logger.info(f"Model loaded from {self.model_dir}")
+
+    async def predict(self, text: str, entity_tag: str = None):
+
+        if not text:
+            return None
+
+        if self.pipeline is None:
+            raise ValueError("Model not loaded. Please call load_model() first.")
+
+        pred = self.pipeline(text)
+
+        if entity_tag:
+            return self.extract_entities(pred, entity_tag)
+        return pred
+
+    def extract_entities(
+        self, result_pred: list[dict[str, any]], entity: str
+    ) -> list[str]:
+        if self.pipeline is None:
+            raise ValueError("Model not loaded. Please call load_model() first.")
+        B_ENTITY = f"B-{entity}"
+        I_ENTITY = f"I-{entity}"
+
+        extracted_entities = []
+        current_entity_tokens = []
+
+        for item in result_pred:
+            word = item["word"]
+            entity_tag = item["entity"]
+
+            if entity_tag == B_ENTITY:
+                if current_entity_tokens:
+                    extracted_entities.append(
+                        self._combine_token(current_entity_tokens)
+                    )
+                current_entity_tokens = [word]
+            elif entity_tag == I_ENTITY and current_entity_tokens:
+                current_entity_tokens.append(word)
+            else:
+                if current_entity_tokens:
+                    extracted_entities.append(
+                        self._combine_token(current_entity_tokens)
+                    )
+                    current_entity_tokens = []
+
+        if current_entity_tokens:
+            extracted_entities.append(self._combine_token(current_entity_tokens))
+
+        return extracted_entities
+
+    def _combine_token(self, tokens: list[str]) -> str:
+        """Combines tokens into a single string, removing leading hashtags from the first token if present.
+        Args:
+            tokens (list[str]): List of tokens to combine.
+
+        Returns:
+            str: Combined string of tokens.
+        """
+        if not tokens:
+            return ""
+
+        words = []
+
+        for token in tokens:
+            if token.strip("#") != token:
+                clean_token = token.strip("#")
+                if words:
+                    words[-1] += clean_token
+                else:
+                    words.append(clean_token)
+            else:
+                words.append(token)
+
+        return " ".join(words)

app/ai/weights/electra/config.json

@@ -0,0 +1,53 @@
+{
+  "_name_or_path": "NlpHUST/ner-vietnamese-electra-base",
+  "architectures": [
+    "ElectraForTokenClassification"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "classifier_dropout": null,
+  "embedding_size": 768,
+  "finetuning_task": "ner",
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "B-LOCATION",
+    "1": "B-MISCELLANEOUS",
+    "2": "B-ORGANIZATION",
+    "3": "B-PERSON",
+    "4": "I-LOCATION",
+    "5": "I-MISCELLANEOUS",
+    "6": "I-ORGANIZATION",
+    "7": "I-PERSON",
+    "8": "O"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "B-LOCATION": 0,
+    "B-MISCELLANEOUS": 1,
+    "B-ORGANIZATION": 2,
+    "B-PERSON": 3,
+    "I-LOCATION": 4,
+    "I-MISCELLANEOUS": 5,
+    "I-ORGANIZATION": 6,
+    "I-PERSON": 7,
+    "O": 8
+  },
+  "layer_norm_eps": 1e-12,
+  "max_position_embeddings": 512,
+  "model_type": "electra",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
+  "pad_token_id": 0,
+  "position_embedding_type": "absolute",
+  "summary_activation": "gelu",
+  "summary_last_dropout": 0.1,
+  "summary_type": "first",
+  "summary_use_proj": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.48.3",
+  "type_vocab_size": 2,
+  "use_cache": true,
+  "vocab_size": 62000
+}

app/ai/weights/electra/ort_config.json

@@ -0,0 +1,37 @@
+{
+  "one_external_file": true,
+  "opset": null,
+  "optimization": {
+    "disable_attention": null,
+    "disable_attention_fusion": false,
+    "disable_bias_gelu": null,
+    "disable_bias_gelu_fusion": false,
+    "disable_bias_skip_layer_norm": null,
+    "disable_bias_skip_layer_norm_fusion": false,
+    "disable_embed_layer_norm": true,
+    "disable_embed_layer_norm_fusion": true,
+    "disable_gelu": null,
+    "disable_gelu_fusion": false,
+    "disable_group_norm_fusion": true,
+    "disable_layer_norm": null,
+    "disable_layer_norm_fusion": false,
+    "disable_packed_kv": true,
+    "disable_rotary_embeddings": false,
+    "disable_shape_inference": false,
+    "disable_skip_layer_norm": null,
+    "disable_skip_layer_norm_fusion": false,
+    "enable_gelu_approximation": true,
+    "enable_gemm_fast_gelu_fusion": false,
+    "enable_transformers_specific_optimizations": true,
+    "fp16": false,
+    "no_attention_mask": false,
+    "optimization_level": 2,
+    "optimize_for_gpu": false,
+    "optimize_with_onnxruntime_only": null,
+    "use_mask_index": false,
+    "use_multi_head_attention": false,
+    "use_raw_attention_mask": false
+  },
+  "quantization": {},
+  "use_external_data_format": false
+}

app/ai/weights/electra/special_tokens_map.json

@@ -0,0 +1,37 @@
+{
+  "cls_token": {
+    "content": "[CLS]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "mask_token": {
+    "content": "[MASK]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "[PAD]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "sep_token": {
+    "content": "[SEP]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "[UNK]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

app/ai/weights/electra/tokenizer_config.json

@@ -0,0 +1,65 @@
+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "[PAD]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "[UNK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "[CLS]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "3": {
+      "content": "[SEP]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "4": {
+      "content": "[MASK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "clean_up_tokenization_spaces": false,
+  "cls_token": "[CLS]",
+  "do_basic_tokenize": true,
+  "do_lower_case": false,
+  "extra_special_tokens": {},
+  "mask_token": "[MASK]",
+  "max_length": 256,
+  "model_max_length": 1000000000000000019884624838656,
+  "never_split": null,
+  "pad_to_multiple_of": null,
+  "pad_token": "[PAD]",
+  "pad_token_type_id": 0,
+  "padding_side": "right",
+  "sep_token": "[SEP]",
+  "stride": 0,
+  "strip_accents": null,
+  "tokenize_chinese_chars": true,
+  "tokenizer_class": "ElectraTokenizer",
+  "truncation_side": "right",
+  "truncation_strategy": "longest_first",
+  "unk_token": "[UNK]"
+}

app/api/router.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+from fastapi import APIRouter
+
+# from app.api.v1.dialogflow import router as v1_dialogflow
+from app.api.v1 import trip, time, user, location
+from core.conf import settings
+
+# from utils.life_span import lifespan
+
+router = APIRouter(prefix=settings.FASTAPI_API_V1_PATH)
+
+router.include_router(trip.router)
+router.include_router(time.router)
+router.include_router(user.router)
+router.include_router(location.router)
+
+
+@router.get("/")
+def get_api_v1_root():
+    """
+    Root endpoint for API v1.
+    """
+    return {"message": "Welcome to the API v1", "version": settings.FASTAPI_VERSION}

app/api/v1/dialogflow.py

@@ -0,0 +1,1247 @@
+# from typing import Dict, List
+# from fastapi import FastAPI, APIRouter, HTTPException, Request, Response  # type: ignore
+# from fastapi.responses import JSONResponse, RedirectResponse, HTMLResponse  # type: ignore
+# from datetime import datetime, timedelta
+# from fastapi.templating import Jinja2Templates
+# from app.services.dialog_service import dialog_service
+# from app.ner.services.ner import NER
+# from utils.format_data_dialog import (
+#     extra_time_dialogflow,
+#     get_weekday_name,
+#     find_surrounding_times,
+# )
+
+# from common.external.external_api import api
+# from app.services.origin_codes import get_origin_id_and_code
+# from app.dto.response import DialogFlowResponseAPI
+
+# router = APIRouter()
+
+# templates = Jinja2Templates(directory="templates")
+
+
+# @router.post("/search/origin-city/from/office")  ##
+# async def search_origin_office(request: Request):
+#     body = await request.json()
+
+#     session_info = body.get("sessionInfo", {})
+#     parameters = session_info.get("parameters")
+
+#     raw_departure_city = parameters.get("departure_city", None)
+#     origin_office = parameters.get("origin_office")
+#     departure_city = None
+#     if origin_office and raw_departure_city is None:
+#         departure_city = await dialog_service.get_origin_city_from_office(origin_office)
+#     parameters = {"departure_city": departure_city}
+#     return DialogFlowResponseAPI(parameters=parameters)
+
+
+# @router.post("/search/destination-city/from/office")  ##
+# async def search_destination_office(request: Request):
+#     body = await request.json()
+#     session_info = body.get("sessionInfo", {})
+#     parameters = session_info.get("parameters")
+
+#     raw_destination_city = parameters.get("destination_city", None)
+#     dest_office = parameters.get("dest_office")
+
+#     destination_city = None
+#     if dest_office and raw_destination_city is None:
+#         destination_city = await dialog_service.get_destination_city_from_office(
+#             dest_office
+#         )
+
+#     parameters = {"destination_city": destination_city}
+#     return DialogFlowResponseAPI(parameters=parameters)
+
+
+# @router.post("/info/confirm")
+# async def confirm(request: Request):
+#     body = await request.json()
+#     session_info = body.get("sessionInfo", {})
+#     parameters = (
+#         session_info.get("parameters", {})
+#         if isinstance(session_info.get("parameters"), dict)
+#         else {}
+#     )
+#     raw_date = parameters.get("date")
+#     departure_city = parameters.get("departure_city")
+#     destination_city = parameters.get("destination_city")
+#     origin_office = parameters.get("origin_office")
+#     dest_office = parameters.get("dest_office")
+#     raw_ticket_number = parameters.get("ticket_number")
+#     time = parameters.get("time_select")
+
+#     time = extra_time_dialogflow(time)
+#     parameters = {}
+
+#     if isinstance(time, list):
+#         parameters["is_time_ambiguous"] = True
+#         return DialogFlowResponseAPI(parameters=parameters)
+
+#     ticket_number = int(raw_ticket_number) if raw_ticket_number else 1
+#     date = dialog_service.to_datetime_from_Dialogflow(raw_date)
+#     date, week_day = get_weekday_name(date)
+#     text = [
+#         f"""**Thời gian:** {time} - {date} -  {week_day}\n**Số vé:** {ticket_number}"""
+#     ]
+#     temp = ""
+#     if origin_office and dest_office:
+#         temp = f"""**Điểm đi:** {origin_office}\n**Điểm đến:** {dest_office}\n"""
+#     elif origin_office and destination_city:
+#         temp = f"""**Điểm đi:** {origin_office}\n**Điểm đến:** {destination_city}\n"""
+#     elif dest_office and departure_city:
+#         temp = f"""**Điểm đi:** {departure_city}\n**Điểm đến:** {dest_office}\n"""
+#     elif departure_city and destination_city:
+#         temp = f"""**Điểm đi:** {departure_city}\n**Điểm đến:** {destination_city}\n"""
+#     text[0] = temp + text[0]
+#     payload = {
+#         "richContent": [
+#             [
+#                 {
+#                     "type": "chips",
+#                     "options": [{"text": "Tìm chuyến xe"}, {"text": "Không, cảm ơn"}],
+#                 }
+#             ]
+#         ]
+#     }
+#     return DialogFlowResponseAPI(text=text, payload=payload)
+
+
+# @router.post("/routes")
+# async def route(request: Request):
+#     body = await request.json()
+#     session_info = body.get("sessionInfo", {})
+#     parameters = session_info.get("parameters")
+
+#     origin_office = parameters.get("origin_office")
+#     dest_office = parameters.get("dest_office")
+
+#     raw_departure_city, raw_destination_city, raw_ticket_number, raw_date, _ = (
+#         dialog_service.get_param_from_dialogflow(body)
+#     )
+
+#     ticket_count = int(raw_ticket_number) if raw_ticket_number else 1
+
+#     if raw_date is None:
+#         from_time, to_time = dialog_service.process_dates_to_timestamp()
+#         date = datetime.today().date().strftime("%m-%d-%Y")
+#     else:
+#         date = raw_date.strftime("%m-%d-%Y")
+#         from_time, to_time = dialog_service.process_dates_to_timestamp(raw_date)
+
+#     origin_code, origin_id, origin_ids, dest_code, dest_id, dest_ids = (
+#         None,
+#         None,
+#         None,
+#         None,
+#         None,
+#         None,
+#     )
+#     if origin_office:
+#         origin_id, origin_code = (
+#             await dialog_service.find_id_and_code_provine_by_name_office(origin_office)
+#         )
+#         origin_ids = await dialog_service.find_id_office_by_name_office(origin_office)
+#     elif raw_departure_city:
+#         origin_id, origin_code = get_origin_id_and_code(raw_departure_city)
+#     if dest_office:
+#         dest_id, dest_code = (
+#             await dialog_service.find_id_and_code_provine_by_name_office(dest_office)
+#         )
+#         dest_ids = await dialog_service.find_id_office_by_name_office(dest_office)
+#     elif raw_destination_city:
+#         dest_id, dest_code = get_origin_id_and_code(raw_destination_city)
+#     route_ids = await dialog_service.search_all_route_ids(
+#         origin_code=origin_code,
+#         from_id=origin_id,
+#         orign_ids=origin_ids,
+#         dest_code=dest_code,
+#         to_id=dest_id,
+#         dest_ids=dest_ids,
+#     )
+
+#     payload = {
+#         "from_time": from_time,
+#         "to_time": to_time,
+#         "route_ids": route_ids,
+#         "ticket_count": 1,
+#         "sort_by": ["price", "departure_time"],
+#     }
+#     try:
+#         respone = await api.post("/search/trips", payload=payload)
+#         data = respone["data"]["items"]
+#         total = respone["data"]["total"]
+#         if total > 0:
+#             price = data[0]["price"]
+#             list_raw_departure_times = sorted(
+#                 list(set([trip["raw_departure_time"] for trip in data]))
+#             )
+#             list_raw_seat_type = list(set([trip["seat_type_name"] for trip in data]))
+#             seat_type_trip_string = (
+#                 "**" + "** | **".join(map(str, list_raw_seat_type)) + "**"
+#             )
+#             schedule_time_trip = (
+#                 "**" + "** | **".join(map(str, list_raw_departure_times)) + "**"
+#             )
+#             duration = data[0]["duration"]
+#             link = f"https://stag.futabus.vn/dat-ve?from={departure_code}&fromTime={date}&isReturn=false&ticketCount={ticket_count}&to={destination_code}&toTime="
+#             text = [
+#                 f"Tuyến xe **{raw_departure_city}** - **{raw_destination_city}**\n \
+#                                      Loại xe: {seat_type_trip_string} \n \
+#                                      Thời gian hành trình: {duration} giờ \n \
+#                                      Giá vé: **{price}** VND"
+#             ]
+#             payload = {
+#                 "richContent": [
+#                     [
+#                         {
+#                             "type": "chips",
+#                             "options": [
+#                                 {"text": "Tìm chuyến xe"},
+#                                 {"text": "Xem lịch trình khác"},
+#                                 {"text": "Không, cảm ơn"},
+#                             ],
+#                         }
+#                     ]
+#                 ]
+#             }
+
+#             return DialogFlowResponseAPI(text=text, payload=payload)
+
+#         text = [
+#             f"Hệ thống không tìm thấy tuyến xe **{raw_departure_city}** - **{raw_destination_city}**.\n Quý khách vui lòng thử lại với lộ trình khác hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."
+#         ]
+#         payload = {
+#             "richContent": [
+#                 [
+#                     {
+#                         "type": "chips",
+#                         "options": [
+#                             {"text": "Xem lịch trình khác"},
+#                             {"text": "Không, cảm ơn"},
+#                         ],
+#                     }
+#                 ]
+#             ]
+#         }
+#         return DialogFlowResponseAPI(text=text, payload=payload)
+#     except Exception as e:
+#         return DialogFlowResponseAPI(
+#             text=[
+#                 "Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."
+#             ]
+#         )
+
+
+# @router.post("/price")
+# async def price(request: Request):
+#     body = await request.json()
+#     session_info = body.get("sessionInfo", {})
+#     parameters = session_info.get("parameters")
+
+#     raw_departure_city, raw_destination_city, _, raw_date, _ = (
+#         dialog_service.get_param_from_dialogflow(body)
+#     )
+
+#     if raw_date is None:
+#         from_time, to_time = dialog_service.process_dates_to_timestamp()
+#     from_time, to_time = dialog_service.process_dates_to_timestamp(raw_date)
+
+#     origin_office = parameters.get("origin_office")
+#     dest_office = parameters.get("dest_office")
+
+#     origin_code, origin_id, origin_ids, dest_code, dest_id, dest_ids = (
+#         None,
+#         None,
+#         None,
+#         None,
+#         None,
+#         None,
+#     )
+#     if origin_office:
+#         origin_id, origin_code = (
+#             await dialog_service.find_id_and_code_provine_by_name_office(origin_office)
+#         )
+#         origin_ids = await dialog_service.find_id_office_by_name_office(origin_office)
+#     elif raw_departure_city:
+#         origin_id, origin_code = get_origin_id_and_code(raw_departure_city)
+#     if dest_office:
+#         dest_id, dest_code = (
+#             await dialog_service.find_id_and_code_provine_by_name_office(dest_office)
+#         )
+#         dest_ids = await dialog_service.find_id_office_by_name_office(dest_office)
+#     elif raw_destination_city:
+#         dest_id, dest_code = get_origin_id_and_code(raw_destination_city)
+#     route_ids = await dialog_service.search_all_route_ids(
+#         origin_code=origin_code,
+#         from_id=origin_id,
+#         orign_ids=origin_ids,
+#         dest_code=dest_code,
+#         to_id=dest_id,
+#         dest_ids=dest_ids,
+#     )
+
+#     payload = {
+#         "from_time": from_time,
+#         "to_time": to_time,
+#         "route_ids": route_ids,
+#         "ticket_count": 1,
+#         "sort_by": ["price", "departure_time"],
+#     }
+#     try:
+#         respone = await api.post("/search/trips", payload=payload)
+#         total = respone["data"]["total"]
+#         if total > 0:
+#             price = respone["data"]["items"][0]["price"]
+#             text = [
+#                 f"Tuyến xe **{raw_departure_city}** - **{raw_destination_city}**\n"
+#                 f"Giá vé: **{price}** VND.\n \
+#                                      Bạn có muốn đặt vé không?"
+#             ]
+#             payload = {
+#                 "richContent": [
+#                     [
+#                         {
+#                             "type": "chips",
+#                             "options": [
+#                                 {"text": "Có, tôi muốn đặt vé"},
+#                                 {"text": "Xem giá vé tuyến xe khác"},
+#                                 {"text": "Không, cảm ơn"},
+#                             ],
+#                         }
+#                     ]
+#                 ]
+#             }
+#             return DialogFlowResponseAPI(text=text, payload=payload)
+#         text = [
+#             f"Hệ thống không tìm thấy tuyến xe **{raw_departure_city}** - **{raw_destination_city}**.\n Quý khách vui lòng thử lại với lộ trình khác hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."
+#         ]
+#         payload = {
+#             "richContent": [
+#                 [
+#                     {
+#                         "type": "chips",
+#                         "options": [
+#                             {"text": "Xem giá vé lịch trình khác"},
+#                             {"text": "Không, cảm ơn"},
+#                         ],
+#                     }
+#                 ]
+#             ]
+#         }
+
+#         return DialogFlowResponseAPI(text=text, payload=payload)
+#     except:
+#         return DialogFlowResponseAPI(
+#             text=[
+#                 "Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."
+#             ]
+#         )
+
+
+# @router.post("/trip/list")  ##
+# async def get_trip_list(request: Request) -> Response:
+#     body = await request.json()
+#     session_info = body.get("sessionInfo", {})
+#     parameters = session_info.get("parameters")
+
+#     raw_departure_city, raw_destination_city, raw_ticket_number, raw_date, _ = (
+#         dialog_service.get_param_from_dialogflow(body)
+#     )
+
+#     origin_office = parameters.get("origin_office")
+#     dest_office = parameters.get("dest_office")
+#     time = parameters.get("time_select")
+
+#     time = extra_time_dialogflow(time)
+
+#     if isinstance(time, list):
+#         parameters = {"is_time_ambiguous": True}
+#         return DialogFlowResponseAPI(parameters=parameters)
+
+#     from_time, to_time = dialog_service.process_dates_to_timestamp(raw_date)
+
+#     ticket_count = int(raw_ticket_number) if raw_ticket_number else 1
+
+#     origin_code, origin_id, origin_ids, dest_code, dest_id, dest_ids = (
+#         None,
+#         None,
+#         None,
+#         None,
+#         None,
+#         None,
+#     )
+#     if origin_office:
+#         origin_id, origin_code = (
+#             await dialog_service.find_id_and_code_provine_by_name_office(origin_office)
+#         )
+#         origin_ids = await dialog_service.find_id_office_by_name_office(origin_office)
+#     elif raw_departure_city:
+#         origin_id, origin_code = get_origin_id_and_code(raw_departure_city)
+#     if dest_office:
+#         dest_id, dest_code = (
+#             await dialog_service.find_id_and_code_provine_by_name_office(dest_office)
+#         )
+#         dest_ids = await dialog_service.find_id_office_by_name_office(dest_office)
+#     elif raw_destination_city:
+#         dest_id, dest_code = get_origin_id_and_code(raw_destination_city)
+#     route_ids = await dialog_service.search_all_route_ids(
+#         origin_code=origin_code,
+#         from_id=origin_id,
+#         orign_ids=origin_ids,
+#         dest_code=dest_code,
+#         to_id=dest_id,
+#         dest_ids=dest_ids,
+#     )
+
+#     try:
+#         data = await dialog_service.search_trip(
+#             from_time, to_time, route_ids, ticket_count
+#         )
+#         if len(data) > 0:
+
+#             if origin_office and dest_office:
+#                 trip_by_time_office = dialog_service.get_trip_by_time_and_office_id(
+#                     data, time, origin_ids, dest_ids
+#                 )
+#                 # Nếu có chuyến xe theo thời gian và văn phòng chỉ định
+#                 if trip_by_time_office:
+#                     parameters = {"is_valid_trip": True, "trip": trip_by_time_office}
+#                     return DialogFlowResponseAPI(parameters=parameters)
+
+#                 # Nếu không có chuyến xe theo thời gian và văn phòng chỉ định
+#                 # Danh sách chuyến xe khớp với văn phòng đón hoặc văn phòng trả
+#             #     data_by_office = dialog_service.get_all_trip_by_office(data, origin_ids, dest_ids)
+#             # elif origin_office:
+#             #     data_by_office = dialog_service.get_all_trip_by_office(data,origin_id=origin_ids)
+#             # elif dest_office:
+#             #     data_by_office = dialog_service.get_all_trip_by_office(data, dest_id=dest_ids)
+#             # else:
+
+#             data_by_office = data
+#             # Tìm 4 chuyến xe gần thời gian chỉ định nhất
+#             trip_surrounding_time = dialog_service.get_surrounding_trip(
+#                 data_by_office, time, num_trip=4
+#             )
+#             trip_dialogflow = []
+
+#             for trip in trip_surrounding_time:
+#                 if ticket_count <= trip["empty_seat_quantity"]:
+#                     trip_dialogflow.append(
+#                         {
+#                             "trip_id": trip["id"],
+#                             "route": f'{trip["raw_departure_time"]} | {trip["route"]["origin_hub_name"]} => {trip["route"]["dest_hub_name"]}',
+#                         }
+#                     )
+#             text = [
+#                 "Quý khách vui lòng lựa chọn chuyến xe\n"
+#                 + "\n".join(
+#                     f'{i+1}. {trip["route"]}' for i, trip in enumerate(trip_dialogflow)
+#                 )
+#             ]
+#             payload = {
+#                 "richContent": [
+#                     [
+#                         {
+#                             "type": "chips",
+#                             "options": [
+#                                 {"text": trip["route"]} for trip in (trip_dialogflow)
+#                             ],
+#                         }
+#                     ]
+#                 ]
+#             }
+#             parameters = {
+#                 "trip_select": trip_dialogflow,
+#                 "trips": trip_surrounding_time,
+#             }
+#             return DialogFlowResponseAPI(
+#                 text=text, payload=payload, parameters=parameters
+#             )
+
+#         text = [
+#             f"Hệ thống không tìm thấy tuyến xe **{raw_departure_city}** - **{raw_destination_city}**.\n Quý khách vui lòng thử lại với lộ trình khác hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."
+#         ]
+#         payload = {
+#             "richContent": [
+#                 [
+#                     {
+#                         "type": "chips",
+#                         "options": [
+#                             {"text": "Xem tuyến xe khác"},
+#                             {"text": "Không, cảm ơn"},
+#                         ],
+#                     }
+#                 ]
+#             ]
+#         }
+
+#         return DialogFlowResponseAPI(text=text, payload=payload)
+#     except Exception as e:
+#         print(e)
+#         return JSONResponse(
+#             {
+#                 "fulfillment_response": {
+#                     "messages": [
+#                         {
+#                             "text": {
+#                                 "text": [
+#                                     "Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."
+#                                 ]
+#                             }
+#                         }
+#                     ]
+#                 },
+#             }
+#         )
+
+
+# # Không xài
+# @router.post("/trip/route/list")
+# async def booking_trip(request: Request) -> Response:
+#     body = await request.json()
+#     session_info = body.get("sessionInfo", {})
+#     parameters = session_info.get("parameters")
+
+#     (
+#         raw_departure_city,
+#         raw_destination_city,
+#         raw_ticket_number,
+#         raw_date,
+#         raw_time_of_day,
+#     ) = dialog_service.get_param_from_dialogflow(body)
+
+#     origin_office = parameters.get("origin_office")
+#     dest_office = parameters.get("dest_office")
+
+#     from_time, to_time = dialog_service.process_dates_to_timestamp(raw_date)
+
+#     ticket_count = int(raw_ticket_number) if raw_ticket_number else 1
+
+#     origin_code, origin_id, origin_ids, dest_code, dest_id, dest_ids = (
+#         None,
+#         None,
+#         None,
+#         None,
+#         None,
+#         None,
+#     )
+#     if origin_office:
+#         origin_id, origin_code = (
+#             await dialog_service.find_id_and_code_provine_by_name_office(origin_office)
+#         )
+#         origin_ids = await dialog_service.find_id_office_by_name_office(origin_office)
+#     elif raw_departure_city:
+#         origin_id, origin_code = get_origin_id_and_code(raw_departure_city)
+#     if dest_office:
+#         dest_id, dest_code = (
+#             await dialog_service.find_id_and_code_provine_by_name_office(dest_office)
+#         )
+#         dest_ids = await dialog_service.find_id_office_by_name_office(dest_office)
+#     elif raw_destination_city:
+#         dest_id, dest_code = get_origin_id_and_code(raw_destination_city)
+#     route_ids = await dialog_service.search_all_route_ids(
+#         origin_code=origin_code,
+#         from_id=origin_id,
+#         orign_ids=origin_ids,
+#         dest_code=dest_code,
+#         to_id=dest_id,
+#         dest_ids=dest_ids,
+#     )
+#     try:
+#         data = await dialog_service.search_trip(
+#             from_time, to_time, route_ids, ticket_count
+#         )
+#         if len(data) > 0:
+#             trips = []
+#             routes_name = []
+#             for trip in data:
+#                 if ticket_count <= trip["empty_seat_quantity"]:
+#                     route = f"{trip['route']['origin_hub_name']} => {trip['route']['dest_hub_name']}"
+#                     if trip["route"] and route not in routes_name:
+#                         routes_name.append(
+#                             f"{trip['route']['origin_hub_name']} => {trip['route']['dest_hub_name']}"
+#                         )
+#                     trips.append(
+#                         {
+#                             "route_name": f"{trip['route']['origin_hub_name']} => {trip['route']['dest_hub_name']}",
+#                             "route_id": trip["route_id"],
+#                             "id": trip["id"],
+#                             "departure_date": trip["raw_departure_date"],
+#                             "departure_time": trip["raw_departure_time"],
+#                             "kind": trip["seat_type_name"],
+#                             "way_id": trip["way_id"],
+#                         }
+#                     )
+#             text = [
+#                 "Quý khách vui lòng lựa chọn chuyến xe\n"
+#                 + "\n".join(f"{i+1}. {name}" for i, name in enumerate(routes_name))
+#             ]
+#             payload = {
+#                 "richContent": [
+#                     [
+#                         {
+#                             "type": "chips",
+#                             "options": [{"text": name} for name in (routes_name)],
+#                         }
+#                     ]
+#                 ]
+#             }
+#             parameters = {"trip_list": trips, "routes_name": routes_name}
+#             return DialogFlowResponseAPI(
+#                 text=text, payload=payload, parameters=parameters
+#             )
+
+#         text = [
+#             f"Hệ thống không tìm thấy tuyến xe **{raw_departure_city}** - **{raw_destination_city}**.\n Quý khách vui lòng thử lại với lộ trình khác hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."
+#         ]
+#         payload = {
+#             "richContent": [
+#                 [
+#                     {
+#                         "type": "chips",
+#                         "options": [
+#                             {"text": "Xem tuyến xe khác"},
+#                             {"text": "Không, cảm ơn"},
+#                         ],
+#                     }
+#                 ]
+#             ]
+#         }
+
+#         return DialogFlowResponseAPI(text=text, payload=payload)
+#     except Exception as e:
+#         print(e)
+#         return JSONResponse(
+#             {
+#                 "fulfillment_response": {
+#                     "messages": [
+#                         {
+#                             "text": {
+#                                 "text": [
+#                                     "Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."
+#                                 ]
+#                             }
+#                         }
+#                     ]
+#                 },
+#             }
+#         )
+
+
+# @router.post("/trip/check-trip")
+# async def is_valid_select_trip(request: Request) -> Response:
+#     body = await request.json()
+#     raw_input = body.get("text", "")
+#     session_info = body.get("sessionInfo", {})
+#     parameters = session_info.get("parameters")
+#     trip_select: list[dict[str, any]] = parameters.get("trip_select")
+#     trips: list[dict[str, any]] = parameters.get("trips")
+
+#     if trip_select is None and trips is None:
+#         return DialogFlowResponseAPI()
+
+#     if raw_input:
+#         raw_input = raw_input.strip()
+#         for item in trip_select:
+#             if item["route"] == raw_input:
+#                 id = int(item["trip_id"])
+#                 trip = dialog_service.get_trip_by_id(id, trips)
+#                 parameters = {"is_valid_trip": True, "trip": trip}
+#                 return DialogFlowResponseAPI(parameters=parameters)
+
+#     parameters = {
+#         "is_valid_trip": False,
+#     }
+#     return DialogFlowResponseAPI(parameters=parameters)
+
+
+# @router.post("/trip/check-time-ambiguous")  #
+# async def check_time_ambiguous(request: Request) -> Response:
+#     try:
+#         body = await request.json()
+#         session_info = body.get("sessionInfo", {})
+#         parameters = session_info.get("parameters")
+#         time = parameters.get("time_select")
+
+#         parameters = {}
+
+#         time = extra_time_dialogflow(time)
+
+#         if isinstance(time, list) or time is None:
+#             parameters["is_time_ambiguous"] = True
+#         else:
+#             parameters["is_time_ambiguous"] = None
+
+#         return DialogFlowResponseAPI(parameters=parameters)
+
+#     except Exception as e:
+#         print(e)
+#         return DialogFlowResponseAPI(
+#             text=[
+#                 "Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."
+#             ]
+#         )
+
+
+# @router.post("/trip/select-time-ambiguous")  #
+# async def select_time(request: Request) -> Response:
+#     try:
+#         body = await request.json()
+#         session_info = body.get("sessionInfo", {})
+#         parameters = session_info.get("parameters")
+#         time = parameters.get("time_select")
+
+#         parameters = {}
+
+#         time = extra_time_dialogflow(time)
+
+#         text = []
+
+#         if time is None:
+#             text = ["Quý khách vui lòng chỉ định thời gian đi cụ thể trong ngày?"]
+
+#         elif isinstance(time, list):
+#             text = [f"Quý khách dự định đi vào lúc"]
+#             payload = {
+#                 "richContent": [
+#                     [{"type": "chips", "options": [{"text": item} for item in (time)]}]
+#                 ]
+#             }
+#         else:
+#             parameters["is_time_ambiguous"] = None
+
+#         return DialogFlowResponseAPI(text=text, payload=payload, parameters=parameters)
+
+#     except Exception as e:
+#         print(e)
+#         return DialogFlowResponseAPI(
+#             text=[
+#                 "Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."
+#             ]
+#         )
+
+
+# # Không xài
+# @router.post("/trip/time-trip-list")
+# async def time_trip(request: Request) -> Response:
+#     try:
+#         body = await request.json()
+#         session_info = body.get("sessionInfo", {})
+#         parameters = session_info.get("parameters")
+#         trip_list: list[Dict[str, any]] = parameters.get("trip_list", [])
+#         raw_route_id = parameters.get("route_id")
+#         route_name = parameters.get("route_name", "")
+#         raw_time = parameters.get("time_select")
+#         time_list = parameters.get("time_list", [])
+#         is_has_time = parameters.get("is_has_time", False)
+
+#         route_id = int(raw_route_id) if raw_route_id else None
+
+#         text = []
+#         quick_time_reply = []
+#         parameters = {}
+
+#         if raw_time:
+#             time = extra_time_dialogflow(raw_time)
+#             quick_time_reply = time_list
+
+#             if isinstance(time, list):
+#                 parameters["time_select"] = None
+#                 parameters["is_time_ambiguous"] = True
+#                 text = [f"Quý khách dự định đi vào lúc"]
+#                 payload = {
+#                     "richContent": [
+#                         [
+#                             {
+#                                 "type": "chips",
+#                                 "options": [{"text": item} for item in (time)],
+#                             }
+#                         ]
+#                     ]
+#                 }
+#                 return DialogFlowResponseAPI(
+#                     text=text, payload=payload, parameters=parameters
+#                 )
+#             else:
+#                 if is_has_time:
+#                     text = [
+#                         f"Quý khách lựa chọn thời gian chuyến {route_name}\n"
+#                         + " | ".join([item["time"] for item in quick_time_reply])
+#                     ]
+#                     parameters["is_has_time"] = None
+#                 else:
+#                     parameters["is_has_time"] = True
+#                 return DialogFlowResponseAPI(text=text, parameters=parameters)
+#         else:
+#             for trip in trip_list:
+#                 if (trip["route_id"]) == route_id:
+#                     time_list.append(
+#                         {"time": trip["departure_time"], "trip_id": trip["id"]}
+#                     )
+#             quick_time_reply = time_list
+#             text = [
+#                 f"Quý khách lựa chọn thời gian chuyến {route_name}\n"
+#                 + " | ".join([item["time"] for item in quick_time_reply])
+#             ]
+
+#         parameters["time_list"] = time_list
+#         payload = {
+#             "richContent": [
+#                 [
+#                     {
+#                         "type": "chips",
+#                         "options": [
+#                             {"text": time["time"]} for time in (quick_time_reply)
+#                         ],
+#                     }
+#                 ]
+#             ]
+#         }
+#         return DialogFlowResponseAPI(text=text, payload=payload, parameters=parameters)
+#     except Exception as e:
+#         print(e)
+#         return DialogFlowResponseAPI(
+#             text=[
+#                 "Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."
+#             ]
+#         )
+
+
+# # Không xài
+# @router.post("/trip/check-time-select")
+# async def is_valid_select_time(request: Request) -> Response:
+#     try:
+#         body = await request.json()
+#         session_info = body.get("sessionInfo", {})
+#         parameters = session_info.get("parameters")
+#         trip_list: list[Dict[str, any]] = parameters.get("trip_list", [])
+#         raw_route_id = parameters.get("route_id")
+
+#         (
+#             raw_departure_city,
+#             raw_destination_city,
+#             raw_ticket_number,
+#             raw_date,
+#             raw_time_of_day,
+#         ) = dialog_service.get_param_from_dialogflow(body)
+#         route_id = int(raw_route_id) if raw_route_id else None
+
+#         from_time, to_time = dialog_service.process_dates_to_timestamp(raw_date)
+#         ticket_count = int(raw_ticket_number) if raw_ticket_number else 1
+
+#         origin_office = parameters.get("origin_office")
+#         dest_office = parameters.get("dest_office")
+
+#         origin_code, origin_id, origin_ids, dest_code, dest_id, dest_ids = (
+#             None,
+#             None,
+#             None,
+#             None,
+#             None,
+#             None,
+#         )
+#         if origin_office:
+#             origin_id, origin_code = (
+#                 await dialog_service.find_id_and_code_provine_by_name_office(
+#                     origin_office
+#                 )
+#             )
+#             origin_ids = await dialog_service.find_id_office_by_name_office(
+#                 origin_office
+#             )
+#         elif raw_departure_city:
+#             origin_id, origin_code = get_origin_id_and_code(raw_departure_city)
+#         if dest_office:
+#             dest_id, dest_code = (
+#                 await dialog_service.find_id_and_code_provine_by_name_office(
+#                     dest_office
+#                 )
+#             )
+#             dest_ids = await dialog_service.find_id_office_by_name_office(dest_office)
+#         elif raw_destination_city:
+#             dest_id, dest_code = get_origin_id_and_code(raw_destination_city)
+#         route_ids = await dialog_service.search_all_route_ids(
+#             origin_code=origin_code,
+#             from_id=origin_id,
+#             orign_ids=origin_ids,
+#             dest_code=dest_code,
+#             to_id=dest_id,
+#             dest_ids=dest_ids,
+#         )
+
+#         time_list: list[Dict[str, any]] = parameters.get("time_list", [])
+#         time_select = parameters.get("time_select")
+#         route_name = parameters.get("route_name")
+#         is_has_time = parameters.get("is_has_time")
+
+#         if len(time_list) == 0:
+#             for trip in trip_list:
+#                 if (trip["route_id"]) == route_id:
+#                     time_list.append(
+#                         {"time": trip["departure_time"], "trip_id": trip["id"]}
+#                     )
+
+#         if time_select:
+#             time_select = extra_time_dialogflow(time_select)
+#             for time in time_list:
+#                 if time_select == time["time"]:
+#                     id = int(time["trip_id"])
+#                     trip = await dialog_service.search_trip_by_id(
+#                         id, from_time, to_time, route_ids, ticket_count
+#                     )
+#                     departure_date = trip["raw_departure_date"]
+#                     parameters = {
+#                         "is_valid_time": True,
+#                         "departure_time": time_select,
+#                         "departure_date": departure_date,
+#                         "trip": trip,
+#                     }
+#                     text = [
+#                         f"Quý khách chọn chuyến **{time_select}** | **{route_name}**"
+#                     ]
+#                     return DialogFlowResponseAPI(text=text, parameters=parameters)
+#             if is_has_time:
+#                 time_list = find_surrounding_times(time_list, time_select)
+#                 parameters = {"time_list": time_list, "is_valid_time": False}
+
+#                 return DialogFlowResponseAPI(parameters=parameters)
+
+#         parameters = {"time_select": None, "is_valid_time": False}
+#         text = []
+#         return DialogFlowResponseAPI(text=text, parameters=parameters)
+#     except Exception as e:
+#         print(e)
+#         return DialogFlowResponseAPI(
+#             text=[
+#                 "Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."
+#             ]
+#         )
+
+
+# @router.post("/trip/seats")  #
+# async def seats_trip(request: Request) -> Response:
+#     try:
+#         body = await request.json()
+#         session_info = body.get("sessionInfo", {})
+#         parameters = session_info.get("parameters")
+
+#         trip = parameters.get("trip", None)
+#         route_id = int(trip.get("route_id")) if trip.get("route_id") else None
+#         trip_id = int(trip.get("id")) if trip.get("id") else None
+
+#         departure_date: str = trip.get("raw_departure_date")
+#         departure_time: str = trip.get("raw_departure_time")
+#         kind: str = trip.get("seat_type_name")
+
+#         seats = await dialog_service.seats_trip(
+#             route_id, trip_id, departure_date, departure_time, kind
+#         )
+#         seats_empty = [seat for seat in seats if seat["bookStatus"] == 0]
+#         seats_empty.sort(key=lambda x: x["chair"])
+#         text = ["Quý khách vui lòng chọn ghế"]
+#         payload = {}
+#         if seats_empty:
+#             payload = {
+#                 "richContent": [
+#                     [
+#                         {
+#                             "type": "chips",
+#                             "options": [
+#                                 {"text": seat["chair"]} for seat in (seats_empty)
+#                             ],
+#                         }
+#                     ]
+#                 ]
+#             }
+#         else:
+#             text = ["Không còn ghế trống. Quý khách vui lòng lựa chọn chuyến khác"]
+#         parameters = {"seat_list": seats}
+#         return DialogFlowResponseAPI(text=text, payload=payload, parameters=parameters)
+#     except Exception as e:
+#         print(e)
+#         return DialogFlowResponseAPI(
+#             text=[
+#                 "Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."
+#             ]
+#         )
+
+
+# @router.post("/trip/check-seat-select")  #
+# async def is_valid_select_seat(request: Request) -> Response:
+#     try:
+#         body = await request.json()
+#         session_info = body.get("sessionInfo", {})
+#         parameters = session_info.get("parameters")
+
+#         trip: list[dict[str, any]] = parameters.get("trip", None)
+
+#         route_id = int(trip.get("route_id")) if trip.get("route_id") else None
+
+#         trip_id = int(trip.get("id")) if trip.get("id") else None
+
+#         departure_date: str = trip.get("raw_departure_date")
+
+#         departure_time: str = trip.get("raw_departure_time")
+
+#         kind: str = trip.get("seat_type_name")
+
+#         seat: str = parameters.get("seat")
+
+#         is_valid = await dialog_service.is_valid_select_seat(
+#             seat, route_id, trip_id, departure_date, departure_time, kind
+#         )
+#         if is_valid:
+#             parameters = {"is_valid_seat": True, "seat": seat}
+#             text = [f"Quý khách chọn ghế **{seat.upper()}**"]
+#         else:
+#             parameters = {"is_valid_seat": False, "seat": None}
+#             text = [
+#                 f"Ghế **{seat.upper()}** không hợp lệ. Quý khách vui lòng chọn ghế khác"
+#             ]
+#         return DialogFlowResponseAPI(text=text, parameters=parameters)
+
+#     except Exception as e:
+#         return DialogFlowResponseAPI(
+#             text=[
+#                 "Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."
+#             ]
+#         )
+
+
+# @router.post("/trip/check-exist-user-info")  #
+# async def check_exist_user_info(request: Request) -> Response:
+#     body = await request.json()
+#     session_info = body.get("sessionInfo", {})
+#     parameters = session_info.get("parameters")
+
+#     is_exist_user_info = await dialog_service.check_exist_user_info()
+
+#     user_info = {}
+
+#     if is_exist_user_info:
+#         user_info = await dialog_service.get_user_info()
+
+#     user_name = user_info.get("user_name")
+#     phone_number = user_info.get("phone_number")
+#     email = user_info.get("email")
+
+#     parameters = {
+#         "is_user_exist": is_exist_user_info,
+#         "user_name": user_name,
+#         "phone_number": phone_number,
+#         "email": email,
+#     }
+
+#     return DialogFlowResponseAPI(parameters=parameters)
+
+
+# @router.post("/trip/extract-user-name")  #
+# async def get_user_name(request: Request) -> Response:
+#     body = await request.json()
+#     session_info = body.get("sessionInfo", {})
+#     parameters = session_info.get("parameters")
+
+#     raw_text_user_name = body.get("text", "")
+
+#     ner: NER = request.app.state.ner
+
+#     user_name = await dialog_service.extract_user_name(text=raw_text_user_name, ner=ner)
+
+#     parameters = {"user_name": user_name}
+
+#     return DialogFlowResponseAPI(parameters=parameters)
+
+
+# @router.post("/trip/stop/pickup")  #
+# async def pickup(request: Request) -> Response:
+#     body = await request.json()
+#     session_info = body.get("sessionInfo", {})
+#     parameters = session_info.get("parameters")
+
+#     trip: list[dict[str, any]] = parameters.get("trip", {})
+
+#     route_id = int(trip.get("route_id")) if trip.get("route_id") else None
+
+#     way_id = int(trip.get("way_id")) if trip.get("way_id") else None
+
+#     pickup_list = await dialog_service.pickup_list(route_id, way_id)
+
+#     text = ["Quý khách vui lòng chọn điểm đón"]
+#     payload = {
+#         "richContent": [
+#             [
+#                 {
+#                     "type": "chips",
+#                     "options": [{"text": pickup["name"]} for pickup in (pickup_list)],
+#                 }
+#             ]
+#         ]
+#     }
+#     parameters = {"pickup_list": pickup_list}
+
+#     return DialogFlowResponseAPI(text=text, payload=payload, parameters=parameters)
+
+
+# @router.post("/trip/check-pickup-select")
+# async def is_valid_select_pickup(request: Request) -> Response:
+#     body = await request.json()
+#     session_info = body.get("sessionInfo", {})
+#     parameters = session_info.get("parameters")
+
+#     trip: list[dict[str, any]] = parameters.get("trip", {})
+
+#     route_id = int(trip.get("route_id")) if trip.get("route_id") else None
+
+#     way_id = int(trip.get("way_id")) if trip.get("way_id") else None
+
+#     raw_input = body.get("text", "")
+#     pickup = raw_input.strip()
+
+#     is_valid = await dialog_service.is_valid_pickup(pickup, route_id, way_id)
+
+#     if is_valid:
+#         parameters = {"is_valid_pickup": True, "pick_up": pickup}
+#         text = [f"Quý khách chọn điểm đón **{pickup}**"]
+#     else:
+#         parameters = {
+#             "is_valid_pickup": False,
+#         }
+#         # text = [f"Điểm đón không hợp lệ. Quý khách vui lòng chọn điểm đón khác"]
+#         text = []
+
+#     return DialogFlowResponseAPI(text=text, parameters=parameters)
+
+
+# @router.post("/trip/stop/dropoff")
+# async def dropoff(request: Request) -> Response:
+#     body = await request.json()
+#     session_info = body.get("sessionInfo", {})
+#     parameters = session_info.get("parameters")
+
+#     trip: list[dict[str, any]] = parameters.get("trip", {})
+
+#     route_id = int(trip.get("route_id")) if trip.get("route_id") else None
+
+#     way_id = int(trip.get("way_id")) if trip.get("way_id") else None
+
+#     dropoff_list = await dialog_service.dropoff_list(route_id, way_id)
+
+#     text = ["Quý khách vui lòng chọn điểm trả khách"]
+#     payload = {
+#         "richContent": [
+#             [
+#                 {
+#                     "type": "chips",
+#                     "options": [
+#                         {"text": dropoff["name"]} for dropoff in (dropoff_list)
+#                     ],
+#                 }
+#             ]
+#         ]
+#     }
+#     parameters = {"dropoff_list": dropoff_list}
+
+#     return DialogFlowResponseAPI(text=text, payload=payload, parameters=parameters)
+
+
+# @router.post("/trip/check-dropoff-select")
+# async def is_valid_select_dropoff(request: Request) -> Response:
+#     body = await request.json()
+#     session_info = body.get("sessionInfo", {})
+#     parameters = session_info.get("parameters")
+
+#     trip: list[dict[str, any]] = parameters.get("trip", {})
+
+#     route_id = int(trip.get("route_id")) if trip.get("route_id") else None
+
+#     way_id = int(trip.get("way_id")) if trip.get("way_id") else None
+
+#     raw_input = body.get("text", "")
+#     dropoff = raw_input.strip()
+
+#     is_valid = await dialog_service.is_valid_dropoff(dropoff, route_id, way_id)
+
+#     if is_valid:
+#         parameters = {"is_valid_dropoff": True, "drop_off": dropoff}
+#         text = [f"Quý khách chọn điểm trả khách **{dropoff}**"]
+#     else:
+#         parameters = {
+#             "is_valid_dropoff": False,
+#         }
+#         # text = [f"Điểm trả khách không hợp lệ. Quý khách vui lòng chọn điểm trả khách khác"]
+#         text = []
+
+#     return DialogFlowResponseAPI(text=text, parameters=parameters)
+
+
+# @router.post("/ticket/info")
+# async def response_ticket_info(request: Request) -> Response:
+#     body = await request.json()
+#     session_info = body.get("sessionInfo", {})
+#     parameters = session_info.get("parameters")
+
+#     raw_user_name = parameters.get("user_name")
+#     user_name = raw_user_name["name"] if raw_user_name else None
+
+#     phone_number = parameters.get("phone_number")
+
+#     email = parameters.get("email")
+
+#     seat: str = parameters.get("seat")
+
+#     pickup = parameters.get("pick_up")
+
+#     dropoff = parameters.get("drop_off")
+
+#     trip = parameters.get("trip", {})
+
+#     route_name = trip["route"]["name"]
+
+#     time = trip["raw_departure_time"]
+
+#     date = trip["raw_departure_date"]
+
+#     price = int(trip.get("price")) if trip.get("price") else None
+
+#     text = [
+#         f" \
+#         **Thông tin hành khách**\n\
+#         **Họ và tên** {user_name} \n\
+#         **Số điện thoại** {phone_number}\n\
+#         **Email** {email} \n\
+#         **Thông tin lượt đi**\n\
+#         **Tuyến xe** {route_name} \n\
+#         **Thời gian xuất bến** {time} {date} \n\
+#         **Số ghế** {seat.upper()} \n\
+#         **Điểm lên xe** {pickup} \n\
+#         **Điểm trả khách** {dropoff} \n\
+#         **Tổng tiền lượt đi** {price} VND \
+#         "
+#     ]
+
+#     payload = {
+#         "richContent": [
+#             [
+#                 {
+#                     "type": "chips",
+#                     "options": [{"text": "Đặt vé"}, {"text": "Không, cảm ơn"}],
+#                 }
+#             ]
+#         ]
+#     }
+
+#     return DialogFlowResponseAPI(text=text, payload=payload)
+
+
+# @router.get("/")
+# def home():
+#     return "Hello World!"
+
+
+# @router.get("/chatbot", response_class=HTMLResponse)
+# async def index(request: Request):
+#     return templates.TemplateResponse(
+#         "index.html", {"request": request, "title": "Dawi Chatbot"}
+#     )

app/api/v1/location.py

@@ -0,0 +1,58 @@
+from loguru import logger
+from fastapi import APIRouter, Depends, HTTPException, Request
+from dependency_injector.wiring import inject, Provide
+
+
+from app.dependencies.containers import Container
+from app.dto.request import LocationRequest
+from app.dto.response import DialogFlowResponseAPI
+from app.services.location_service import LocationService
+
+
+router = APIRouter(prefix="/location", tags=["Location"])
+
+
+@router.post("/origin-city-from-office")
+@inject
+async def get_origin_city_from_office(
+    request: Request,
+    service: LocationService = Depends(Provide[Container.location_service]),
+):
+    body = await request.json()
+    params = body.get("sessionInfo", {}).get("parameters", {})
+    try:
+        data = LocationRequest(**params)
+    except Exception:
+        logger.error("Invalid location parameters", exc_info=True)
+        raise HTTPException(status_code=400, detail="Invalid location parameters")
+
+    if data.origin_office and not data.departure_city:
+        _, _, city = await service.get_city_by_office(
+            office_name=data.origin_office, role="origin"
+        )
+        return DialogFlowResponseAPI(parameters={"departure_city": city})
+
+    return DialogFlowResponseAPI(parameters={"departure_city": data.departure_city})
+
+
+@router.post("/dest-city-from-office")
+@inject
+async def get_dest_city_from_office(
+    request: Request,
+    service: LocationService = Depends(Provide[Container.location_service]),
+):
+    body = await request.json()
+    params = body.get("sessionInfo", {}).get("parameters", {})
+    try:
+        data = LocationRequest.model_validate(params)
+    except Exception:
+        logger.error("Invalid location parameters", exc_info=True)
+        raise HTTPException(status_code=400, detail="Invalid location parameters")
+
+    if data.dest_office and not data.destination_city:
+        city = await service.get_city_by_office(
+            office_name=data.dest_office, role="dest"
+        )
+        return DialogFlowResponseAPI(parameters={"destination_city": city})
+
+    return DialogFlowResponseAPI(parameters={"destination_city": data.destination_city})

app/api/v1/time.py

@@ -0,0 +1,63 @@
+from loguru import logger
+
+from fastapi import APIRouter, Depends, HTTPException, Request
+from dependency_injector.wiring import inject, Provide
+
+
+from app.dependencies.containers import Container
+from app.dto.request import DialogflowWebhookRequest, TimeRequest
+from app.dto.response import (
+    DialogServiceResult,
+    DialogflowResponseBuilder,
+)
+from app.services.time_service import TimeService
+
+
+router = APIRouter(prefix="/time", tags=["Time"])
+
+
+@router.post("/check-ambiguity")
+@inject
+async def check_time_ambiguity(
+    request: Request, service: TimeService = Depends(Provide[Container.time_service])
+):
+    body = await request.json()
+    body = DialogflowWebhookRequest(**body)
+    params = body.get_parameters()
+    time = TimeRequest(**params)
+    if not time:
+        logger.error("Time selection is empty")
+        raise HTTPException(status_code=400, detail="Time selection is required")
+
+    is_ambiguous = service.check_time_ambiguity(time=time.time_select)
+
+    return (
+        DialogflowResponseBuilder()
+        .set_parameters({"is_time_ambiguous": is_ambiguous})
+        .build()
+    )
+
+
+@router.post("/ambiguity-selection")
+@inject
+async def time_ambiguity_selection(
+    request: Request, service: TimeService = Depends(Provide[Container.time_service])
+):
+    body = await request.json()
+    body = DialogflowWebhookRequest(**body)
+    params = body.get_parameters()
+
+    time = params.get("time_select")
+    if not time:
+        logger.error("Time selection is required")
+        raise HTTPException(status_code=400, detail="Time selection is required")
+
+    rs: DialogServiceResult = service.get_time_ambiguous_selection(time)
+
+    return (
+        DialogflowResponseBuilder()
+        .add_text(rs.text)
+        .add_choice_chips(options=rs.options)
+        .set_parameters(rs.parameters)
+        .build()
+    )

app/api/v1/trip.py

@@ -0,0 +1,277 @@
+from typing import Annotated
+from loguru import logger
+from fastapi import APIRouter, Depends, HTTPException, Request
+from dependency_injector.wiring import inject, Provide
+
+from app.dependencies.containers import Container
+from app.dto.request import (
+    DialogflowWebhookRequest,
+    SeatQueryRequest,
+    StopRequest,
+    TicketRequest,
+    TripQueryRequest,
+)
+from app.dto.response import (
+    DialogServiceResult,
+    DialogflowResponseBuilder,
+)
+from app.services.trip_service import TripService
+
+
+router = APIRouter(prefix="/trip", tags=["Trip"])
+
+
+@router.post("/option")
+@inject
+async def trips_option(
+    request: Request, service: TripService = Depends(Provide[Container.trip_service])
+):
+    body = await request.json()
+    params = body.get("sessionInfo", {}).get("parameters", {})
+    try:
+        data = TripQueryRequest(**params)
+    except Exception as e:
+        raise HTTPException(status_code=400, detail="Invalid location parameters")
+
+    rs: DialogServiceResult = await service.get_trips_option(data)
+
+    return (
+        DialogflowResponseBuilder()
+        .add_text(rs.text)
+        .add_choice_chips(options=rs.options, key="route")
+        .set_parameters(rs.parameters)
+        .build()
+    )
+
+
+@router.post("/confirm-info")
+@inject
+async def confirm(
+    request: Request, service: TripService = Depends(Provide[Container.trip_service])
+):
+    body = await request.json()
+    body = DialogflowWebhookRequest(**body)
+    parameters = body.get_parameters()
+    try:
+        data = TripQueryRequest(**parameters)
+    except Exception as e:
+        logger.error(f"Error parsing trip parameters {e}", exc_info=True)
+        raise HTTPException(status_code=400, detail="Invalid trip query parameters")
+
+    rs: DialogServiceResult = service.confirm_trip_search_input(data=data)
+
+    return (
+        DialogflowResponseBuilder()
+        .add_text(rs.text)
+        .add_choice_chips(options=rs.options)
+        .build()
+    )
+
+
+@router.post("/validate-trip-selection")
+@inject
+async def validate_trip_selection(
+    request: Request, service: TripService = Depends(Provide[Container.trip_service])
+):
+    body = await request.json()
+    body = DialogflowWebhookRequest(**body)
+    params = body.get_parameters()
+    try:
+        data = TripQueryRequest(**params)
+    except Exception as e:
+        raise HTTPException(status_code=400, detail="Invalid trip selection parameters")
+
+    rs: DialogServiceResult = await service.validate_trip_selection(
+        id_selected=data.id_selected, data=data
+    )
+    return DialogflowResponseBuilder().set_parameters(rs.parameters).build()
+
+
+@router.post("/available-seat")
+@inject
+async def fetch_available_seat(
+    request: Request, service: TripService = Depends(Provide[Container.trip_service])
+):
+    body = await request.json()
+    body = DialogflowWebhookRequest(**body)
+    params = body.get_parameters()
+
+    try:
+        logger.info(f"Received parameters: {params}")
+        data = SeatQueryRequest(**params)
+
+    except Exception as e:
+        logger.error(f"Error parsing trip parameters {e}", exc_info=True)
+        raise HTTPException(status_code=400, detail="Invalid trip query parameters")
+
+    rs: DialogServiceResult = await service.get_available_seat(data.trip_data)
+    return (
+        DialogflowResponseBuilder()
+        .add_text(rs.text)
+        .add_choice_chips(options=rs.options, key="chair")
+        .set_parameters(rs.parameters)
+        .build()
+    )
+
+
+@router.post("/validate-seat-selection")
+@inject
+async def validate_seat_selection(
+    request: Request, service: TripService = Depends(Provide[Container.trip_service])
+):
+    body = await request.json()
+    body = DialogflowWebhookRequest(**body)
+    params = body.get_parameters()
+
+    try:
+        data = SeatQueryRequest(**params)
+    except Exception as e:
+        logger.error("Error parsing trip parameters", exc_info=True)
+        raise HTTPException(status_code=400, detail="Invalid trip query parameters")
+
+    rs: DialogServiceResult = await service.validate_seat_selection(
+        selected=data.seat, data=data.trip_data
+    )
+    return (
+        DialogflowResponseBuilder()
+        .add_text(rs.text)
+        .set_parameters(rs.parameters)
+        .build()
+    )
+
+
+@router.post("/pick-up")
+@inject
+async def pick_up_stop(
+    request: Request, service: TripService = Depends(Provide[Container.trip_service])
+):
+    try:
+        body = await request.json()
+        body = DialogflowWebhookRequest(**body)
+        params = body.get_parameters()
+        data = StopRequest(**params)
+    except Exception as e:
+        logger.error(f"Error parsing trip parameters {e}", exc_info=True)
+        raise HTTPException(status_code=400, detail="Invalid trip query parameters")
+
+    rs: DialogServiceResult = await service.get_pick_up_points(
+        route_id=data.trip_data.route_id, way_id=data.trip_data.way_id
+    )
+    return (
+        DialogflowResponseBuilder()
+        .add_text(rs.text)
+        .add_choice_chips(options=rs.options, key="name")
+        .set_parameters(rs.parameters)
+        .build()
+    )
+
+
+@router.post("/pick-up/validate-selection")
+@inject
+async def validate_pick_up_selection(
+    request: Request, service: TripService = Depends(Provide[Container.trip_service])
+):
+    body = await request.json()
+    body = DialogflowWebhookRequest(**body)
+    params = body.get_parameters()
+
+    try:
+        data = StopRequest(**params)
+    except Exception as e:
+        logger.error("Error parsing trip parameters", exc_info=True)
+        raise HTTPException(status_code=400, detail="Invalid trip query parameters")
+
+    selected = body.get_text()
+    rs: DialogServiceResult = await service.validate_stop_selection(
+        id_selected=data.id_selected, data=data.trip_data, role="pick-up"
+    )
+
+    return (
+        DialogflowResponseBuilder()
+        .add_text(rs.text)
+        .add_choice_chips(options=rs.options)
+        .set_parameters(rs.parameters)
+        .build()
+    )
+
+
+@router.post("/drop-off")
+@inject
+async def drop_off_stop(
+    request: Request, service: TripService = Depends(Provide[Container.trip_service])
+):
+    body = await request.json()
+    body = DialogflowWebhookRequest(**body)
+    params = body.get_parameters()
+
+    try:
+        data = StopRequest(**params)
+    except Exception as e:
+        logger.error("Error parsing trip parameters", exc_info=True)
+        raise HTTPException(status_code=400, detail="Invalid trip query parameters")
+
+    rs: DialogServiceResult = await service.get_drop_off_points(
+        route_id=data.trip_data.route_id, way_id=data.trip_data.way_id
+    )
+    logger.info(f"Drop-off points: {rs.options}")
+
+    return (
+        DialogflowResponseBuilder()
+        .add_text(rs.text)
+        .add_choice_chips(options=rs.options, key="name")
+        .set_parameters(rs.parameters)
+        .build()
+    )
+
+
+@router.post("/drop-off/validate-selection")
+@inject
+async def validate_drop_off_selection(
+    request: Request, service: TripService = Depends(Provide[Container.trip_service])
+):
+    body = await request.json()
+    body = DialogflowWebhookRequest(**body)
+    params = body.get_parameters()
+
+    try:
+        data = StopRequest(**params)
+    except Exception as e:
+        logger.error("Error parsing trip parameters", exc_info=True)
+        raise HTTPException(status_code=400, detail="Invalid trip query parameters")
+
+    rs: DialogServiceResult = await service.validate_stop_selection(
+        id_selected=data.id_selected, data=data.trip_data, role="drop-off"
+    )
+
+    return (
+        DialogflowResponseBuilder()
+        .add_text(rs.text)
+        .add_choice_chips(options=rs.options)
+        .set_parameters(rs.parameters)
+        .build()
+    )
+
+
+@router.post("/ticket")
+@inject
+async def show_ticket(
+    request: Request, service: TripService = Depends(Provide[Container.trip_service])
+):
+    body = await request.json()
+    body = DialogflowWebhookRequest(**body)
+    params = body.get_parameters()
+
+    try:
+        data = TicketRequest(**params)
+    except Exception as e:
+        logger.error("Error parsing trip parameters", exc_info=True)
+        raise HTTPException(status_code=400, detail="Invalid trip query parameters")
+
+    rs: DialogServiceResult = await service.make_ticket_summary(data)
+
+    return (
+        DialogflowResponseBuilder()
+        .add_text(rs.text)
+        .set_parameters(rs.parameters)
+        .build()
+    )

app/api/v1/user.py

@@ -0,0 +1,60 @@
+from fastapi.params import Depends
+from loguru import logger
+from typing_extensions import Annotated
+from fastapi import APIRouter, Request
+from dependency_injector.wiring import inject, Provide
+
+
+from app.dependencies.containers import Container
+from app.dto.request import DialogflowWebhookRequest
+from app.dto.response import DialogflowResponseBuilder
+from app.dto.user import UserDTO
+from app.services.user_service import UserService
+from app.ner.services.ner import NER
+
+
+router = APIRouter(prefix="/user", tags=["User"])
+
+
+def get_ner_model(request: Request) -> NER:
+    """
+    Dependency to get the NER model.
+    This can be used to inject the NER model into the endpoint.
+    """
+    return request.app.state.ner
+
+
+@router.post("/info")
+@inject
+async def get_user_info(
+    request: Request, service: UserService = Depends(Provide[Container.user_service])
+):
+    body = await request.json()
+    body = DialogflowWebhookRequest(**body)
+    params = body.get_parameters()
+
+    user: UserDTO = await service.get_user()
+    user_exists: bool = user is not None
+
+    parameters = {
+        "is_user_exist": user_exists,
+        "user_name": user.user_name if user_exists else None,
+        "phone_number": user.phone_number if user_exists else None,
+        "email": user.email if user_exists else None,
+    }
+    return DialogflowResponseBuilder().set_parameters(parameters).build()
+
+
+@router.post("/extract-name")
+@inject
+async def extract_user_name(
+    request: Request, service: UserService = Depends(Provide[Container.user_service])
+):
+    body = await request.json()
+    body = DialogflowWebhookRequest(**body)
+
+    text = body.get_text()
+
+    user_name = await service.extract_user_name(text)
+
+    return DialogflowResponseBuilder().set_parameters({"user_name": user_name}).build()

app/client/futabus_client.py

@@ -0,0 +1,210 @@
+import logging
+from app.dto.trip import TripDTO
+from common.external.external_api import api
+from datetime import datetime
+
+from core.conf import settings
+
+
+logger = logging.getLogger(__name__)
+
+
+class FutabusClient:
+    @staticmethod
+    async def get_pickup_points(
+        origin: str | None = None, dest: str | None = None
+    ) -> dict:
+        """
+        Gọi API để lấy metadata điểm đón/trả.
+
+        Args:
+            origin (str): tên văn phòng đi
+            dest (str): tên văn phòng đến
+
+        Returns:
+            dict: dữ liệu metadata chứa các điểm đón/trả
+        """
+        params = {
+            "origin": origin,
+            "dest": dest,
+            "session_id": str(int(datetime.now().timestamp())),
+        }
+        try:
+            response = await api.get(
+                "/search/metadata/pickup-points",
+                params={k: v for k, v in params.items() if v is not None},
+            )
+            if response.get("status") == 200:
+                logger.debug(f"Fetched pickup points for: origin={origin}, dest={dest}")
+                return response.get("data", {})
+            logger.warning(
+                f"Unexpected response status: {response.get('status')} - origin={origin}, dest={dest}"
+            )
+            return {}
+        except Exception as e:
+            logger.exception(
+                f"[FutabusClient] Error fetching pickup points for origin={origin}, dest={dest}: {e}"
+            )
+            return {}
+
+    @staticmethod
+    async def get_route_ids_by_metadata(
+        origin_code: str | None = None,
+        from_id: int | None = None,
+        origin_ids: int | list[int] | None = None,
+        dest_code: str | None = None,
+        to_id: int | None = None,
+        dest_ids: int | list[int] | None = None,
+    ) -> list[int]:
+        """
+        Gọi API `/metadata/office/routes` để lấy routeId theo ID/code văn phòng hoặc tỉnh.
+        """
+        params = {
+            k: v
+            for k, v in {
+                "OriginCode": origin_code,
+                "FromId": from_id,
+                "OriginIds": origin_ids,
+                "DestCode": dest_code,
+                "ToId": to_id,
+                "DestIds": dest_ids,
+            }.items()
+            if v is not None
+        }
+
+        try:
+            response = await api.get("/metadata/office/routes", params=params)
+            if isinstance(response, list):
+                return [item.get("routeId") for item in response if "routeId" in item]
+        except Exception as e:
+            logger.exception("Error fetching route ids by metadata")
+        return []
+
+    @staticmethod
+    async def get_route_ids_by_code(origin_code: str, dest_code: str) -> list:
+        """
+        Gọi API `/booking/api/v2/routes/from/{origin}/to/{dest}` để lấy danh sách route khi metadata không có.
+        """
+        try:
+            endpoint = f"/booking/api/v2/routes/from/{origin_code}/to/{dest_code}"
+            response = await api.get(endpoint)
+            if response.get("Status") == 200 and response.get("Data"):
+                data = response.get("Data", [])
+                route_ids = [item.get("Id", None) for item in data]
+                return route_ids
+            return []
+        except Exception as e:
+            logger.exception("Error fetching route ids by province codes")
+            return []
+
+    @staticmethod
+    async def search_trips(
+        from_time: int, to_time: int, route_ids: list, ticket_count: int = 1
+    ) -> list[dict[str, any]]:
+        """
+        Gọi external API để tìm chuyến xe theo danh sách route_id và thời gian.
+
+        Args:
+            from_time (int): Mốc thời gian bắt đầu (timestamp in ms)
+            to_time (int): Mốc thời gian kết thúc (timestamp in ms)
+            route_ids (list[int]): Danh sách route ID để lọc chuyến
+            ticket_count (int): Số lượng vé cần đặt (default = 1)
+
+        Returns:
+            list[dict[str, any]]: Danh sách chuyến xe phù hợp
+        """
+        if not route_ids:
+            return []
+
+        payload = {
+            "channel": "web_client",
+            "size": 300,
+            "only_online_trip": True,
+            "from_time": from_time,
+            "to_time": to_time,
+            "route_ids": route_ids,
+            "ticket_count": int(ticket_count),
+            "sort_by": ["price", "departure_time"],
+        }
+        logger.info(f"Searching trips with payload: {payload}")
+
+        try:
+            response = await api.post("/search/trips", payload=payload)
+            if response.get("status") == 200:
+                data = response.get("data", {})
+                if data.get("total", 0) > 0:
+                    return data.get("items", [])
+            logger.warning(
+                f"Unexpected response status: {response.get('status')} - No trips found"
+            )
+            return []
+        except Exception as e:
+            logger.exception("Failed to fetch trips from external API")
+            return []
+
+    @staticmethod
+    async def get_seat_trip(trip: TripDTO) -> list[dict]:
+        """
+        Gọi API để lấy thông tin ghế của chuyến xe.
+
+        Args:
+            trip (TripDTO): Thông tin chuyến xe cần lấy ghế.
+
+        Returns:
+            list[dict]: Danh sách ghế của chuyến xe.
+        """
+        params = {
+            k: v
+            for k, v in {
+                "departureDate": trip.raw_departure_date,
+                "departureTime": trip.raw_departure_time,
+                "kind": trip.seat_type_name,
+            }.items()
+            if v is not None
+        }
+        logger.info(f"Parameters for fetching seats: {params}")
+        logger.debug(f"Parameters for fetching seats: {params}")
+        try:
+            rs = await api.get(
+                api_base=settings.API_BASE_URL_VATO,
+                # api_base="https://api-busline.vato.vn/api/buslines/futa/booking",
+                endpoint=f"/seats/{trip.route_id}/{trip.id}",
+                params=params,
+            )
+
+            return rs.get("data", [])
+        except Exception as e:
+            logger.exception(f"Error fetching seats for trip {trip.id}: {e}")
+            return []
+
+    @staticmethod
+    async def get_stop_by_route_and_way(route_id: int, way_id: int) -> list:
+        """
+        Lấy danh sách các điểm dừng của chuyến xe theo route_id và way_id.
+
+        Args:
+            route_id (int): ID của tuyến xe.
+            way_id (int): ID của lộ trình.
+
+        Returns:
+            list[dict]: Danh sách các điểm dừng.
+        """
+        try:
+            params = {
+                k: v
+                for k, v in {
+                    "wayId": way_id,
+                }.items()
+                if v is not None
+            }
+            rs = await api.get(
+                api_base=settings.API_BASE_URL_VATO,
+                endpoint=f"/stops/{route_id}",
+                params=params,
+            )
+            return rs.get("data", [])
+        except Exception as e:
+            logger.exception(
+                f"Error fetching stops for route {route_id}, way {way_id}: {e}"
+            )
+            return []

app/dependencies/containers.py

@@ -0,0 +1,50 @@
+from dependency_injector import containers, providers
+
+from app.client.futabus_client import FutabusClient
+from app.repositories.trip_repository import TripRepository
+from app.resolvers.location_resolver import LocationResolver
+from app.services.location_service import LocationService
+from app.services.time_service import TimeService
+from app.services.trip_service import TripService
+from app.services.user_service import UserService
+from app.ner.services.ner import NER
+
+
+class Container(containers.DeclarativeContainer):
+    wiring_config = containers.WiringConfiguration(
+        modules=[
+            "app.api.v1.trip",
+            "app.api.v1.user",
+            "app.api.v1.location",
+            "app.api.v1.time",
+        ]
+    )
+
+    ner_model = providers.Singleton(NER)
+
+    futabus_client = providers.Singleton(FutabusClient)
+
+    location_service = providers.Factory(
+        LocationService,
+        client=futabus_client,
+    )
+
+    location_resolver = providers.Factory(
+        LocationResolver, location_service=location_service
+    )
+    # NER model (singleton nếu model lớn)
+
+    trip_repository = providers.Factory(TripRepository, client=futabus_client)
+
+    trip_service = providers.Factory(
+        TripService,
+        trip_repository=trip_repository,
+        futabus_client=futabus_client,
+        location_resolver=location_resolver,
+    )
+
+    time_service = providers.Factory(
+        TimeService,
+    )
+
+    user_service = providers.Factory(UserService, model=ner_model)

app/dialogflow/api/router.py

@@ -1,12 +0,0 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-from fastapi import APIRouter
-
-from app.dialogflow.api.v1.dialogflow import router as dialogflow_router
-from core.conf import settings
-
-from utils.life_span import lifespan
-
-v1 = APIRouter(prefix=settings.FASTAPI_API_V1_PATH, lifespan=lifespan)
-
-v1.include_router(dialogflow_router)

app/dialogflow/api/v1/dialogflow.py

@@ -1,1059 +0,0 @@
-from typing import Dict, List
-from fastapi import FastAPI, APIRouter, HTTPException, Request, Response # type: ignore
-from fastapi.responses import JSONResponse, RedirectResponse, HTMLResponse # type: ignore
-from datetime import datetime, timedelta
-from fastapi.templating import Jinja2Templates
-from app.dialogflow.services.dialog_service import dialog_service
-from app.ner.services.ner import NER
-from utils.format_data_dialog import extra_time_dialogflow, get_weekday_name, find_surrounding_times
-
-from common.external.external_api import api
-from app.dialogflow.services.origin_codes import origin_codes, get_origin_id_and_code
-from common.response.respone_dialogflow import DialogFlowResponseAPI
-router = APIRouter()
-
-templates = Jinja2Templates(directory="templates")
-
-
-@router.post('/search/origin-city/from/office')
-async def search_origin_office(request: Request):
-    body = await request.json()
-    
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters")
-    
-    raw_departure_city = parameters.get("departure_city", None)
-    origin_office = parameters.get("origin_office")
-    departure_city = None
-    if origin_office and raw_departure_city is None:
-        departure_city = await dialog_service.get_origin_city_from_office(origin_office)
-    parameters = {
-            "departure_city": departure_city
-        }
-    return DialogFlowResponseAPI(parameters=parameters)
-    
-@router.post('/search/destination-city/from/office')
-async def search_destination_office(request: Request):
-    body = await request.json()
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters")
-    
-    raw_destination_city = parameters.get("destination_city", None)
-    dest_office = parameters.get("dest_office")
-    
-    destination_city = None
-    if dest_office and raw_destination_city is None:
-        destination_city = await dialog_service.get_destination_city_from_office(dest_office)
-            
-    parameters = {
-                "destination_city": destination_city
-            }
-    return DialogFlowResponseAPI(parameters=parameters)
-@router.post('/info/confirm')
-async def confirm(request: Request):
-    body = await request.json()
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters", {}) if isinstance(session_info.get("parameters"), dict) else {}
-    raw_date = parameters.get("date")
-    departure_city = parameters.get("departure_city")
-    destination_city = parameters.get("destination_city")
-    origin_office = parameters.get("origin_office")
-    dest_office = parameters.get("dest_office")
-    raw_ticket_number = parameters.get("ticket_number")
-    time = parameters.get("time-select")
-        
-    time = extra_time_dialogflow(time)
-    parameters = {}
-            
-    if isinstance(time, list):
-        parameters["is_time_ambiguous"] = True
-        return DialogFlowResponseAPI(parameters=parameters)
-                
-    ticket_number = int(raw_ticket_number) if raw_ticket_number else 1
-    date = dialog_service.to_datetime_from_Dialogflow(raw_date)
-    date, week_day = get_weekday_name(date)
-    text = [f"""**Thời gian:** {time} - {date} -  {week_day}\n**Số vé:** {ticket_number}"""]
-    temp = ""
-    if origin_office and dest_office:
-        temp = f"""**Điểm đi:** {origin_office}\n**Điểm đến:** {dest_office}\n"""
-    elif origin_office and destination_city:
-        temp = f"""**Điểm đi:** {origin_office}\n**Điểm đến:** {destination_city}\n"""  
-    elif dest_office and departure_city:
-        temp = f"""**Điểm đi:** {departure_city}\n**Điểm đến:** {dest_office}\n"""
-    elif departure_city and destination_city:
-        temp = f"""**Điểm đi:** {departure_city}\n**Điểm đến:** {destination_city}\n"""
-    text[0] = temp + text[0] 
-    payload = {
-                                "richContent": [
-                                    [
-                                        {
-                                            "type": "chips",
-                                            "options": [
-                                                {"text": "Tìm chuyến xe"},
-                                                {"text": "Không, cảm ơn"}
-                                            ]
-                                        }
-                                    ]
-                                ]
-                            }
-    return DialogFlowResponseAPI(text=text, payload=payload)
-@router.post('/routes')
-async def route(request: Request):
-    body = await request.json()
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters")
-    
-    origin_office = parameters.get("origin_office")
-    dest_office = parameters.get("dest_office")
-    
-    raw_departure_city, raw_destination_city, raw_ticket_number , raw_date, _ = dialog_service.get_param_from_dialogflow(body)
-    
-    ticket_count = int(raw_ticket_number) if raw_ticket_number else 1
-
-    if raw_date is None:
-        from_time, to_time = dialog_service.process_dates_to_timestamp()
-        date = datetime.today().date().strftime('%m-%d-%Y')
-    else:
-        date = raw_date.strftime('%m-%d-%Y')
-        from_time, to_time = dialog_service.process_dates_to_timestamp(raw_date)
-    
-    origin_code, origin_id, origin_ids, dest_code, dest_id, dest_ids = None, None, None, None, None, None
-    if origin_office:
-        origin_id, origin_code = await dialog_service.find_id_and_code_provine_by_name_office(origin_office)
-        origin_ids = await dialog_service.find_id_office_by_name_office(origin_office)
-    elif raw_departure_city:
-        origin_id, origin_code = get_origin_id_and_code(raw_departure_city)
-    if dest_office:
-        dest_id, dest_code  = await dialog_service.find_id_and_code_provine_by_name_office(dest_office)
-        dest_ids = await dialog_service.find_id_office_by_name_office(dest_office)
-    elif raw_destination_city:
-        dest_id, dest_code = get_origin_id_and_code(raw_destination_city)
-    route_ids = await dialog_service.search_all_route_ids(origin_code=origin_code, from_id=origin_id, orign_ids=origin_ids, dest_code=dest_code, to_id=dest_id, dest_ids=dest_ids)
-    
-    payload = {
-        "from_time": from_time,
-        "to_time": to_time,
-        "route_ids": route_ids,
-        "ticket_count": 1,
-        "sort_by": ["price", "departure_time"]
-    }
-    try:
-        respone = await api.post("/search/trips", payload=payload)
-        data = respone["data"]["items"]
-        total = respone["data"]["total"]
-        if total > 0:
-            price = data[0]["price"]
-            list_raw_departure_times = sorted(list(set([ trip["raw_departure_time"] for trip in data])))
-            list_raw_seat_type = list(set([ trip["seat_type_name"] for trip in data]))
-            seat_type_trip_string = "**" + "** | **".join(map(str, list_raw_seat_type)) + "**"
-            schedule_time_trip = "**" + "** | **".join(map(str, list_raw_departure_times)) + "**"
-            duration = data[0]["duration"]
-            link = f'https://stag.futabus.vn/dat-ve?from={departure_code}&fromTime={date}&isReturn=false&ticketCount={ticket_count}&to={destination_code}&toTime='
-            text = [f"Tuyến xe **{raw_departure_city}** - **{raw_destination_city}**\n \
-                                     Loại xe: {seat_type_trip_string} \n \
-                                     Thời gian hành trình: {duration} giờ \n \
-                                     Giá vé: **{price}** VND"]
-            payload = {
-                                "richContent": [
-                                    [
-                                        {
-                                            "type": "chips",
-                                            "options": [
-                                                {"text": "Tìm chuyến xe"},
-                                                {"text": "Xem lịch trình khác"},
-                                                {"text": "Không, cảm ơn"}
-                                            ]
-                                        }
-                                    ]
-                                ]
-                            }
-            
-            return DialogFlowResponseAPI(text=text, payload=payload)
-
-        text = [f"Hệ thống không tìm thấy tuyến xe **{raw_departure_city}** - **{raw_destination_city}**.\n Quý khách vui lòng thử lại với lộ trình khác hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."]
-        payload = {
-                                "richContent": [
-                                    [
-                                        {
-                                            "type": "chips",
-                                            "options": [
-                                                {"text": "Xem lịch trình khác"},
-                                                {"text": "Không, cảm ơn"}
-                                            ]
-                                        }
-                                    ]
-                                ]
-                            }
-        return DialogFlowResponseAPI(text=text, payload=payload)
-    except Exception as e:
-        return DialogFlowResponseAPI(text=["Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."])
-
-
-@router.post('/price')
-async def price(request: Request):
-    body = await request.json()
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters")
-    
-    raw_departure_city, raw_destination_city, _, raw_date, _ = dialog_service.get_param_from_dialogflow(body)
-    
-    if raw_date is None:
-        from_time, to_time = dialog_service.process_dates_to_timestamp()
-    from_time, to_time = dialog_service.process_dates_to_timestamp(raw_date)
-
-    origin_office = parameters.get("origin_office")
-    dest_office = parameters.get("dest_office")
-    
-    origin_code, origin_id, origin_ids, dest_code, dest_id, dest_ids = None, None, None, None, None, None
-    if origin_office:
-        origin_id, origin_code = await dialog_service.find_id_and_code_provine_by_name_office(origin_office)
-        origin_ids = await dialog_service.find_id_office_by_name_office(origin_office)
-    elif raw_departure_city:
-        origin_id, origin_code = get_origin_id_and_code(raw_departure_city)
-    if dest_office:
-        dest_id, dest_code  = await dialog_service.find_id_and_code_provine_by_name_office(dest_office)
-        dest_ids = await dialog_service.find_id_office_by_name_office(dest_office)
-    elif raw_destination_city:
-        dest_id, dest_code = get_origin_id_and_code(raw_destination_city)
-    route_ids = await dialog_service.search_all_route_ids(origin_code=origin_code, from_id=origin_id, orign_ids=origin_ids, dest_code=dest_code, to_id=dest_id, dest_ids=dest_ids)
-    
-    payload = {
-        "from_time": from_time,
-        "to_time": to_time,
-        "route_ids": route_ids,
-        "ticket_count": 1,
-        "sort_by": ["price", "departure_time"]
-    }
-    try:
-        respone = await api.post("/search/trips", payload=payload)
-        total = respone["data"]["total"]
-        if total > 0:
-            price = respone["data"]["items"][0]["price"]
-            text = [f"Tuyến xe **{raw_departure_city}** - **{raw_destination_city}**\n"
-                                     f"Giá vé: **{price}** VND.\n \
-                                     Bạn có muốn đặt vé không?"
-                                     ]
-            payload = {
-                                "richContent": [
-                                    [
-                                        {
-                                            "type": "chips",
-                                            "options": [
-                                                {"text": "Có, tôi muốn đặt vé"},
-                                                {"text": "Xem giá vé tuyến xe khác"},
-                                                {"text": "Không, cảm ơn"}
-                                            ]
-                                        }
-                                    ]
-                                ]
-                            }
-            return DialogFlowResponseAPI(text=text, payload=payload)
-        text = [f"Hệ thống không tìm thấy tuyến xe **{raw_departure_city}** - **{raw_destination_city}**.\n Quý khách vui lòng thử lại với lộ trình khác hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."]
-        payload={
-                                "richContent": [
-                                    [
-                                        {
-                                            "type": "chips",
-                                            "options": [
-                                                {"text": "Xem giá vé lịch trình khác"},
-                                                {"text": "Không, cảm ơn"}
-                                            ]
-                                        }
-                                    ]
-                                ]
-                            }
-        
-        return DialogFlowResponseAPI(text=text, payload=payload)
-    except:
-        return DialogFlowResponseAPI(text=["Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."])
-
-@router.post('/trip/list')
-async def get_trip_list(request: Request) -> Response:
-    body = await request.json()
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters")
-    
-    raw_departure_city, raw_destination_city, raw_ticket_number, raw_date, _ = dialog_service.get_param_from_dialogflow(body)
-    
-    origin_office = parameters.get("origin_office")
-    dest_office = parameters.get("dest_office")
-    time = parameters.get("time-select")
-    
-    time = extra_time_dialogflow(time)
-    
-    if isinstance(time, list):
-        parameters = {
-            "is_time_ambiguous": True
-        }
-        return DialogFlowResponseAPI(parameters=parameters)
-    
-    from_time, to_time = dialog_service.process_dates_to_timestamp(raw_date)
-    
-    ticket_count = int(raw_ticket_number) if raw_ticket_number else 1
-
-    origin_code, origin_id, origin_ids, dest_code, dest_id, dest_ids = None, None, None, None, None, None
-    if origin_office:
-        origin_id, origin_code = await dialog_service.find_id_and_code_provine_by_name_office(origin_office)
-        origin_ids = await dialog_service.find_id_office_by_name_office(origin_office)
-    elif raw_departure_city:
-        origin_id, origin_code = get_origin_id_and_code(raw_departure_city)
-    if dest_office:
-        dest_id, dest_code  = await dialog_service.find_id_and_code_provine_by_name_office(dest_office)
-        dest_ids = await dialog_service.find_id_office_by_name_office(dest_office)
-    elif raw_destination_city:
-        dest_id, dest_code = get_origin_id_and_code(raw_destination_city)
-    route_ids = await dialog_service.search_all_route_ids(origin_code=origin_code, from_id=origin_id, orign_ids=origin_ids, dest_code=dest_code, to_id=dest_id, dest_ids=dest_ids)    
-    
-    try:
-        data = await dialog_service.search_trip(from_time, to_time, route_ids, ticket_count)
-        if len(data) > 0:
-            
-            if origin_office and dest_office:
-                trip_by_time_office = dialog_service.get_trip_by_time_and_office_id(data, time, origin_ids, dest_ids)
-                # Nếu có chuyến xe theo thời gian và văn phòng chỉ định
-                if trip_by_time_office:                    
-                    parameters = {
-                            "is_valid_trip": True,
-                            "trip": trip_by_time_office
-                        }
-                    return DialogFlowResponseAPI(parameters=parameters)
-                
-                # Nếu không có chuyến xe theo thời gian và văn phòng chỉ định   
-                # Danh sách chuyến xe khớp với văn phòng đón hoặc văn phòng trả
-            #     data_by_office = dialog_service.get_all_trip_by_office(data, origin_ids, dest_ids)
-            # elif origin_office:
-            #     data_by_office = dialog_service.get_all_trip_by_office(data,origin_id=origin_ids)
-            # elif dest_office:
-            #     data_by_office = dialog_service.get_all_trip_by_office(data, dest_id=dest_ids)
-            # else:
-            
-            data_by_office = data
-            # Tìm 4 chuyến xe gần thời gian chỉ định nhất
-            trip_surrounding_time = dialog_service.get_surrounding_trip(data_by_office, time, num_trip=4)
-            trip_dialogflow = []    
-                
-            for trip in trip_surrounding_time:
-                if ticket_count <= trip["empty_seat_quantity"]:
-                    trip_dialogflow.append({"trip_id": trip["id"], "route":f'{trip["raw_departure_time"]} | {trip["route"]["origin_hub_name"]} => {trip["route"]["dest_hub_name"]}'})
-            text = ["Quý khách vui lòng lựa chọn chuyến xe\n" + "\n".join(f'{i+1}. {trip["route"]}' for i, trip in enumerate(trip_dialogflow))]
-            payload={
-                                "richContent": [
-                                    [
-                                        {
-                                            "type": "chips",
-                                            "options": [ 
-                                                {"text": trip["route"]} for trip in (trip_dialogflow)
-                                            ]
-                                        }
-                                    ]
-                                ]
-                            }
-            parameters = {
-                    "trip_select": trip_dialogflow,
-                    "trips": trip_surrounding_time,
-                }
-            return DialogFlowResponseAPI(text=text, payload=payload,parameters=parameters)                
-            
-        text = [f"Hệ thống không tìm thấy tuyến xe **{raw_departure_city}** - **{raw_destination_city}**.\n Quý khách vui lòng thử lại với lộ trình khác hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."]
-        payload={
-                                "richContent": [
-                                    [
-                                        {
-                                            "type": "chips",
-                                            "options": [
-                                                {"text": "Xem tuyến xe khác"},
-                                                {"text": "Không, cảm ơn"}
-                                            ]
-                                        }
-                                    ]
-                                ]
-                            }
-                        
-        return DialogFlowResponseAPI(text=text, payload=payload)
-    except Exception as e:
-        print(e)
-        return JSONResponse(
-                    {
-                    "fulfillment_response": {
-                        "messages": [
-                        {
-                            "text": {
-                            "text": ["Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."]
-                            }
-                        }
-                        ]
-                    },
-                }
-        )
-
-# Không xài 
-@router.post('/trip/route/list')
-async def booking_trip(request: Request) -> Response:
-    body = await request.json()
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters")
-    
-    raw_departure_city, raw_destination_city, raw_ticket_number, raw_date, raw_time_of_day = dialog_service.get_param_from_dialogflow(body)
-    
-    origin_office = parameters.get("origin_office")
-    dest_office = parameters.get("dest_office")
-    
-    from_time, to_time = dialog_service.process_dates_to_timestamp(raw_date)
-    
-    ticket_count = int(raw_ticket_number) if raw_ticket_number else 1
-
-    origin_code, origin_id, origin_ids, dest_code, dest_id, dest_ids = None, None, None, None, None, None
-    if origin_office:
-        origin_id, origin_code = await dialog_service.find_id_and_code_provine_by_name_office(origin_office)
-        origin_ids = await dialog_service.find_id_office_by_name_office(origin_office)
-    elif raw_departure_city:
-        origin_id, origin_code = get_origin_id_and_code(raw_departure_city)
-    if dest_office:
-        dest_id, dest_code  = await dialog_service.find_id_and_code_provine_by_name_office(dest_office)
-        dest_ids = await dialog_service.find_id_office_by_name_office(dest_office)
-    elif raw_destination_city:
-        dest_id, dest_code = get_origin_id_and_code(raw_destination_city)
-    route_ids = await dialog_service.search_all_route_ids(origin_code=origin_code, from_id=origin_id, orign_ids=origin_ids, dest_code=dest_code, to_id=dest_id, dest_ids=dest_ids)
-    try:
-        data = await dialog_service.search_trip(from_time, to_time, route_ids, ticket_count)
-        if len(data) > 0:
-                trips = []
-                routes_name = []
-                for trip in data:
-                    if ticket_count <= trip["empty_seat_quantity"]:
-                        route = f"{trip['route']['origin_hub_name']} => {trip['route']['dest_hub_name']}"
-                        if trip["route"] and route not in routes_name:
-                            routes_name.append(f"{trip['route']['origin_hub_name']} => {trip['route']['dest_hub_name']}")
-                        trips.append({
-                                "route_name": f"{trip['route']['origin_hub_name']} => {trip['route']['dest_hub_name']}",
-                                "route_id": trip["route_id"],
-                                "id": trip["id"],
-                                "departure_date": trip["raw_departure_date"],
-                                "departure_time": trip["raw_departure_time"],
-                                "kind": trip["seat_type_name"],
-                                "way_id": trip["way_id"]
-
-                                })
-                text = ["Quý khách vui lòng lựa chọn chuyến xe\n" + "\n".join(f"{i+1}. {name}" for i, name in enumerate(routes_name))]
-                payload={
-                                "richContent": [
-                                    [
-                                        {
-                                            "type": "chips",
-                                            "options": [ 
-                                                {"text": name} for name in (routes_name)
-                                            ]
-                                        }
-                                    ]
-                                ]
-                            }
-                parameters = {
-                    "trip_list": trips,
-                    "routes_name": routes_name
-                }
-                return DialogFlowResponseAPI(text=text, payload=payload,parameters=parameters)                
-            
-        text = [f"Hệ thống không tìm thấy tuyến xe **{raw_departure_city}** - **{raw_destination_city}**.\n Quý khách vui lòng thử lại với lộ trình khác hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."]
-        payload={
-                                "richContent": [
-                                    [
-                                        {
-                                            "type": "chips",
-                                            "options": [
-                                                {"text": "Xem tuyến xe khác"},
-                                                {"text": "Không, cảm ơn"}
-                                            ]
-                                        }
-                                    ]
-                                ]
-                            }
-                        
-        return DialogFlowResponseAPI(text=text, payload=payload)
-    except Exception as e:
-        print(e)
-        return JSONResponse(
-                    {
-                    "fulfillment_response": {
-                        "messages": [
-                        {
-                            "text": {
-                            "text": ["Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."]
-                            }
-                        }
-                        ]
-                    },
-                }
-        )
-
-@router.post('/trip/check-trip')
-async def is_valid_select_trip(request: Request) -> Response:
-    body = await request.json()
-    raw_input = body.get("text", "")
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters")
-    trip_select: list[dict[str, any]] = parameters.get("trip_select")
-    trips: list[dict[str, any]] = parameters.get("trips")
-    
-    if trip_select is None and trips is None:
-        return DialogFlowResponseAPI()
-    
-    if raw_input:
-        raw_input = raw_input.strip()
-        for item in trip_select:
-            if item["route"] == raw_input:
-                id = int(item["trip_id"])
-                trip = dialog_service.get_trip_by_id(id, trips)
-                parameters = {
-                    "is_valid_trip": True,
-                    "trip": trip
-                }
-                return DialogFlowResponseAPI(parameters=parameters)
-                
-    parameters = {
-                "is_valid_trip": False,
-            }
-    return DialogFlowResponseAPI(parameters=parameters)
-
-@router.post('/trip/check-time-ambiguous')
-async def check_time_ambiguous(request: Request) -> Response:
-    try:
-        body = await request.json()
-        session_info = body.get("sessionInfo", {})
-        parameters = session_info.get("parameters")
-        time = parameters.get("time-select")
-        
-        parameters = {}
-       
-        time = extra_time_dialogflow(time)
-                    
-        
-        if isinstance(time, list) or time is None:
-            parameters["is_time_ambiguous"] = True
-        else:
-            parameters["is_time_ambiguous"] = None  
-              
-        return DialogFlowResponseAPI(parameters=parameters)
-        
-    except Exception as e:
-        print(e)
-        return DialogFlowResponseAPI(text=["Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."])
-@router.post('/trip/select-time-ambiguous')
-async def select_time(request: Request) -> Response:
-    try:
-        body = await request.json()
-        session_info = body.get("sessionInfo", {})
-        parameters = session_info.get("parameters")
-        time = parameters.get("time-select")
-        
-        parameters = {}
-       
-        time = extra_time_dialogflow(time)
-        
-        text = []
-        
-        if time is None:
-            text = ["Quý khách vui lòng chỉ định thời gian đi cụ thể trong ngày?"]
-        
-        elif isinstance(time, list):
-            text = [f"Quý khách dự định đi vào lúc"]
-            payload={
-                                    "richContent": [
-                                        [
-                                            {
-                                                "type": "chips",
-                                                "options": [ 
-                                                    {"text": item} for item in (time)
-                                                ]
-                                            }
-                                        ]
-                                    ]
-                                }
-        else:
-            parameters["is_time_ambiguous"] = None  
-              
-        return DialogFlowResponseAPI(text=text,payload=payload,parameters=parameters)
-        
-    except Exception as e:
-        print(e)
-        return DialogFlowResponseAPI(text=["Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."])
-
-
-# Không xài
-@router.post('/trip/time-trip-list')
-async def time_trip(request: Request) -> Response:
-    try:
-        body = await request.json()
-        session_info = body.get("sessionInfo", {})
-        parameters = session_info.get("parameters")
-        trip_list: list[Dict[str, any]] = parameters.get("trip_list", [])
-        raw_route_id = parameters.get("route_id")
-        route_name = parameters.get("route_name", "")
-        raw_time = parameters.get("time-select")
-        time_list = parameters.get("time_list", [])
-        is_has_time = parameters.get("is_has_time", False)
-        
-        route_id = int(raw_route_id) if raw_route_id else None
-        
-        text = []
-        quick_time_reply = []
-        parameters = {}
-       
-        
-        
-        if raw_time:
-            time = extra_time_dialogflow(raw_time)
-            quick_time_reply = time_list
-            
-            if isinstance(time, list):
-                parameters["time_select"] = None
-                parameters["is_time_ambiguous"] = True
-                text = [f"Quý khách dự định đi vào lúc"]
-                payload={
-                                    "richContent": [
-                                        [
-                                            {
-                                                "type": "chips",
-                                                "options": [ 
-                                                    {"text": item} for item in (time)
-                                                ]
-                                            }
-                                        ]
-                                    ]
-                                }
-                return DialogFlowResponseAPI(text=text, payload=payload ,parameters=parameters)
-            else:
-                if is_has_time:
-                    text = [f"Quý khách lựa chọn thời gian chuyến {route_name}\n" + " | ".join([item["time"] for item in quick_time_reply])]
-                    parameters["is_has_time"] = None
-                else:
-                    parameters["is_has_time"] = True
-                return DialogFlowResponseAPI(text=text,parameters=parameters)
-        else:
-            for trip in trip_list:
-                if (trip["route_id"]) == route_id:
-                    time_list.append({"time": trip["departure_time"], "trip_id": trip["id"]})
-            quick_time_reply = time_list
-            text = [f"Quý khách lựa chọn thời gian chuyến {route_name}\n" + " | ".join([item["time"] for item in quick_time_reply])]
-        
-        parameters["time_list"] =  time_list
-        payload={
-                                    "richContent": [
-                                        [
-                                            {
-                                                "type": "chips",
-                                                "options": [ 
-                                                    {"text": time["time"]} for time in (quick_time_reply)
-                                                ]
-                                            }
-                                        ]
-                                    ]
-                                }
-        return DialogFlowResponseAPI(text=text, payload=payload ,parameters=parameters)
-    except Exception as e:
-        print(e)
-        return DialogFlowResponseAPI(text=["Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."])
-
-# Không xài
-@router.post('/trip/check-time-select')
-async def is_valid_select_time(request: Request) -> Response:
-    try:
-        body = await request.json()
-        session_info = body.get("sessionInfo", {})
-        parameters = session_info.get("parameters")
-        trip_list: list[Dict[str, any]] = parameters.get("trip_list", [])
-        raw_route_id = parameters.get("route_id")
-        
-        raw_departure_city, raw_destination_city, raw_ticket_number, raw_date, raw_time_of_day = dialog_service.get_param_from_dialogflow(body)
-        route_id = int(raw_route_id) if raw_route_id else None
-
-
-        from_time, to_time = dialog_service.process_dates_to_timestamp(raw_date)
-        ticket_count = int(raw_ticket_number) if raw_ticket_number else 1
-        
-        origin_office = parameters.get("origin_office")
-        dest_office = parameters.get("dest_office")
-
-        origin_code, origin_id, origin_ids, dest_code, dest_id, dest_ids = None, None, None, None, None, None
-        if origin_office:
-            origin_id, origin_code = await dialog_service.find_id_and_code_provine_by_name_office(origin_office)
-            origin_ids = await dialog_service.find_id_office_by_name_office(origin_office)
-        elif raw_departure_city:
-            origin_id, origin_code = get_origin_id_and_code(raw_departure_city)
-        if dest_office:
-            dest_id, dest_code  = await dialog_service.find_id_and_code_provine_by_name_office(dest_office)
-            dest_ids = await dialog_service.find_id_office_by_name_office(dest_office)
-        elif raw_destination_city:
-            dest_id, dest_code = get_origin_id_and_code(raw_destination_city)
-        route_ids = await dialog_service.search_all_route_ids(origin_code=origin_code, from_id=origin_id, orign_ids=origin_ids, dest_code=dest_code, to_id=dest_id, dest_ids=dest_ids)
-        
-
-        time_list: list[Dict[str, any]] = parameters.get("time_list", [])
-        time_select = parameters.get("time-select")
-        route_name = parameters.get("route_name")
-        is_has_time = parameters.get("is_has_time")
-        
-        if len(time_list) == 0:
-            for trip in trip_list:
-                if (trip["route_id"]) == route_id:
-                    time_list.append({"time": trip["departure_time"], "trip_id": trip["id"]})
-        
-        if time_select:
-            time_select = extra_time_dialogflow(time_select)
-            for time in time_list:
-                if time_select == time["time"]:
-                    id = int(time["trip_id"])
-                    trip = await dialog_service.search_trip_by_id(id, from_time, to_time, route_ids, ticket_count)
-                    departure_date = trip["raw_departure_date"]
-                    parameters = {
-                        "is_valid_time": True,
-                        "departure_time": time_select,
-                        "departure_date": departure_date,
-                        "trip": trip,
-                    }
-                    text = [f'Quý khách chọn chuyến **{time_select}** | **{route_name}**']
-                    return DialogFlowResponseAPI(text=text, parameters=parameters)
-            if is_has_time:
-                time_list = find_surrounding_times(time_list, time_select)
-                parameters = {
-                    "time_list": time_list,
-                    "is_valid_time": False
-                }
-                
-                return DialogFlowResponseAPI(parameters=parameters)
-                
-        parameters = {
-                "time_select": None,
-                "is_valid_time": False
-            }
-        text = []
-        return DialogFlowResponseAPI(text=text, parameters=parameters)
-    except Exception as e:
-        print(e)
-        return DialogFlowResponseAPI(text=["Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."])
-
-
-@router.post('/trip/seats')
-async def seats_trip(request: Request) -> Response:
-    try:
-        body = await request.json()
-        session_info = body.get("sessionInfo", {})
-        parameters = session_info.get("parameters")
-        
-        trip = parameters.get("trip", None)
-        route_id = int(trip.get("route_id")) if trip.get("route_id") else None
-        trip_id = int(trip.get("id")) if trip.get("id") else None
-        
-        departure_date: str = trip.get("raw_departure_date")
-        departure_time: str = trip.get("raw_departure_time")
-        kind: str = trip.get("seat_type_name")
-
-        seats = await dialog_service.seats_trip(route_id, trip_id, departure_date, departure_time, kind)
-        seats_empty = [ seat for seat in seats if seat["bookStatus"] == 0 ]
-        seats_empty.sort(key=lambda x: x["chair"])
-        text=["Quý khách vui lòng chọn ghế"]
-        payload = {}
-        if seats_empty:
-            payload={
-                                    "richContent": [
-                                        [
-                                            {
-                                                "type": "chips",
-                                                "options": [ 
-                                                    {"text": seat["chair"]} for seat in (seats_empty)
-                                                ]
-                                            }
-                                        ]
-                                    ]
-                                }
-        else:
-            text = ["Không còn ghế trống. Quý khách vui lòng lựa chọn chuyến khác"]
-        parameters = {
-            "seat_list": seats
-        }
-        return DialogFlowResponseAPI(text=text, payload=payload, parameters=parameters)
-    except Exception as e:
-        print(e)
-        return DialogFlowResponseAPI(text=["Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."])
-@router.post('/trip/check-seat-select')
-async def is_valid_select_seat(request: Request) -> Response:
-    try:
-        body = await request.json()
-        session_info = body.get("sessionInfo", {})
-        parameters = session_info.get("parameters")
-        
-        trip: list[dict[str, any]] = parameters.get("trip", None)
-        
-        route_id = int(trip.get("route_id")) if trip.get("route_id") else None
-        
-        trip_id = int(trip.get("id")) if trip.get("id") else None
-        
-        departure_date: str = trip.get("raw_departure_date")
-        
-        departure_time: str = trip.get("raw_departure_time")
-        
-        kind: str = trip.get("seat_type_name")
-        
-        seat: str = parameters.get("seat")
-
-        is_valid = await dialog_service.is_valid_select_seat(seat, route_id, trip_id, departure_date, departure_time, kind)
-        if is_valid:
-            parameters = {
-                "is_valid_seat": True,
-                "seat": seat
-            }
-            text = [f"Quý khách chọn ghế **{seat.upper()}**"]
-        else:
-            parameters = {
-                "is_valid_seat": False,
-                "seat": None        
-            }
-            text = [f"Ghế **{seat.upper()}** không hợp lệ. Quý khách vui lòng chọn ghế khác"]
-        return DialogFlowResponseAPI(text=text, parameters=parameters)
-        
-    except Exception as e:
-        return DialogFlowResponseAPI(text=["Hệ thống xảy ra lỗi. Quý khách vui lòng thử lại sau hoặc liên hệ Trung tâm tổng đài 1900 6067 để được hỗ trợ."])
-
-@router.post('/trip/check-exist-user-info')
-async def check_exist_user_info(request: Request) -> Response:
-    body = await request.json()
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters")
-    
-    is_exist_user_info = await dialog_service.check_exist_user_info()
-    
-    user_info = {}
-    
-    if is_exist_user_info:
-        user_info = await dialog_service.get_user_info()
-
-    user_name = user_info.get("user_name")
-    phone_number = user_info.get("phone_number")
-    email = user_info.get("email")
-    
-    parameters = {
-        "is_user_exist": is_exist_user_info,
-        "user_name":  user_name,
-        "phone_number": phone_number,
-        "email": email
-    }
-        
-    return DialogFlowResponseAPI(parameters=parameters)
-
-@router.post('/trip/extract-user-name')
-async def get_user_name(request: Request) -> Response:
-    body = await request.json()
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters")
-    
-    raw_text_user_name = (body.get("text",""))
-    
-    ner: NER = request.app.state.ner
-    
-    user_name = await dialog_service.extract_user_name(text=raw_text_user_name, ner=ner)
-    
-    parameters = {
-        "user_name": user_name
-    }
-        
-    return DialogFlowResponseAPI(parameters=parameters)
-
-@router.post('/trip/stop/pickup')
-async def pickup(request: Request) -> Response:
-    body = await request.json()
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters")
-    
-    trip: list[dict[str, any]] = parameters.get("trip", {})
-    
-    route_id = int(trip.get("route_id")) if trip.get("route_id") else None
-    
-    way_id = int(trip.get("way_id")) if trip.get("way_id") else None
-    
-    pickup_list = await dialog_service.pickup_list(route_id, way_id)
-    
-    text=["Quý khách vui lòng chọn điểm đón"]
-    payload={
-                                "richContent": [
-                                    [
-                                        {
-                                            "type": "chips",
-                                            "options": [ 
-                                                {"text": pickup["name"]} for pickup in (pickup_list)
-                                            ]
-                                        }
-                                    ]
-                                ]
-                            }
-    parameters = {
-        "pickup_list": pickup_list
-    }
-    
-    return DialogFlowResponseAPI(text=text, payload=payload, parameters=parameters)
-
-@router.post('/trip/check-pickup-select')
-async def is_valid_select_pickup(request: Request) -> Response:
-    body = await request.json()
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters")
-    
-    trip: list[dict[str, any]] = parameters.get("trip", {})
-    
-    route_id = int(trip.get("route_id")) if trip.get("route_id") else None
-    
-    way_id = int(trip.get("way_id")) if trip.get("way_id") else None
-    
-    raw_input = (body.get("text",""))
-    pickup = raw_input.strip()
-
-    is_valid = await dialog_service.is_valid_pickup(pickup, route_id, way_id)
-
-    if is_valid:
-        parameters = {
-            "is_valid_pickup": True,
-            "pick_up": pickup
-        }
-        text = [f"Quý khách chọn điểm đón **{pickup}**"]
-    else:
-        parameters = {
-            "is_valid_pickup": False,
-        }
-        # text = [f"Điểm đón không hợp lệ. Quý khách vui lòng chọn điểm đón khác"]
-        text = []
-
-    return DialogFlowResponseAPI(text=text, parameters=parameters)
-
-@router.post('/trip/stop/dropoff')
-async def dropoff(request: Request) -> Response:
-    body = await request.json()
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters")
-    
-    trip: list[dict[str, any]] = parameters.get("trip", {})
-    
-    route_id = int(trip.get("route_id")) if trip.get("route_id") else None
-    
-    way_id = int(trip.get("way_id")) if trip.get("way_id") else None
-    
-    dropoff_list = await dialog_service.dropoff_list(route_id, way_id)
-
-    text=["Quý khách vui lòng chọn điểm trả khách"]
-    payload={
-                                "richContent": [
-                                    [
-                                        {
-                                            "type": "chips",
-                                            "options": [ 
-                                                {"text": dropoff["name"]} for dropoff in (dropoff_list)
-                                            ]
-                                        }
-                                    ]
-                                ]
-                            }
-    parameters = {
-        "dropoff_list": dropoff_list
-    }
-    
-    return DialogFlowResponseAPI(text=text, payload=payload, parameters=parameters)
-
-@router.post('/trip/check-dropoff-select')
-async def is_valid_select_dropoff(request: Request) -> Response:
-    body = await request.json()
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters")
-    
-    trip: list[dict[str, any]] = parameters.get("trip", {})
-    
-    route_id = int(trip.get("route_id")) if trip.get("route_id") else None
-    
-    way_id = int(trip.get("way_id")) if trip.get("way_id") else None
-    
-    raw_input = (body.get("text",""))
-    dropoff = raw_input.strip()
-
-    is_valid = await dialog_service.is_valid_dropoff(dropoff, route_id, way_id)
-
-    if is_valid:
-        parameters = {
-            "is_valid_dropoff": True,
-            "drop_off": dropoff
-        }
-        text = [f"Quý khách chọn điểm trả khách **{dropoff}**"]
-    else:
-        parameters = {
-            "is_valid_dropoff": False,
-        }
-        # text = [f"Điểm trả khách không hợp lệ. Quý khách vui lòng chọn điểm trả khách khác"]
-        text = []
-
-    return DialogFlowResponseAPI(text=text, parameters=parameters)
-
-@router.post('/ticket/info')
-async def response_ticket_info(request: Request) -> Response:
-    body = await request.json()
-    session_info = body.get("sessionInfo", {})
-    parameters = session_info.get("parameters")
-
-    raw_user_name = parameters.get("user_name")
-    user_name = raw_user_name["name"] if raw_user_name else None
-    
-    phone_number = parameters.get("phone_number")
-    
-    email = parameters.get("email")
-    
-    seat: str = parameters.get("seat")
-    
-    pickup = parameters.get("pick_up")
-    
-    dropoff = parameters.get("drop_off")
-
-    trip = parameters.get("trip", {})
-
-    route_name = trip["route"]["name"]
-    
-    time = trip["raw_departure_time"]
-    
-    date = trip["raw_departure_date"]
-    
-    price = int(trip.get("price")) if trip.get("price") else None
-
-    text = [
-        f" \
-        **Thông tin hành khách**\n\
-        **Họ và tên** {user_name} \n\
-        **Số điện thoại** {phone_number}\n\
-        **Email** {email} \n\
-        **Thông tin lượt đi**\n\
-        **Tuyến xe** {route_name} \n\
-        **Thời gian xuất bến** {time} {date} \n\
-        **Số ghế** {seat.upper()} \n\
-        **Điểm lên xe** {pickup} \n\
-        **Điểm trả khách** {dropoff} \n\
-        **Tổng tiền lượt đi** {price} VND \
-        "
-        ]
-    
-    payload={
-                                "richContent": [
-                                    [
-                                        {
-                                            "type": "chips",
-                                            "options": [ 
-                                                {"text": "Đặt vé"},
-                                                {"text": "Không, cảm ơn"}
-                                            ]
-                                        }
-                                    ]
-                                ]
-                            }
-
-
-    return DialogFlowResponseAPI(text=text, payload=payload)
-
-
-@router.get("/")
-def home():
-    return "Hello World!"
-
-@router.get('/chatbot', response_class=HTMLResponse)
-async def index(request: Request):
-    return templates.TemplateResponse("index.html", {"request": request, "title": "Dawi Chatbot"})

app/dialogflow/services/dialog_service.py

@@ -1,432 +0,0 @@
-from bisect import bisect_left
-from datetime import datetime, timedelta
-
-from fastapi import logger
-
-from app.ner.services.ner import NER
-
-from common.external.external_api import api
-from core.conf import settings
-
-class DialogService:
-    @staticmethod
-    def to_datetime_from_Dialogflow(time: dict):
-        date_time = datetime(int(time["year"]), int(time["month"]), int(time["day"]))
-        return date_time
-    
-    @staticmethod
-    def process_dates_to_timestamp(from_time: datetime = None, to_time: datetime = None):
-        if to_time is None and from_time is not None:
-            to_time = from_time.replace(hour=23, minute=59, second=59)
-        
-        if from_time is None:
-            today = datetime.today().date()
-            from_time = datetime.combine(today, datetime.min.time())
-            to_time = datetime.combine(today, datetime.max.time()) - timedelta(microseconds=1)
-
-        return int(from_time.timestamp()) * 1000 , int(to_time.timestamp()) * 1000
-
-    def get_param_from_dialogflow(self, body: any):
-        session_info = body.get("sessionInfo", {})
-        parameters = session_info.get("parameters", {}) if isinstance(session_info.get("parameters"), dict) else {}
-        raw_date = parameters.get("date")
-        if raw_date is not None:
-            raw_date = self.to_datetime_from_Dialogflow(raw_date)
-        raw_departure_city = parameters.get("departure_city")
-        raw_destination_city = parameters.get("destination_city")
-        raw_ticket_number = parameters.get("ticket_number")
-        raw_time_of_day = parameters.get("time_of_day")
-        return raw_departure_city, raw_destination_city, raw_ticket_number, raw_date, raw_time_of_day
-
-    # Danh sách route_id API v1
-    @staticmethod
-    async def search_route_ids_one_way_by_metadata(origin_code: str = None, from_id: int = None, orign_ids: int = None, dest_code: str = None, to_id: str = None, dest_ids: str = None):
-        params = {k: v for k, v in {
-                    "OriginCode": origin_code,
-                    "FromId": from_id,
-                    "OriginIds": orign_ids,
-                    "DestCode": dest_code,
-                    "ToId": to_id,
-                    "DestIds": dest_ids
-                }.items() if v is not None}
-        response = await api.get(f'/metadata/office/routes' , params=params)
-        route_ids = []
-        if isinstance(response, list):
-                route_ids = [route.get("routeId", None) for route in response]
-                return route_ids
-        return []
-    
-    # Danh sách route_id v2 khi API route v1 không có dữ liệu
-    @staticmethod
-    async def search_route_ids_one_way_by_origin_dest_code(origin: str, dest: str):
-        try:
-            route_ids = []
-            response = await api.get(f'/booking/api/v2/routes/from/{origin}/to/{dest}')
-            if response.get("Status") == 200:
-                if response.get("Data"):
-                    data = response.get("Data")
-                    route_ids = [ route.get("Id", None) for route in data ]
-            return route_ids
-        except Exception as e:
-            print(e)
-            raise Exception("Error fetching seats data")
-    
-    async def search_all_route_ids(self, origin_code: str = None, from_id: int = None, orign_ids: int = None, dest_code: str = None, to_id: str = None, dest_ids: str = None):
-        route_ids = await self.search_route_ids_one_way_by_metadata(origin_code, from_id, orign_ids, dest_code, to_id, dest_ids)
-        if len(route_ids) == 0:
-            route_ids = await self.search_route_ids_one_way_by_origin_dest_code(origin_code, dest_code)
-        return route_ids
-    # Danh sách ghế 
-    @staticmethod
-    async def seats_trip(route_id: int, trip_id:int, departure_date: str, departure_time: str, kind: str):
-        try:
-            params ={k: v for k, v in {
-                    "departureDate": departure_date,
-                    "departureTime": departure_time,
-                    "kind": kind
-                }.items() if v is not None}
-            if params is None:
-                return []
-            response = await api.get(api_base=settings.API_BASE_URL_VATO, endpoint=f"/seats/{route_id}/{trip_id}" , params=params)
-            if not response.get("data"):
-                return []
-            seats = response["data"]
-            return seats
-        except Exception as e:
-            print(e)
-            raise Exception("Error fetching seats data")
-
-    # Call API để lấy danh sách chuyến đi
-    @staticmethod
-    async def search_trip(from_time: int, to_time: int, route_ids: list[int], ticket_count: int = 1):
-        try:
-            if len(route_ids) == 0:
-                return []
-            ###
-            payload = { k: v for k, v in {
-                "channel": "web_client",
-                "size":300,
-                "only_online_trip": True,
-                "from_time": from_time,
-                "to_time": to_time,
-                "route_ids": route_ids,
-                "ticket_count": ticket_count,
-                "sort_by": ["price", "departure_time"]
-                }.items() if v is not None
-            }
-
-            data = []       
-            response = await api.post("/search/trips", payload=payload)
-            if response.get("status") == 200:
-                if response.get("data"):
-                    if response["data"].get("total") > 0:
-                        data = response["data"]["items"]
-            return data
-        except Exception as e:
-            print(e)
-            raise Exception("Error fetching trip data")
-        
-    # Chuyến đi khớp với thời gian và văn phòng đón trả 
-    @staticmethod
-    def get_trip_by_time_and_office_id(trips: list, time: str ,origin_id: int, dest_id: int) -> dict:
-        if time is None or origin_id is None or dest_id is None:
-            return {}
-
-        for trip in trips:
-            if trip and trip["raw_departure_time"] == time and trip["route"]["origin_hub_office_id"] == origin_id and trip["route"]["dest_hub_office_id"] == dest_id:
-                return trip
-        return {}
-    
-    
-    # Danh sách chuyến đi có thể chọn được theo văn phòng
-    @staticmethod
-    def get_all_trip_by_office(trips: list, origin_id: int = None, dest_id: int = None) -> list:
-        if origin_id is None and dest_id is None:
-            return trips
-        
-        result = []
-        for trip in trips:
-            pickup_points = trip.get("pickup_points", [])
-            pickup_point_ids = {p["OfficeId"] for p in pickup_points if "OfficeId" in p}
-
-            origin_match = (
-                (origin_id is None) or
-                (trip["route"]["origin_hub_office_id"] == origin_id) or
-                (origin_id in pickup_point_ids)
-            )
-
-            dest_match = (
-                (dest_id is None) or
-                (trip["route"]["dest_hub_office_id"] == dest_id) or
-                (dest_id in pickup_point_ids)
-            )
-
-            if origin_match and dest_match:
-                result.append(trip)
-
-        return result if result else trips
-    # Danh sách 4 chuyến đi xung quanh thời gian chỉ định
-    @staticmethod
-    def get_surrounding_trip(trips: list, time: str, num_trip: int = 4) -> list:
-        if not time or not trips or num_trip <= 0:
-            return []
-        
-        time_trips = [ trip["raw_departure_time"] for trip in trips]
-        index = bisect_left(time_trips, time)
-        
-        half = num_trip // 2
-        start = max(0, index - half)
-        end = min(len(trips), index + half + (num_trip % 2))
-        
-        missing = num_trip - (end - start)
-        
-        if missing > 0:
-            extra_start = min(missing, start)
-            start -= extra_start
-            missing -= extra_start
-        
-        if missing > 0:
-            extra_end = min(missing, len(trips) - end)
-            end += extra_end
-        
-        return trips[start:end]
-        
-        
-
-    async def is_valid_select_seat(self, seat: str,route_id: int, trip_id:int, departure_date: str, departure_time: str, kind: str):
-        if seat is None:
-            return False
-        seats = await self.seats_trip(route_id, trip_id, departure_date, departure_time, kind)
-        for seat_data in seats:
-            if seat_data['chair'].lower() == seat.lower() and seat_data['bookStatus'] == 0 :
-                return True
-        return False
-
-    async def search_trip_by_id(self, trip_id: int, from_time: int, to_time: int, route_ids: list[int], ticket_count: int = 1):
-        trip = await self.search_trip(from_time, to_time, route_ids, ticket_count)
-        for item in trip:
-            if trip_id == item["id"]:
-                return item
-        return None
-    @staticmethod
-    def get_trip_by_id(trip_id: int, trips: list):
-        for trip in trips:
-            if trip and trip["id"] == trip_id:
-                return trip
-        return {}
-    
-    @staticmethod
-    async def stops(route_id: int, way_id: int):
-        try:
-            if route_id is None or way_id is None:
-                return []
-            params ={ k: v for k,v in {
-                        "wayId": way_id,
-                        }.items() if v is not None}
-            response = await api.get(api_base=settings.API_BASE_URL_VATO, endpoint=f"/stops/{route_id}", params=params)
-            if not response.get("data"):
-                return []
-            data = response["data"]
-            return data
-        except Exception as e:
-            print(e)
-            raise Exception("Error fetching stops data")
-
-    async def pickup_list(self, route_id: int, way_id: int):
-        try:
-            data = await self.stops(route_id, way_id)
-            if not data:
-                return []
-            
-            pickup_list = []
-
-            for pickup in data:
-                if pickup["type"] == 0 or pickup["type"] == -1:
-                    pickup_list.append(pickup)
-            return pickup_list
-        
-        except Exception as e:
-            print(e)
-            raise Exception("Error fetching pickup list data")
-    
-    async def is_valid_pickup(self, pickup: str, route_id: int, way_id: int):
-        if pickup is None:
-            return False
-        pickup_list = await self.pickup_list(route_id, way_id)
-        for pickup_data in pickup_list:
-            pickup_name: str = pickup_data['name']
-            if pickup_name.lower() == pickup.lower():
-                return True
-        return False
-
-    async def dropoff_list(self, route_id: int, way_id: int):
-        try:
-            data = await self.stops(route_id, way_id)
-            if not data:
-                return []
-            
-            dropoff_list = []
-
-            for dropoff in data:
-                if dropoff["type"] == 1 or ( dropoff["type"] == -1 and dropoff["presentBeforeMinutes"] >= 0 ):
-                    dropoff_list.append(dropoff)
-            return dropoff_list
-        
-        except Exception as e:
-            print(e)
-            raise Exception("Error fetching dropoff list data")
-        
-    async def is_valid_dropoff(self, dropoff: str, route_id: int, way_id: int):
-        if dropoff is None:
-            return False
-        dropoff_list = await self.dropoff_list(route_id, way_id)
-        for dropoff_data in dropoff_list:
-            if dropoff_data['name'] == dropoff:
-                return True
-        return False
-    @staticmethod
-    async def search_pickup_points(origin: str = None, dest: str = None) -> dict:
-        session_id = str(int(datetime.now().timestamp()))
-        params =    {k: v for k, v in {
-                    "origin": origin,
-                    "dest": dest,
-                    "session_id": session_id,
-                    }.items() if v is not None
-                }
-        
-        response = await api.get('/search/metadata/pickup-points', params=params)
-        if response.get("status") == 200:
-            data = response.get("data")
-            return data
-        return {}
-    
-    async def find_id_office_by_name_office(self, office_name: str):
-        data = await self.search_pickup_points(origin=office_name)
-        if data.get("origin"):
-            origins = data["origin"]
-            for origin in origins:
-                if origin.get("group"):
-                    groups = origin["group"]
-                    for group in groups:
-                        if group.get("name"):
-                            name: str = group["name"]
-                            if name.lower() == office_name.lower():
-                                office_id = group["officeId"]
-                                office_id = office_id[0] if isinstance(office_id, tuple) else office_id
-                                return office_id
-        return None
-    
-    async def find_id_provine_by_name_office(self, office_name: str):
-        data = await self.search_pickup_points(origin=office_name)
-        if data.get("origin"):
-            origins = data["origin"]
-            for origin in origins:
-                if origin.get("group"):
-                    groups = origin["group"]
-                    for group in groups:
-                        if group.get("name"):
-                            name: str = group["name"]
-                            if name.lower() == office_name.lower():
-                                province_id = group["provinceId"]
-                                return province_id
-        return None
-
-    async def find_all_origin_by_name_office(self, office_name: str):
-        data = await self.search_pickup_points(origin=office_name)
-        if data.get("origin"):
-            origins = data["origin"]
-            return origins
-        return None
-
-    async def find_id_and_code_provine_by_name_office(self, office_name: str):
-        data = await self.search_pickup_points(origin=office_name)
-        if data.get("origin"):
-            origins = data["origin"]
-            for origin in origins:
-                if origin.get("group"):
-                    groups = origin["group"]
-                    for group in groups:
-                        if group.get("name"):
-                            name: str = group["name"]
-                            if name.lower() == office_name.lower():
-                                province_id = group["provinceId"]
-                                province_code = group["provinceCode"]
-                                province_id = province_id[0] if isinstance(province_id, tuple) else province_id
-                                return province_id, province_code
-        return None, None
-    
-    async def get_origin_city_from_office(self, origin_office: str):
-        data = await self.search_pickup_points(origin=origin_office)
-        if data.get("origin"):
-            origins = data["origin"]
-            for origin in origins:
-                if origin.get("group"):
-                    groups = origin["group"]
-                    for group in groups:
-                        if group.get("name"):
-                            name:str = group["name"]
-                            if name.lower() == origin_office.lower():
-                                return group["provinceName"]
-        return None
-
-    async def get_destination_city_from_office(self, dest_office: str):
-        data = await self.search_pickup_points(dest=dest_office)
-        if data.get("dest"):
-            dests = data["dest"]
-            for dest in dests:
-                if dest.get("group"):
-                    groups = dest["group"]
-                    for group in groups:
-                        if group.get("name"):
-                            name: str = group["name"]
-                            if name.lower() == dest_office.lower():
-                                return group["provinceName"]
-        return None
-
-    async def check_exist_user_info(self, user_id: str = None):
-        try:
-            # response = await api.get(f'/user/{user_id}')
-            # if response.get("status") == 200:
-            #     return True
-            return True
-            
-            # return False
-        except Exception as e:
-            logger.error(f"Error checking user info: {e}")
-            return False
-    
-    async def get_user_info(self, user_id: str = None):
-        try:
-            # response = await api.get(f'/user/{user_id}')
-            # if response.get("status") == 200:
-            #     return response.get("data")
-            user_info = {
-                "user_name": { "name": "Đại", "original": "Đại" },
-                "phone_number": "0987654321",
-                "email": "[email protected]"
-            }
-            
-            return user_info
-            
-            # return None
-        except Exception as e:
-            logger.error(f"Error fetching user info: {e}")
-            return None
-    @staticmethod
-    async def extract_user_name(text: str, ner: NER):
-        if text is None:
-            return None
-        
-        user_name_pred = await ner.predict(text=text, entity_tag="PERSON")
-        
-        if user_name_pred:
-            user_name = user_name_pred[0]
-            
-            if user_name:
-                user_name_resp = { "name": user_name, "original": user_name }
-                return user_name_resp
-        
-        return None
-
-dialog_service: DialogService = DialogService()
-    

app/dialogflow/services/origin_codes.py

@@ -1,14 +0,0 @@
-import json
-
-with open("static/files/origin-codes.json", "r", encoding="utf-8") as file:
-    origin_codes = json.load(file)
-
-def get_origin_id_and_code(origin: str):
-    for x in origin_codes:
-        if x and x.get("name"):
-            if origin == x["name"]:
-                    id = x["id"],
-                    code = x["code"]
-                    id = id[0] if isinstance(id, tuple) else id
-                    return id, code
-    return None, None

app/dto/request.py

@@ -0,0 +1,96 @@
+from pydantic import BaseModel, field_validator
+from typing import Any
+from pydantic import BaseModel, Field
+
+
+from app.dto.trip import TripDTO
+
+
+class LocationRequest(BaseModel):
+    origin_office: str | None = None
+    departure_city: str | None = None
+    dest_office: str | None = None
+    destination_city: str | None = None
+
+
+class TimeRequest(BaseModel):
+    time_select: dict | None = Field(default_factory=dict)
+
+
+class TripQueryRequest(BaseModel):
+    departure_city: str | None = None
+    destination_city: str | None = None
+    origin_office: str | None = None
+    dest_office: str | None = None
+    ticket_number: int | None = None
+    date: dict | None = Field(default_factory=dict)
+    time_select: dict | None = Field(default_factory=dict)
+    id_selected: int | None = None
+
+    @field_validator("ticket_number", "id_selected", mode="before")
+    def validate_ticket_number(cls, v):
+        return int(v) if v else None
+
+    @field_validator("id_selected", mode="before")
+    def validate_selected_pickup(cls, v):
+        return int(v) if v else None
+
+
+class TicketRequest(BaseModel):
+    user_name: str | None = None
+    phone_number: str | None = None
+    email: str | None = None
+    seat: str | None = None
+    pickup: str | None = None
+    dropoff: str | None = None
+    trip_data: TripDTO | None = None
+
+
+class SeatQueryRequest(BaseModel):
+    seat: str | None = None
+    trip_data: TripDTO | None = None
+
+
+class SessionInfo(BaseModel):
+    session: str | None = None
+    parameters: dict[str, Any] = Field(default_factory=dict)
+
+
+class FulfillmentInfo(BaseModel):
+    tag: str | None = None
+
+
+class DialogflowWebhookRequest(BaseModel):
+    text: str | None = None
+    sessionInfo: SessionInfo | None = None
+    fulfillmentInfo: FulfillmentInfo | None = None
+    detectIntentResponseId: str | None = None
+
+    def get_parameters(self) -> dict[str, Any]:
+        return self.sessionInfo.parameters if self.sessionInfo else {}
+
+    def get_session(self) -> str:
+        return (
+            self.sessionInfo.session
+            if self.sessionInfo and self.sessionInfo.session
+            else ""
+        )
+
+    def get_text(self) -> str:
+        return self.text.strip() if self.text else ""
+
+    def get_tag(self) -> str:
+        return (
+            self.fulfillmentInfo.tag
+            if self.fulfillmentInfo and self.fulfillmentInfo.tag
+            else ""
+        )
+
+
+class StopRequest(BaseModel):
+    trip_data: TripDTO | None = None
+    id_selected: int | None = None
+
+    @field_validator("id_selected", mode="before")
+    def validate_selected_pickup(cls, v):
+        return int(v) if v else None

app/dto/response.py

@@ -0,0 +1,148 @@
+from fastapi.responses import JSONResponse
+from pydantic import BaseModel, Field
+from typing import List, Dict, Any, Optional
+
+
+class MessageText(BaseModel):
+    text: list[str] = Field(default_factory=list)
+
+
+class Message(BaseModel):
+    text: MessageText | None = None
+    payload: dict | None = None
+
+
+class FulfillmentResponse(BaseModel):
+    messages: list[Message] = Field(default_factory=list)
+
+
+class SessionInfo(BaseModel):
+    parameters: dict[str, Any] = Field(default_factory=dict)
+
+
+class DialogflowResponse(BaseModel):
+    fulfillment_response: FulfillmentResponse
+    sessionInfo: Optional[SessionInfo] = Field(default_factory=dict)
+
+
+def DialogFlowResponseAPI(
+    text: list[str] = None,
+    payload: dict | None = None,
+    parameters: dict | None = None,
+):
+
+    messages = []
+    if text:
+        messages.append(Message(text=MessageText(text=text)))
+
+    if payload:
+        messages.append(Message(payload=payload))
+
+    response_data = DialogflowResponse(
+        fulfillment_response=FulfillmentResponse(messages=messages),
+        sessionInfo=SessionInfo(parameters=parameters or {}),
+    )
+
+    return JSONResponse(content=response_data.model_dump())
+
+
+def dialogflow_response(
+    text: list[str] | None = None,
+    payload: dict | None = None,
+    parameters: dict | None = None,
+) -> JSONResponse:
+    messages: List[Message] = []
+    if text:
+        messages.append(Message(text=MessageText(text=text)))
+    if payload:
+        messages.append(Message(payload=payload))
+
+    response = DialogflowResponse(
+        fulfillment_response=FulfillmentResponse(messages=messages),
+        sessionInfo=SessionInfo(parameters=parameters or {}),
+    )
+    return JSONResponse(content=response.model_dump(mode="json"))
+
+
+def dialogflow_error(message: str) -> JSONResponse:
+    return dialogflow_response(text=[message])
+
+
+def dialogflow_choice(
+    prompt: str, options: list[str], parameters: dict | None = None
+) -> JSONResponse:
+    chips = {
+        "richContent": [[{"type": "chips", "options": [{"text": o} for o in options]}]]
+    }
+    return dialogflow_response(text=[prompt], payload=chips, parameters=parameters)
+
+
+class DialogflowResponseBuilder:
+    def __init__(self):
+        self._messages: list[Message] = []
+        self._parameters: dict = {}
+
+    def add_text(self, lines: list[str]) -> "DialogflowResponseBuilder":
+        if lines:
+            self._messages.append(Message(text=MessageText(text=lines)))
+        return self
+
+    def add_payload(self, payload: dict) -> "DialogflowResponseBuilder":
+        if payload:
+            self._messages.append(Message(payload=payload))
+        return self
+
+    def add_choice_chips(
+        self,
+        options: list[str] | list[dict],
+        key: str = None,
+        prompt: str | None = None,
+    ) -> "DialogflowResponseBuilder":
+        if options:
+            chips = {
+                "richContent": [
+                    [
+                        {
+                            "type": "chips",
+                            "options": [
+                                (
+                                    {"text": option.get(key, "")}
+                                    if isinstance(option, dict)
+                                    else {"text": option}
+                                )
+                                for option in options
+                            ],
+                        }
+                    ]
+                ]
+            }
+            self.add_payload(chips)
+        if prompt:
+            self.add_text([prompt])
+        return self
+
+    def set_parameters(self, parameters: dict[str, any]) -> "DialogflowResponseBuilder":
+        if parameters:
+            self._parameters = parameters
+        return self
+
+    def build(self) -> JSONResponse:
+        response = DialogflowResponse(
+            fulfillment_response=FulfillmentResponse(messages=self._messages),
+            sessionInfo=SessionInfo(parameters=self._parameters),
+        )
+        return JSONResponse(content=response.model_dump(mode="json"))
+
+
+## Use Builder return (
+#     DialogflowResponseBuilder()
+#     .add_choice_chips(["Xem vé", "Đặt vé"], prompt="Bạn muốn làm gì tiếp?")
+#     .set_parameters({"next_step": "booking"})
+#     .build()
+# )
+
+
+class DialogServiceResult(BaseModel):
+    text: list[str] | None = None
+    options: list[str] | list[dict] | None = None
+    parameters: dict | None = None

app/dto/seat.py

@@ -0,0 +1,17 @@
+from pydantic import BaseModel
+
+
+class SeatDTO(BaseModel):
+    bookStatus: int
+    id: int
+    chair: str
+    rowNo: int
+    columnNo: int
+    discount: float
+    floorNo: int
+    inSelect: int
+    lockChair: int
+    lock: bool
+    originPrice: float
+    price: float
+    promotion: str | None = None

app/dto/ticket.py

@@ -0,0 +1,80 @@
+from pydantic import BaseModel
+
+
+class PassengerDTO(BaseModel):
+    CustName: str
+    CustMobile: str
+    CustEmail: str
+    CustDestinationAddress: str
+
+
+class DropoffDTO(BaseModel):
+    office_id: int
+
+
+class InfoTicketDTO(BaseModel):
+    PickUpName: str
+    PickUpStreet: str
+    OfficePickupId: int
+    DropOffName: str
+    DropOffStreet: str
+    OfficeDropOffId: int
+    Dropoff: DropoffDTO
+    CarBookingId: int
+    DepartureDate: str
+    DepartureTime: str
+    Kind: str
+    WayId: int
+    DestCode: str
+    DestName: str
+    OriginCode: str
+    OriginName: str
+    Price: int
+    RouteId: int
+    RouteName: str
+    NumOfTicket: int
+    SeatIds: list[int]
+    SeatNames: list[str]
+    Passengers: list[PassengerDTO]
+    Channel: str
+
+
+class TicketBookingDTO(BaseModel):
+    CustName: str
+    CustMobile: str
+    CustEmail: str
+    CustDestinationAddress: str
+    CustId: str
+    PickUpName: str
+    PickUpStreet: str
+    OfficePickupId: int
+    DropOffName: str
+    DropOffStreet: str
+    OfficeDropOffId: int
+    Dropoff: DropoffDTO
+    CarBookingId: int
+    DepartureDate: str
+    DepartureTime: str
+    Kind: str
+    WayId: int
+    DestCode: str
+    DestName: str
+    OriginCode: str
+    OriginName: str
+    Price: int
+    RouteId: int
+    RouteName: str
+    NumOfTicket: int
+    SeatIds: list[int]
+    SeatNames: list[str]
+    Passengers: list[PassengerDTO]
+    Channel: str
+    DeviceId: str | None = None
+    EnglishTicket: int
+    SeatDiscounts: list[str]  # assuming it's list of strings; adjust as needed
+    CustAddress: str
+    CustCode: str
+    CustHomeAddress: str
+    CustMobile2: str
+    CustSN: str
+    InfoTicket: list[InfoTicketDTO]

app/dto/trip.py

@@ -0,0 +1,136 @@
+from pydantic import BaseModel, Field, field_validator
+
+
+class TripSearchDTO(BaseModel):
+    channel: str = "web_client"
+    size: int = 300
+    only_online_trip: bool = True
+    ticket_count: int
+    seat_type_id: list = Field(default_factory=list)
+    from_time: int
+    to_time: int
+    route_ids: list[int]
+    origin_office_id: list = Field(default_factory=list)
+    dest_office_id: list = Field(default_factory=list)
+    postion: list = Field(default_factory=list)
+    floor: list = Field(default_factory=list)
+    sort_by: list[str] = ["price", "departure_time"]
+
+
+class TripOptionDTO(BaseModel):
+    trip_id: int
+    route: str
+
+    @field_validator("trip_id", mode="before")
+    def convert_float_to_int(cls, v):
+        return int(v) if v is not None else None
+
+
+class Route(BaseModel):
+    name: str | None = None
+
+
+class TripDTO(BaseModel):
+    id: int | None = None
+    route_id: int | None = None
+    way_id: int | None = None
+    route: Route | None = None
+    raw_departure_date: str | None = None
+    raw_departure_time: str | None = None
+    seat_type_name: str | None = None
+    price: int | None = None
+
+    @field_validator("id", "price", "route_id", "way_id", mode="before")
+    @classmethod
+    def convert_float_to_int(cls, v):
+        return int(v) if v is not None else None
+
+
+class CoordinateDTO(BaseModel):
+    lat: float
+    lon: float
+
+
+class PickupPointDTO(BaseModel):
+    OfficeId: int | None = None
+    Name: str | None = None
+    Address: str | None = None
+    Phone: str | None = None
+    PickUp: int | None = None
+    TimeOffice: int | None = None
+    PointKind: int | None = None
+    Note: str | None = None
+    PointKindName: str | None = None
+
+
+class RouteDTO(BaseModel):
+    name: str
+    route_id: int
+    origin_code: str
+    origin_name: str
+    origin_hub_id: int
+    origin_hub_name: str
+    origin_hub_office_id: int
+    origin_hub_office_name: str
+    origin_hub_coords: CoordinateDTO
+    dest_code: str
+    dest_name: str
+    dest_hub_id: int
+    dest_hub_name: str
+    dest_hub_office_id: int
+    dest_hub_office_name: str
+    dest_hub_coords: CoordinateDTO
+    shuttle_enable: bool
+    allow_desktop: bool
+    allow_mobile_app: bool
+    allow_web_client: bool
+    allow_web_admin: bool
+
+
+class ShuttleZoneResultDTO(BaseModel):
+    is_in_zone: bool
+    is_allow_user_to_toggle: bool
+    distance_to_user: float
+    zone: str | None = None
+    nearest_pickup_point: str | None = None
+
+
+class ShuttleOptionDTO(BaseModel):
+    is_enable_shuttle: bool
+    route: RouteDTO
+    origin_result: ShuttleZoneResultDTO
+    dest_result: ShuttleZoneResultDTO
+
+
+class TripItemDTO(BaseModel):
+    id: int
+    departure_time: int
+    raw_departure_time: str
+    raw_departure_date: str
+    arrival_time: int
+    duration: int
+    seat_type_id: int
+    seat_type_name: str
+    price: int
+    empty_seat_quantity: int
+    route_id: int
+    distance: int
+    route: RouteDTO
+    way_id: int
+    allow_online_booking: bool
+    online_booking_before: int
+    max_seats_per_booking: int
+    point_details: str | None = None
+    num_price: int
+    from_bus_station_address: str
+    to_bus_station_address: str
+    top_first_floor_quantity: int
+    middle_first_floor_quantity: int
+    last_first_floor_quantity: int
+    top_second_floor_quantity: int
+    middle_second_floor_quantity: int
+    last_second_floor_quantity: int
+    way_name: str
+    way_note: str
+    pickup_points: list[PickupPointDTO]
+    shuttle_option: ShuttleOptionDTO

app/dto/user.py

@@ -0,0 +1,7 @@
+from pydantic import BaseModel
+
+
+class UserDTO(BaseModel):
+    user_name: str
+    phone_number: str
+    email: str

app/ner/services/ner.py

@@ -1,11 +1,15 @@
+from fastapi import Request
+from loguru import logger
 from core.conf import settings
 
+
 class NER:
     def __init__(self, model_dir: str = settings.NER_MODEL_DIR):
         self.model_dir = model_dir
         self.model = None
         self.tokenizer = None
         self.pipeline = None
+        self.load_model()
 
     def load_model(self):
         from transformers import AutoTokenizer
@@ -13,68 +17,79 @@ class NER:
         from optimum.pipelines import pipeline
 
         self.tokenizer = AutoTokenizer.from_pretrained(
-                                                    self.model_dir,
-                                                    truncation=settings.TRUNCATE, 
-                                                    max_length=settings.MAX_LENGTH
-                                                    )
+            self.model_dir, truncation=settings.TRUNCATE, max_length=settings.MAX_LENGTH
+        )
         self.model = ORTModelForTokenClassification.from_pretrained(self.model_dir)
-        self.pipeline = pipeline(task=settings.TASK_NAME, 
-                                 model=self.model, 
-                                 tokenizer=self.tokenizer,
-                                 device=settings.DEVICE)
-        
+        self.pipeline = pipeline(
+            task=settings.TASK_NAME,
+            model=self.model,
+            tokenizer=self.tokenizer,
+            device=settings.DEVICE,
+        )
+        logger.info(f"Model loaded from {self.model_dir}")
+
     async def predict(self, text: str, entity_tag: str = None):
+
+        if not text:
+            return None
+
         if self.pipeline is None:
             raise ValueError("Model not loaded. Please call load_model() first.")
-        
+
         pred = self.pipeline(text)
-        
+
         if entity_tag:
             return self.extract_entities(pred, entity_tag)
         return pred
-    
-    def extract_entities(self, result_pred: list[dict[str, any]], entity: str) -> list[str]:
+
+    def extract_entities(
+        self, result_pred: list[dict[str, any]], entity: str
+    ) -> list[str]:
         if self.pipeline is None:
             raise ValueError("Model not loaded. Please call load_model() first.")
         B_ENTITY = f"B-{entity}"
         I_ENTITY = f"I-{entity}"
-        
+
         extracted_entities = []
         current_entity_tokens = []
-        
+
         for item in result_pred:
             word = item["word"]
             entity_tag = item["entity"]
-            
+
             if entity_tag == B_ENTITY:
                 if current_entity_tokens:
-                    extracted_entities.append(self._combine_token(current_entity_tokens))
+                    extracted_entities.append(
+                        self._combine_token(current_entity_tokens)
+                    )
                 current_entity_tokens = [word]
             elif entity_tag == I_ENTITY and current_entity_tokens:
                 current_entity_tokens.append(word)
             else:
                 if current_entity_tokens:
-                    extracted_entities.append(self._combine_token(current_entity_tokens))
+                    extracted_entities.append(
+                        self._combine_token(current_entity_tokens)
+                    )
                     current_entity_tokens = []
-                    
+
         if current_entity_tokens:
             extracted_entities.append(self._combine_token(current_entity_tokens))
-        
+
         return extracted_entities
 
     def _combine_token(self, tokens: list[str]) -> str:
         """Combines tokens into a single string, removing leading hashtags from the first token if present.
         Args:
             tokens (list[str]): List of tokens to combine.
-            
+
         Returns:
             str: Combined string of tokens.
         """
         if not tokens:
             return ""
-        
+
         words = []
-        
+
         for token in tokens:
             if token.strip("#") != token:
                 clean_token = token.strip("#")
@@ -84,5 +99,13 @@ class NER:
                     words.append(clean_token)
             else:
                 words.append(token)
-        
+
         return " ".join(words)
+
+
+def get_ner_model(request: Request) -> NER:
+    """
+    Dependency to get the NER model.
+    This can be used to inject the NER model into the endpoint.
+    """
+    return request.app.state.ner

app/repositories/trip_repository.py

@@ -0,0 +1,88 @@
+from typing import Annotated
+from fastapi import Depends
+from typing_extensions import Literal
+from app.client.futabus_client import FutabusClient
+
+
+class TripRepository:
+
+    def __init__(self, client: FutabusClient):
+        self.client = client
+
+    async def get_all_route_ids(
+        self,
+        origin_code: str | None = None,
+        from_id: int | None = None,
+        origin_ids: int | None = None,
+        dest_code: str | None = None,
+        to_id: int | None = None,
+        dest_ids: int | None = None,
+    ) -> list[int]:
+        """
+        Lấy danh sách route_ids từ hệ thống external FUTABUS
+
+        Returns:
+            List[int]: Danh sách ID tuyến xe.
+        """
+
+        route_ids = await self.client.get_route_ids_by_metadata(
+            origin_code=origin_code,
+            from_id=from_id,
+            origin_ids=origin_ids,
+            dest_code=dest_code,
+            to_id=to_id,
+            dest_ids=dest_ids,
+        )
+
+        if not route_ids:
+            route_ids = await self.client.get_route_ids_by_code(
+                origin_code=origin_code,
+                dest_code=dest_code,
+            )
+        return route_ids
+
+    async def get_trip_list(
+        self, from_time: int, to_time: int, route_ids: list[int], ticket_count: int = 1
+    ) -> list:
+        return await self.client.search_trips(
+            from_time=from_time,
+            to_time=to_time,
+            route_ids=route_ids,
+            ticket_count=ticket_count,
+        )
+
+    async def get_stop(
+        self, route_id: int, way_id: int, role: Literal["pick-up", "drop-off"]
+    ) -> list:
+        """
+        Lấy danh sách các điểm đón trả khách của một tuyến xe cụ thể.
+
+        Args:
+            route_id (int): ID của tuyến xe.
+            way_id (int): ID của chiều đi hoặc về của tuyến xe.
+            role (Literal["pick-up", "drop-off"]): Vai trò của điểm dừng, có thể là "pick-up" hoặc "drop-off".
+        Returns:
+            list: Danh sách các điểm đón trả khách.
+        """
+        stops = await self.client.get_stop_by_route_and_way(
+            route_id=route_id, way_id=way_id
+        )
+        if role == "pick-up":
+            pickups = []
+
+            for stop in stops:
+                if stop["type"] == 0 or stop["type"] == -1:
+                    pickups.append(stop)
+            return pickups
+
+        if role == "drop-off":
+            dropoffs = []
+
+            for stop in stops:
+                if stop["type"] == 1 or (
+                    stop["type"] == -1 and stop["presentBeforeMinutes"] >= 0
+                ):
+                    dropoffs.append(stop)
+            return dropoffs
+        else:
+            raise ValueError("Role must be either 'pick-up' or 'drop-off'")

app/resolvers/location_resolver.py

@@ -0,0 +1,27 @@
+from typing import Literal, Optional
+from fastapi.params import Depends
+from typing_extensions import Annotated
+from app.services.location_service import LocationService
+from app.resolvers.origin_code import get_city_id_and_code
+
+
+class LocationResolver:
+    def __init__(self, location_service: LocationService):
+        self.location_service = location_service
+
+    async def resolve_info(
+        self,
+        office: Optional[str],
+        city: Optional[str],
+        role: Literal["origin", "dest"],
+    ) -> tuple[Optional[int], Optional[str], Optional[int]]:
+        if office:
+            id, code, _ = await self.location_service.get_city_by_office(
+                office_name=office, role=role
+            )
+            ids = await self.location_service.get_office_id_by_name(office)
+            return id, code, ids
+        if city:
+            id, code = get_city_id_and_code(city)
+            return id, code, None
+        return None, None, None

app/resolvers/origin_code.py

@@ -0,0 +1,37 @@
+import json
+import logging
+import os
+from typing import Optional, Tuple
+
+
+logger = logging.getLogger(__name__)
+
+ORIGIN_CODES_PATH = "static/files/origin-codes.json"
+
+
+def load_origin_codes() -> list[dict]:
+    try:
+        with open(ORIGIN_CODES_PATH, "r", encoding="utf-8") as file:
+            return json.load(file)
+    except (FileNotFoundError, json.JSONDecodeError):
+        logger.error(f"Error loading origin codes from {ORIGIN_CODES_PATH}")
+        return []
+
+
+def get_city_id_and_code(origin_name: str) -> Tuple[Optional[int], Optional[str]]:
+    """
+    Truy xuất province_id và province_code từ file JSON tĩnh theo tên tỉnh/thành phố.
+
+    Args:
+        origin_name (str): Tên thành phố gốc
+
+    Returns:
+        Tuple[province_id, province_code] hoặc (None, None) nếu không tìm thấy
+    """
+    origin_name = origin_name.strip().lower()
+    for entry in load_origin_codes():
+        name = entry.get("name", "").lower()
+        if name == origin_name:
+            return entry.get("id"), entry.get("code")
+    logger.warning(f"ID and Code not found for name: {origin_name}")
+    return None, None

app/router.py

@@ -1,9 +0,0 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-from fastapi import APIRouter
-
-from app.dialogflow.api.router import v1 as dialogflow_v1
-
-router = APIRouter()
-
-router.include_router(dialogflow_v1)

app/services/dialog_service.py

@@ -0,0 +1,537 @@
+# from bisect import bisect_left
+# from datetime import datetime, timedelta
+
+# from fastapi import logger
+
+# from app.ner.services.ner import NER
+
+# from common.external.external_api import api
+# from core.conf import settings
+
+
+# class DialogService:
+#     @staticmethod
+#     def to_datetime_from_Dialogflow(time: dict):
+#         date_time = datetime(int(time["year"]), int(time["month"]), int(time["day"]))
+#         return date_time
+
+#     @staticmethod
+#     def process_dates_to_timestamp(
+#         from_time: datetime = None, to_time: datetime = None
+#     ):
+#         if to_time is None and from_time is not None:
+#             to_time = from_time.replace(hour=23, minute=59, second=59)
+
+#         if from_time is None:
+#             today = datetime.today().date()
+#             from_time = datetime.combine(today, datetime.min.time())
+#             to_time = datetime.combine(today, datetime.max.time()) - timedelta(
+#                 microseconds=1
+#             )
+
+#         return int(from_time.timestamp()) * 1000, int(to_time.timestamp()) * 1000
+
+#     def get_param_from_dialogflow(self, body: any):
+#         session_info = body.get("sessionInfo", {})
+#         parameters = (
+#             session_info.get("parameters", {})
+#             if isinstance(session_info.get("parameters"), dict)
+#             else {}
+#         )
+#         raw_date = parameters.get("date")
+#         if raw_date is not None:
+#             raw_date = self.to_datetime_from_Dialogflow(raw_date)
+#         raw_departure_city = parameters.get("departure_city")
+#         raw_destination_city = parameters.get("destination_city")
+#         raw_ticket_number = parameters.get("ticket_number")
+#         raw_time_of_day = parameters.get("time_of_day")
+#         return (
+#             raw_departure_city,
+#             raw_destination_city,
+#             raw_ticket_number,
+#             raw_date,
+#             raw_time_of_day,
+#         )
+
+#     # Danh sách route_id API v1
+#     @staticmethod
+#     async def search_route_ids_one_way_by_metadata(
+#         origin_code: str = None,
+#         from_id: int = None,
+#         orign_ids: int = None,
+#         dest_code: str = None,
+#         to_id: str = None,
+#         dest_ids: str = None,
+#     ):
+#         params = {
+#             k: v
+#             for k, v in {
+#                 "OriginCode": origin_code,
+#                 "FromId": from_id,
+#                 "OriginIds": orign_ids,
+#                 "DestCode": dest_code,
+#                 "ToId": to_id,
+#                 "DestIds": dest_ids,
+#             }.items()
+#             if v is not None
+#         }
+#         response = await api.get(f"/metadata/office/routes", params=params)
+#         route_ids = []
+#         if isinstance(response, list):
+#             route_ids = [route.get("routeId", None) for route in response]
+#             return route_ids
+#         return []
+
+#     # Danh sách route_id v2 khi API route v1 không có dữ liệu
+#     @staticmethod
+#     async def search_route_ids_one_way_by_origin_dest_code(origin: str, dest: str):
+#         try:
+#             route_ids = []
+#             response = await api.get(f"/booking/api/v2/routes/from/{origin}/to/{dest}")
+#             if response.get("Status") == 200:
+#                 if response.get("Data"):
+#                     data = response.get("Data")
+#                     route_ids = [route.get("Id", None) for route in data]
+#             return route_ids
+#         except Exception as e:
+#             print(e)
+#             raise Exception("Error fetching seats data")
+
+#     async def search_all_route_ids(
+#         self,
+#         origin_code: str = None,
+#         from_id: int = None,
+#         orign_ids: int = None,
+#         dest_code: str = None,
+#         to_id: str = None,
+#         dest_ids: str = None,
+#     ):
+#         route_ids = await self.search_route_ids_one_way_by_metadata(
+#             origin_code, from_id, orign_ids, dest_code, to_id, dest_ids
+#         )
+#         if len(route_ids) == 0:
+#             route_ids = await self.search_route_ids_one_way_by_origin_dest_code(
+#                 origin_code, dest_code
+#             )
+#         return route_ids
+
+#     # Danh sách ghế
+#     @staticmethod
+#     async def seats_trip(
+#         route_id: int, trip_id: int, departure_date: str, departure_time: str, kind: str
+#     ):
+#         try:
+#             params = {
+#                 k: v
+#                 for k, v in {
+#                     "departureDate": departure_date,
+#                     "departureTime": departure_time,
+#                     "kind": kind,
+#                 }.items()
+#                 if v is not None
+#             }
+#             if params is None:
+#                 return []
+#             response = await api.get(
+#                 api_base=settings.API_BASE_URL_VATO,
+#                 endpoint=f"/seats/{route_id}/{trip_id}",
+#                 params=params,
+#             )
+#             if not response.get("data"):
+#                 return []
+#             seats = response["data"]
+#             return seats
+#         except Exception as e:
+#             print(e)
+#             raise Exception("Error fetching seats data")
+
+#     # Call API để lấy danh sách chuyến đi
+#     @staticmethod
+#     async def search_trip(
+#         from_time: int, to_time: int, route_ids: list[int], ticket_count: int = 1
+#     ):
+#         try:
+#             if len(route_ids) == 0:
+#                 return []
+#             ###
+#             payload = {
+#                 k: v
+#                 for k, v in {
+#                     "channel": "web_client",
+#                     "size": 300,
+#                     "only_online_trip": True,
+#                     "from_time": from_time,
+#                     "to_time": to_time,
+#                     "route_ids": route_ids,
+#                     "ticket_count": ticket_count,
+#                     "sort_by": ["price", "departure_time"],
+#                 }.items()
+#                 if v is not None
+#             }
+
+#             data = []
+#             response = await api.post("/search/trips", payload=payload)
+#             if response.get("status") == 200:
+#                 if response.get("data"):
+#                     if response["data"].get("total") > 0:
+#                         data = response["data"]["items"]
+#             return data
+#         except Exception as e:
+#             print(e)
+#             raise Exception("Error fetching trip data")
+
+#     # Chuyến đi khớp với thời gian và văn phòng đón trả
+#     @staticmethod
+#     def get_trip_by_time_and_office_id(
+#         trips: list, time: str, origin_id: int, dest_id: int
+#     ) -> dict:
+#         if time is None or origin_id is None or dest_id is None:
+#             return {}
+
+#         for trip in trips:
+#             if (
+#                 trip
+#                 and trip["raw_departure_time"] == time
+#                 and trip["route"]["origin_hub_office_id"] == origin_id
+#                 and trip["route"]["dest_hub_office_id"] == dest_id
+#             ):
+#                 return trip
+#         return {}
+
+#     # Danh sách chuyến đi có thể chọn được theo văn phòng
+#     @staticmethod
+#     def get_all_trip_by_office(
+#         trips: list, origin_id: int = None, dest_id: int = None
+#     ) -> list:
+#         if origin_id is None and dest_id is None:
+#             return trips
+
+#         result = []
+#         for trip in trips:
+#             pickup_points = trip.get("pickup_points", [])
+#             pickup_point_ids = {p["OfficeId"] for p in pickup_points if "OfficeId" in p}
+
+#             origin_match = (
+#                 (origin_id is None)
+#                 or (trip["route"]["origin_hub_office_id"] == origin_id)
+#                 or (origin_id in pickup_point_ids)
+#             )
+
+#             dest_match = (
+#                 (dest_id is None)
+#                 or (trip["route"]["dest_hub_office_id"] == dest_id)
+#                 or (dest_id in pickup_point_ids)
+#             )
+
+#             if origin_match and dest_match:
+#                 result.append(trip)
+
+#         return result if result else trips
+
+#     # Danh sách 4 chuyến đi xung quanh thời gian chỉ định
+#     @staticmethod
+#     def get_surrounding_trip(trips: list, time: str, num_trip: int = 4) -> list:
+#         if not time or not trips or num_trip <= 0:
+#             return []
+
+#         time_trips = [trip["raw_departure_time"] for trip in trips]
+#         index = bisect_left(time_trips, time)
+
+#         half = num_trip // 2
+#         start = max(0, index - half)
+#         end = min(len(trips), index + half + (num_trip % 2))
+
+#         missing = num_trip - (end - start)
+
+#         if missing > 0:
+#             extra_start = min(missing, start)
+#             start -= extra_start
+#             missing -= extra_start
+
+#         if missing > 0:
+#             extra_end = min(missing, len(trips) - end)
+#             end += extra_end
+
+#         return trips[start:end]
+
+#     async def is_valid_select_seat(
+#         self,
+#         seat: str,
+#         route_id: int,
+#         trip_id: int,
+#         departure_date: str,
+#         departure_time: str,
+#         kind: str,
+#     ):
+#         if seat is None:
+#             return False
+#         seats = await self.seats_trip(
+#             route_id, trip_id, departure_date, departure_time, kind
+#         )
+#         for seat_data in seats:
+#             if (
+#                 seat_data["chair"].lower() == seat.lower()
+#                 and seat_data["bookStatus"] == 0
+#             ):
+#                 return True
+#         return False
+
+#     async def search_trip_by_id(
+#         self,
+#         trip_id: int,
+#         from_time: int,
+#         to_time: int,
+#         route_ids: list[int],
+#         ticket_count: int = 1,
+#     ):
+#         trip = await self.search_trip(from_time, to_time, route_ids, ticket_count)
+#         for item in trip:
+#             if trip_id == item["id"]:
+#                 return item
+#         return None
+
+#     @staticmethod
+#     def get_trip_by_id(trip_id: int, trips: list):
+#         for trip in trips:
+#             if trip and trip["id"] == trip_id:
+#                 return trip
+#         return {}
+
+#     @staticmethod
+#     async def stops(route_id: int, way_id: int):
+#         try:
+#             if route_id is None or way_id is None:
+#                 return []
+#             params = {
+#                 k: v
+#                 for k, v in {
+#                     "wayId": way_id,
+#                 }.items()
+#                 if v is not None
+#             }
+#             response = await api.get(
+#                 api_base=settings.API_BASE_URL_VATO,
+#                 endpoint=f"/stops/{route_id}",
+#                 params=params,
+#             )
+#             if not response.get("data"):
+#                 return []
+#             data = response["data"]
+#             return data
+#         except Exception as e:
+#             print(e)
+#             raise Exception("Error fetching stops data")
+
+#     async def pickup_list(self, route_id: int, way_id: int):
+#         try:
+#             data = await self.stops(route_id, way_id)
+#             if not data:
+#                 return []
+
+#             pickup_list = []
+
+#             for pickup in data:
+#                 if pickup["type"] == 0 or pickup["type"] == -1:
+#                     pickup_list.append(pickup)
+#             return pickup_list
+
+#         except Exception as e:
+#             print(e)
+#             raise Exception("Error fetching pickup list data")
+
+#     async def is_valid_pickup(self, pickup: str, route_id: int, way_id: int):
+#         if pickup is None:
+#             return False
+#         pickup_list = await self.pickup_list(route_id, way_id)
+#         for pickup_data in pickup_list:
+#             pickup_name: str = pickup_data["name"]
+#             if pickup_name.lower() == pickup.lower():
+#                 return True
+#         return False
+
+#     async def dropoff_list(self, route_id: int, way_id: int):
+#         try:
+#             data = await self.stops(route_id, way_id)
+#             if not data:
+#                 return []
+
+#             dropoff_list = []
+
+#             for dropoff in data:
+#                 if dropoff["type"] == 1 or (
+#                     dropoff["type"] == -1 and dropoff["presentBeforeMinutes"] >= 0
+#                 ):
+#                     dropoff_list.append(dropoff)
+#             return dropoff_list
+
+#         except Exception as e:
+#             print(e)
+#             raise Exception("Error fetching dropoff list data")
+
+#     async def is_valid_dropoff(self, dropoff: str, route_id: int, way_id: int):
+#         if dropoff is None:
+#             return False
+#         dropoff_list = await self.dropoff_list(route_id, way_id)
+#         for dropoff_data in dropoff_list:
+#             if dropoff_data["name"] == dropoff:
+#                 return True
+#         return False
+
+#     @staticmethod
+#     async def search_pickup_points(origin: str = None, dest: str = None) -> dict:
+#         session_id = str(int(datetime.now().timestamp()))
+#         params = {
+#             k: v
+#             for k, v in {
+#                 "origin": origin,
+#                 "dest": dest,
+#                 "session_id": session_id,
+#             }.items()
+#             if v is not None
+#         }
+
+#         response = await api.get("/search/metadata/pickup-points", params=params)
+#         if response.get("status") == 200:
+#             data = response.get("data")
+#             return data
+#         return {}
+
+#     async def find_id_office_by_name_office(self, office_name: str):
+#         data = await self.search_pickup_points(origin=office_name)
+#         if data.get("origin"):
+#             origins = data["origin"]
+#             for origin in origins:
+#                 if origin.get("group"):
+#                     groups = origin["group"]
+#                     for group in groups:
+#                         if group.get("name"):
+#                             name: str = group["name"]
+#                             if name.lower() == office_name.lower():
+#                                 office_id = group["officeId"]
+#                                 office_id = (
+#                                     office_id[0]
+#                                     if isinstance(office_id, tuple)
+#                                     else office_id
+#                                 )
+#                                 return office_id
+#         return None
+
+#     async def find_id_provine_by_name_office(self, office_name: str):
+#         data = await self.search_pickup_points(origin=office_name)
+#         if data.get("origin"):
+#             origins = data["origin"]
+#             for origin in origins:
+#                 if origin.get("group"):
+#                     groups = origin["group"]
+#                     for group in groups:
+#                         if group.get("name"):
+#                             name: str = group["name"]
+#                             if name.lower() == office_name.lower():
+#                                 province_id = group["provinceId"]
+#                                 return province_id
+#         return None
+
+#     async def find_all_origin_by_name_office(self, office_name: str):
+#         data = await self.search_pickup_points(origin=office_name)
+#         if data.get("origin"):
+#             origins = data["origin"]
+#             return origins
+#         return None
+
+#     async def find_id_and_code_provine_by_name_office(self, office_name: str):
+#         data = await self.search_pickup_points(origin=office_name)
+#         if data.get("origin"):
+#             origins = data["origin"]
+#             for origin in origins:
+#                 if origin.get("group"):
+#                     groups = origin["group"]
+#                     for group in groups:
+#                         if group.get("name"):
+#                             name: str = group["name"]
+#                             if name.lower() == office_name.lower():
+#                                 province_id = group["provinceId"]
+#                                 province_code = group["provinceCode"]
+#                                 province_id = (
+#                                     province_id[0]
+#                                     if isinstance(province_id, tuple)
+#                                     else province_id
+#                                 )
+#                                 return province_id, province_code
+#         return None, None
+
+#     async def get_origin_city_from_office(self, origin_office: str):
+#         data = await self.search_pickup_points(origin=origin_office)
+#         if data.get("origin"):
+#             origins = data["origin"]
+#             for origin in origins:
+#                 if origin.get("group"):
+#                     groups = origin["group"]
+#                     for group in groups:
+#                         if group.get("name"):
+#                             name: str = group["name"]
+#                             if name.lower() == origin_office.lower():
+#                                 return group["provinceName"]
+#         return None
+
+#     async def get_destination_city_from_office(self, dest_office: str):
+#         data = await self.search_pickup_points(dest=dest_office)
+#         if data.get("dest"):
+#             dests = data["dest"]
+#             for dest in dests:
+#                 if dest.get("group"):
+#                     groups = dest["group"]
+#                     for group in groups:
+#                         if group.get("name"):
+#                             name: str = group["name"]
+#                             if name.lower() == dest_office.lower():
+#                                 return group["provinceName"]
+#         return None
+
+#     async def check_exist_user_info(self, user_id: str = None):
+#         try:
+#             # response = await api.get(f'/user/{user_id}')
+#             # if response.get("status") == 200:
+#             #     return True
+#             return True
+
+#             # return False
+#         except Exception as e:
+#             logger.error(f"Error checking user info: {e}")
+#             return False
+
+#     async def get_user_info(self, user_id: str = None):
+#         try:
+#             # response = await api.get(f'/user/{user_id}')
+#             # if response.get("status") == 200:
+#             #     return response.get("data")
+#             user_info = {
+#                 "user_name": {"name": "Đại", "original": "Đại"},
+#                 "phone_number": "0987654321",
+#                 "email": "[email protected]",
+#             }
+
+#             return user_info
+
+#             # return None
+#         except Exception as e:
+#             logger.error(f"Error fetching user info: {e}")
+#             return None
+
+#     @staticmethod
+#     async def extract_user_name(text: str, ner: NER):
+#         if text is None:
+#             return None
+
+#         user_name_pred = await ner.predict(text=text, entity_tag="PERSON")
+
+#         if user_name_pred:
+#             user_name = user_name_pred[0]
+
+#             if user_name:
+#                 user_name_resp = {"name": user_name, "original": user_name}
+#                 return user_name_resp
+
+#         return None
+
+
+# dialog_service: DialogService = DialogService()

app/services/helper.py

@@ -0,0 +1,97 @@
+from datetime import datetime
+
+
+def extra_time_dialogflow(time: dict) -> list | str | None:
+    """
+    Format time dictionary to string.
+    Nếu time có ambiguous (past, future) => trả về list [past_time_str, future_time_str]
+    Nếu time rõ ràng => trả về string "HH:mm"
+    """
+
+    def is_time_ambiguous(time_obj: dict) -> bool:
+        return any(time_obj.get(key) for key in ["past", "future", "partial"])
+
+    if time is None:
+        return None
+
+    if is_time_ambiguous(time):
+        past_time = time.get("past", {})
+        future_time = time.get("future", {})
+
+        past_hours = int(past_time.get("hours", 0))
+        past_minutes = int(past_time.get("minutes", 0))
+        past_time_str = f"{past_hours:02d}:{past_minutes:02d}"
+
+        future_hours = int(future_time.get("hours", 0))
+        future_minutes = int(future_time.get("minutes", 0))
+        future_time_str = f"{future_hours:02d}:{future_minutes:02d}"
+
+        return sorted([past_time_str, future_time_str])
+
+    else:
+        hours = int(time["hours"])
+        minutes = int(time["minutes"])
+        return f"{hours:02d}:{minutes:02d}"
+
+
+def get_time_range(from_time: datetime = None, to_time: datetime = None):
+    """
+    Chuyển thời gian từ Dialogflow về dạng timestap cho search trip.
+    Thông thường sẽ truyền vào ngày và trả về thời gian đầu và cuối ngày
+
+    Args:
+        from_time (datetime): Thời gian bắt đầu
+        to_time (datetime): Thời gian kết thúc
+    Returns:
+        from_time (timestamp)
+        to_time (timestamp)
+    """
+    if to_time is None and from_time is not None:
+        to_time = from_time.replace(hour=23, minute=59, second=59)
+
+    if from_time is None:
+        today = datetime.today().date()
+        from_time = datetime.combine(today, datetime.min.time())
+        to_time = datetime.combine(today, datetime.max.time()) - timedelta(
+            microseconds=1
+        )
+
+    return int(from_time.timestamp()) * 1000, int(to_time.timestamp()) * 1000
+
+
+def to_datetime_from_Dialogflow(raw_date: dict) -> datetime:
+    """
+    Convert raw date from Dialogflow to datetime object.
+    """
+    if not raw_date:
+        return None
+
+    year = int(raw_date.get("year", 0))
+    month = int(raw_date.get("month", 1))
+    day = int(raw_date.get("day", 1))
+
+    return datetime(year, month, day)
+
+
+def get_weekday_name(date: datetime) -> tuple:
+    """
+    Get the name of the weekday from its number.
+    :param weekday: Weekday number (0-6).
+    :return: Name of the weekday.
+    """
+    vietnam_weekdays = {
+        0: "Thứ hai",
+        1: "Thứ ba",
+        2: "Thứ tư",
+        3: "Thứ năm",
+        4: "Thứ sáu",
+        5: "Thứ bảy",
+        6: "Chủ nhật",
+    }
+
+    weekday_ids = date.weekday()
+    weekday = (
+        vietnam_weekdays.get(weekday_ids) if vietnam_weekdays.get(weekday_ids) else ""
+    )
+
+    return date.strftime("%d-%m-%Y"), weekday