update

app.py

@@ -20,10 +20,13 @@ wiki_state_variables = {
     'wiki_text' : [],
     'nodes':[],
     "topics":[],
+    "html":""
 }
 
 free_text_state_variables = {
     'has_run_free':False,
+    "html":""
+
 }
 
 def wiki_init_state_variables():
@@ -35,6 +38,7 @@ def wiki_init_state_variables():
         if k not in st.session_state:
             st.session_state[k] = v
 
+
 def wiki_generate_graph():
     st.session_state["GRAPH_FILENAME"] = str(dt.now().timestamp()*1000) + ".html"
 
@@ -47,6 +51,10 @@ def wiki_generate_graph():
         texts = st.session_state['wiki_text']
         st.session_state['nodes'] = []
         nodes = rebel.generate_knowledge_graph(texts, st.session_state["GRAPH_FILENAME"])
+        HtmlFile = open(st.session_state["GRAPH_FILENAME"], 'r', encoding='utf-8')
+        source_code = HtmlFile.read()
+        st.session_state["html"] = source_code
+        os.remove(st.session_state["GRAPH_FILENAME"])
         print("gen_graph", nodes)
         for n in nodes:
             n = n.lower()
@@ -100,6 +108,10 @@ def free_text_generate():
     st.session_state["GRAPH_FILENAME"] = str(dt.now().timestamp()*1000) + ".html"
     text = st.session_state['free_text'][0:500]
     rebel.generate_knowledge_graph([text], st.session_state["GRAPH_FILENAME"])
+    HtmlFile = open(st.session_state["GRAPH_FILENAME"], 'r', encoding='utf-8')
+    source_code = HtmlFile.read()
+    st.session_state["html"] = source_code
+    os.remove(st.session_state["GRAPH_FILENAME"])
     st.session_state['has_run_free'] = True
 
 def free_text_layout():
@@ -167,10 +179,8 @@ def show_wiki_hub_page():
     print(st.session_state)
 
     if st.session_state['has_run_wiki']:
-        HtmlFile = open(st.session_state["GRAPH_FILENAME"], 'r', encoding='utf-8')
-        source_code = HtmlFile.read()
-        components.html(source_code, width=720, height=600)
-        os.remove(st.session_state["GRAPH_FILENAME"])
+
+        components.html(st.session_state("html"), width=720, height=600)
         num_buttons = len(st.session_state["nodes"])
         num_cols = num_buttons if 0 < num_buttons < 7 else 7
         columns = st.columns([1] * num_cols + [1])