41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
from database import search_database
|
|
from geocoder import validate_address
|
|
from new_parser_test import parse_request
|
|
from route import build_route
|
|
from ouput_chat import RouteDescriber
|
|
import json
|
|
|
|
if __name__=='__main__':
|
|
parsed = parse_request(input())
|
|
user_input, conversation_history = parsed['result'], parsed['conversation_history'],
|
|
#user_input = json.loads(user_input)
|
|
print(user_input)
|
|
query =user_input['tags']
|
|
user_address =user_input['user_location']
|
|
user_time =user_input['time']
|
|
val_output = validate_address('addresses.sqlite',user_address)
|
|
print(val_output)
|
|
found_points,mapping = search_database('output.json', query)
|
|
print(len(found_points))
|
|
print(mapping)
|
|
user_position =[]
|
|
if val_output['valid']:
|
|
user_position.append(val_output['coordinates']['lat'])
|
|
user_position.append(val_output['coordinates']['lon'])
|
|
else:
|
|
print('Адрес не найден')
|
|
n_nodes = len(mapping)
|
|
allow_extend = any(v[1] for v in mapping.values())
|
|
print(len(mapping),allow_extend)
|
|
route_otp =0
|
|
if allow_extend:
|
|
while n_nodes <= 5 and (route_otp := build_route(found_points, mapping, user_position, user_time, n_nodes,strategy='random') or route_otp): n_nodes += 1
|
|
else:
|
|
route_otp= build_route(found_points, mapping, user_position, user_time, n_nodes,strategy='random')
|
|
route, places = route_otp
|
|
dscb = RouteDescriber()
|
|
print(conversation_history)
|
|
description = dscb.generate_route_description(places,conversation_history)
|
|
print(route)
|
|
print(description)
|