| import gradio as gr | |
| import pandas as pd | |
| names = pd.read_csv('vertices.txt',delimiter=" ",names=["i","name"]) | |
| relations = pd.read_csv('arcs.txt',delimiter=" ",names=["source","target","times"]) | |
| def get_targets_from_source(source): | |
| return relations[relations.source==source].target.values | |
| def get_index(term): | |
| return names[names.name==term].i.values[0] | |
| def get_target_names(listtargets,top_words): | |
| return [names[names.i==target].name.values[0] for target in listtargets][:top_words] | |
| def get_associations(term,top_words=10): | |
| term = term.upper() | |
| return pd.DataFrame(get_target_names(get_targets_from_source(get_index(term)),top_words),columns=['associations']) | |
| iface = gr.Interface(fn=get_associations, inputs="text", outputs=gr.Dataframe(headers=['associations']),examples=["cat"]).launch(share=False) | |