| import streamlit as st | |
| from pyvis.network import Network | |
| from functions import * | |
| import streamlit.components.v1 as components | |
| import pickle, math | |
| st.set_page_config(page_title="Earnings Knowledge Graph", page_icon="📈") | |
| st.sidebar.header("Knowledge Graph") | |
| st.markdown("## Earnings Knowledge Graph") | |
| filename = "earnings_network.html" | |
| if "earnings_passages" in st.session_state: | |
| with st.spinner(text='Loading Babelscape/rebel-large which can take a few minutes to generate the graph..'): | |
| st.session_state.kb_text = from_text_to_kb(st.session_state['earnings_passages'], kg_model, kg_tokenizer, "", verbose=True) | |
| save_network_html(st.session_state.kb_text, filename=filename) | |
| st.session_state.kb_chart = filename | |
| with st.container(): | |
| st.subheader("Generated Knowledge Graph") | |
| st.markdown("*You can interact with the graph and zoom.*") | |
| html_source_code = open(st.session_state.kb_chart, 'r', encoding='utf-8').read() | |
| components.html(html_source_code, width=700, height=700) | |
| st.markdown(st.session_state.kb_text) | |
| else: | |
| st.write('No earnings text detected, please regenerate from Home page..') | |