visualize_amr_corpus.py

from argparse import ArgumentParser

import penman

from vulcan.server.basic_layout import BasicLayout
from vulcan.server.server import Server
from vulcan.data_handling.data_corpus import from_dict_list
from vulcan.data_handling.format_names import FORMAT_NAME_STRING, FORMAT_NAME_GRAPH


def main():
    parser = ArgumentParser()
    parser.add_argument("corpus_filename", help="Path to the AMR corpus file you want to visualize.")
    parser.add_argument("-p", "--port", type=int, action="store", dest="port", default=5050,
                        help="Specify the port to use for this visualization.")
    parser.add_argument("-pf", "--propbank-frames", type=str,
                        action="store", dest="propbank_frames", default=None,
                        help="Path to a folder containing XML files with Propbank frames, "
                             "such as data/frames/propbank-amr-frames-xml-2018-01-25 in the AMR3.0 corpus. "
                             "If given, vulcan will show frame definitions on node mouseover ")
    parser.add_argument("-wiki", "--show-wikipedia-articles", action="store_true", dest="show_wikipedia_articles",
                        default=False, help="If given, vulcan will show Wikipedia articles on node mouseover."
                                            "Can take quite a while to load in the beginning!")
    args = parser.parse_args()

    penman_corpus = penman.load(args.corpus_filename)

    # print(penman_corpus[0].metadata.keys())
    # print(penman_corpus[0].metadata["snt"])

    graphs = []
    sentences = []
    for graph in penman_corpus:
        graphs.append(graph)
        if "snt" in graph.metadata:
            sentences.append(graph.metadata["snt"])
        else:
            sentences.append("SENTENCE_NOT_FOUND_IN_CORPUS")

    input_dicts = [
        {
            "type": "data",
            "name": "graphs",
            "instances": graphs,
            "format": FORMAT_NAME_GRAPH
        },
        {
            "type": "data",
            "name": "sentences",
            "instances": sentences,
            "format": FORMAT_NAME_STRING
        },
    ]

    data_corpus = from_dict_list(input_dicts, propbank_frames_path=args.propbank_frames,
                                 show_wikipedia=args.show_wikipedia_articles)

    layout = BasicLayout(data_corpus.slices.values(), data_corpus.linkers, data_corpus.size)

    # print(layout.layout)

    server = Server(layout, port=args.port)
    server.start()  # at this point, the server is running on this thread, and nothing below will be executed


if __name__ == '__main__':
    main()