import numpy as np
import gradio as gr
from src.utils import Utils
from swiplserver import PrologMQI
import plotly.graph_objects as go

facts_path = "./src/facts.pl"
search_algorithm_path = "./src/search_algorithms.pl"

utils = Utils()
possible_routes = []

def query(origin, destination):
    with PrologMQI() as mqi:
        with mqi.create_thread() as prolog_thread:
            result = prolog_thread.query(f'consult("{facts_path}").')
            print(f"Loaded facts: {result}")
            result = prolog_thread.query(f'consult("{search_algorithm_path}").')
            print(f"Loaded rules: {result}")
            print(f"Query Path: {origin} - - to - - {destination}")
            prolog_thread.query_async(f"a_star_search({utils.remove_spl_chars(origin)}, {utils.remove_spl_chars(destination)}, Path).", find_all=False)
            result = prolog_thread.query_async_result()
            if (result == False) or (len(result) == 0):
                return []
            print(f"Total Results: {len(result)}")
            routes = result[0]['Path']
            print(f"Result route: {routes}")
            return routes

def load_map():
    center_lat, center_lon = utils.get_map_center()
    fig = go.Figure(layout=utils.get_default_map_layout(center_lat, center_lon))
    fig.add_traces(utils.get_traces())
    fig.add_trace(utils.get_markers())
    fig = utils.filter_duplicate_traces(fig)
    return fig

def filter_map(origin, destination):
    center_lat, center_lon = utils.get_map_center()
    output_text = ""    
    traces = []
    if (len(origin) > 0) and (len(destination) > 0):
        route = query(origin, destination)
        if len(route) == 0:
            output_text = "No Route Found"
        else:
            traces, center_lat, center_lon, output_text = utils.filter_marker_and_traces(route)
    else:
        traces = utils.get_traces()

    fig = go.Figure(data=traces,layout=utils.get_default_map_layout(center_lat, center_lon))
    fig = utils.filter_duplicate_traces(fig)
    return fig, output_text

with gr.Blocks() as demo:
    with gr.Column():
        gr.Label(value="London Underground Stations Route Finder", label="")
        map = gr.Plot(label="London Underground Station Map")
        with gr.Row():
            origin = gr.Dropdown(
                choices=utils.get_all_station_names(), label="Origin 🔵")
            destination = gr.Dropdown(
                choices=utils.get_all_station_names(), label="Destination 🔴")
        with gr.Row():
            btn_clear = gr.Button(value="Clear")
            btn_search = gr.Button(value="Search")
            btn_clear.click(load_map, [], map)
        output_label = gr.Label(label="Optimal Route")
        btn_search.click(filter_map, [origin, destination], [map, output_label])
    demo.load(load_map, [], map)

demo.launch()