final
This commit is contained in:
40
input-to-route/wrapper.py
Normal file
40
input-to-route/wrapper.py
Normal file
@@ -0,0 +1,40 @@
|
||||
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)
|
||||
Reference in New Issue
Block a user