add

app/dialogflow/api/v1/dialogflow.py

@@ -14,6 +14,40 @@ 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 = session_info.get("departure_city")
+    origin_office = parameters.get("origin_office")
+    
+    if origin_office and raw_departure_city is None:
+        departure_city = 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 = session_info.get("destination_city")
+    dest_office = parameters.get("dest_office")
+    
+    if dest_office and raw_destination_city is None:
+        destination_city = 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()

app/dialogflow/services/dialog_service.py

@@ -155,7 +155,47 @@ class DialogService:
             if dropoff_data['name'] == dropoff:
                 return True
         return False
+    async def search_pickup_points(origin: str = None, dest: str = None):
+        session_id = str(int(datetime.now().timestamp()))
+        params = {
+            "origin": origin,
+            "dest": dest,
+            "session_id": session_id,
+        }
+        response = await api.get('search/metadata/pickup-points', params=params)
+        if response.get("status") == 200:
+            data = response.get("data")
+            return data
+        return None
+    
+    
+    async def get_origin_city_from_office(self, origin_office: str):
+        data = 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 = group["name"]
+                            if name == origin_office:
+                                return group["provinceName"]
+        return None
 
+    async def get_destination_city_from_office(self, dest_office: str):
+        data = 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 = group["name"]
+                            if name == dest_office:
+                                return group["provinceName"]
+        return None
 
 dialog_service: DialogService = DialogService()
     

static/files/code_province.json

@@ -52,6 +52,7 @@
   "Bà Rịa - Vũng Tàu": "VUNGTAU",
   "Hồ Chí Minh": "TPHCM",
   "Thành phố Hồ Chí Minh": "TPHCM",
+  "TP. Hồ Chí Minh": "TPHCM",
   "Long An": "LONGAN",
   "Tiền Giang": "TIENGIANG",
   "Bến Tre": "BENTRE",