updating app

app.py

@@ -78,8 +78,11 @@ def main():
             st.header('Entity-Relation Representation')
             st.graphviz_chart(st.session_state['graph'], use_container_width=True)
 
+            st.session_state['text'] = txt
+
         else:
             st.session_state['text'] = txt
+            st.session_state['graph'] = None
 
             # preprocess
             sentence = ie.preprocess(txt)
@@ -121,8 +124,8 @@ def main():
             single_submit=False
         )
 
-        st.session_state['text'] = ''
-        st.session_state['graph'] = None
+        # st.session_state['text'] = ''
+        # st.session_state['graph'] = None
 
 
 if __name__ == '__main__':